Update app.py

app.py

@@ -10,6 +10,8 @@ from langchain_huggingface import HuggingFaceEmbeddings, HuggingFaceEndpoint
 from docx import Document as DocxDocument
 import gradio as gr
 from langchain_community.vectorstores import FAISS
+import textwrap
+
 
 # -------------------------------
 # 2. 環境變數與資料路徑
@@ -81,8 +83,9 @@ def get_hf_rate_limit():
     except Exception:
         return "無法取得 API 速率資訊"
 
+
 # -------------------------------
-# 6. 生成文章（進度條 + 文字更新）
+# 6. 生成文章（進度條 + 美化換行）
 # -------------------------------
 def generate_article_with_progress_bar(query, segments=5):
     docx_file = "/tmp/generated_article.docx"
@@ -105,8 +108,10 @@ def generate_article_with_progress_bar(query, segments=5):
         except Exception as e:
             paragraph = f"（本段生成失敗：{e}）"
 
-        all_text.append(paragraph)
-        doc.add_paragraph(paragraph)
+        # 美化換行，每行約 60 字
+        wrapped_paragraph = "\n".join(textwrap.wrap(paragraph, width=60))
+        all_text.append(wrapped_paragraph)
+        doc.add_paragraph(paragraph)  # DOCX 保留原文，不換行
         prompt = f"請接續上一段生成下一段：\n{paragraph}\n\n下一段："
 
         # 更新進度
@@ -119,6 +124,7 @@ def generate_article_with_progress_bar(query, segments=5):
     full_text = "\n\n".join(all_text)
     yield 100, gr.Textbox.update(value=f"{rate_info}\n\n{full_text}"), docx_file
 
+
 # -------------------------------
 # 7. Gradio 介面（進度條 + 文字 + 文件）
 # -------------------------------