download_content

app.py

@@ -1,9 +1,11 @@
 import os
+from docx import Document
 import gradio as gr
 from openai import OpenAI
 
-from difflib import Differ
 import random
+import uuid
+
 
 import json
 import tempfile
@@ -650,6 +652,21 @@ def generate_chinese_essay_idea(model, user_prompt, chinese_essay_title_input):
     return content
 
 
+# Download doc
+def create_word(content):
+    unique_filename = str(uuid.uuid4())
+    word_file_path = f"/tmp/{unique_filename}.docx"
+    doc = Document()
+    doc.add_paragraph(content)
+    doc.save(word_file_path)
+    return word_file_path
+
+def download_content(content):
+    word_path = create_word(content)
+    return word_path
+
+
+# === INIT PARAMS ===
 def init_params(request: gr.Request):
     if request:
         print("Request headers dictionary:", request.headers)
@@ -1593,6 +1610,19 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue=gr.themes.colors.blue, secondary
                         with gr.Column():
                             with gr.Row():
                                 gr.Markdown("# Step 9. 寫作完成 Save and Share")
+                            with gr.Row():
+                                # 顯示最後段落寫作結果
+                                with gr.Column():
+                                    paragraph_save_to_doc_button = gr.Button("點擊建立 doc", variant="primary")
+                                with gr.Column():
+                                    paragraph_doc_download_link = gr.File(label="請點擊右下角連結(ex: 37KB)，進行下載")
+
+                                    paragraph_save_to_doc_button.click(
+                                        fn=download_content,
+                                        inputs=[paragraph_refine_input],
+                                        outputs=[paragraph_doc_download_link]
+                                    )
+
                             with gr.Row():
                                 gr.Markdown("## 完成修訂！你按部就班地完成了一次段落寫作練習，太棒了！")
                             with gr.Row():

requirements.txt

@@ -1,2 +1,3 @@
 gradio==4.37.2
-openai>=1.16.2
+openai>=1.16.2
+python-docx