Update app.py

app.py

@@ -10,8 +10,6 @@ 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. 環境變數與資料路徑
@@ -83,50 +81,40 @@ def get_hf_rate_limit():
     except Exception:
         return "無法取得 API 速率資訊"
 
-
 # -------------------------------
-# 6. 生成文章（進度條 + 美化換行）
+# 6. 生成文章（可即時顯示進度）
 # -------------------------------
-def generate_article_with_progress_bar(query, segments=5):
+def generate_article_progress(query, segments=5):
     docx_file = "/tmp/generated_article.docx"
     doc = DocxDocument()
     doc.add_heading(query, level=1)
 
     all_text = []
     prompt = f"請依據下列主題生成段落：{query}\n\n每段約150-200字。"
-    rate_info = get_hf_rate_limit()
-
-    # 初始化
-    yield 0, gr.Textbox.update(value=f"{rate_info}\n\n開始生成文章...\n"), None
 
     for i in range(int(segments)):
         try:
             result = qa_chain({"query": prompt})
             paragraph = result.get("result", "").strip()
             if not paragraph:
-                paragraph = "（本段生成失敗，請稍後再試。）"
+                paragraph = "（本段生成失敗，請嘗試減少段落或改用較小模型。）"
         except Exception as e:
             paragraph = f"（本段生成失敗：{e}）"
 
-        # 美化換行，每行約 60 字
-        wrapped_paragraph = "\n".join(textwrap.wrap(paragraph, width=60))
-        all_text.append(wrapped_paragraph)
-        doc.add_paragraph(paragraph)  # DOCX 保留原文，不換行
+        all_text.append(paragraph)
+        doc.add_paragraph(paragraph)
         prompt = f"請接續上一段生成下一段：\n{paragraph}\n\n下一段："
 
-        # 更新進度
-        progress = int((i + 1) / int(segments) * 100)
-        current_text = "\n\n".join(all_text)
-        yield progress, gr.Textbox.update(value=f"{rate_info}\n\n{current_text}\n\n正在生成第 {i+1} 段 / {segments} ..."), None
+        # yield 即時更新 Textbox
+        yield "\n\n".join(all_text), None
 
-    # 保存 DOCX
     doc.save(docx_file)
-    full_text = "\n\n".join(all_text)
-    yield 100, gr.Textbox.update(value=f"{rate_info}\n\n{full_text}"), docx_file
+    rate_info = get_hf_rate_limit()
+    yield f"{rate_info}\n\n" + "\n\n".join(all_text), docx_file
 
 
 # -------------------------------
-# 7. Gradio 介面（進度條 + 文字 + 文件）
+# 7. Gradio 介面（支援進度更新）
 # -------------------------------
 with gr.Blocks() as demo:
     gr.Markdown("# 佛教經論 RAG 系統 (HF API)")
@@ -134,21 +122,15 @@ with gr.Blocks() as demo:
 
     query_input = gr.Textbox(lines=2, placeholder="請輸入文章主題", label="文章主題")
     segments_input = gr.Slider(minimum=1, maximum=10, step=1, value=5, label="段落數")
-    progress_bar = gr.Progress()
     output_text = gr.Textbox(label="生成文章 + API 剩餘次數")
     output_file = gr.File(label="下載 DOCX")
 
     btn = gr.Button("生成文章")
-    btn.click(
-        generate_article_with_progress_bar,
-        [query_input, segments_input],
-        [progress_bar, output_text, output_file]
-    )
+    btn.click(generate_article_progress, [query_input, segments_input], [output_text, output_file])
+
 
 # -------------------------------
-# 8. 啟動 Gradio
+# 8. 啟動 Gradio（Hugging Face Space 適用）
 # -------------------------------
 if __name__ == "__main__":
     demo.launch()
-
-