Update app.py

app.py

@@ -82,35 +82,45 @@ def get_hf_rate_limit():
         return "無法取得 API 速率資訊"
 
 # -------------------------------
-# 6. 生成文章
+# 6. 生成文章（加入進度顯示）
 # -------------------------------
-def generate_article_with_rate(query, segments=5):
+def generate_article_with_progress(query, segments=5):
+    import time
     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 gr.Textbox.update(value=f"{rate_info}\n\n開始生成文章...\n")
+    
     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}）"
+        
         all_text.append(paragraph)
         doc.add_paragraph(paragraph)
         prompt = f"請接續上一段生成下一段：\n{paragraph}\n\n下一段："
 
+        # 更新進度文字
+        current_text = "\n\n".join(all_text)
+        yield gr.Textbox.update(value=f"{rate_info}\n\n{current_text}\n\n正在生成第 {i+1} 段 / {segments} ...")
+    
+    # 保存 DOCX
     doc.save(docx_file)
     full_text = "\n\n".join(all_text)
-    rate_info = get_hf_rate_limit()
-    return f"{rate_info}\n\n{full_text}", docx_file
+    yield gr.Textbox.update(value=f"{rate_info}\n\n{full_text}"), docx_file
 
 # -------------------------------
-# 7. Gradio 介面
+# 7. Gradio 介面（更新按鈕綁定 generator）
 # -------------------------------
 with gr.Blocks() as demo:
     gr.Markdown("# 佛教經論 RAG 系統 (HF API)")
@@ -122,10 +132,11 @@ with gr.Blocks() as demo:
     output_file = gr.File(label="下載 DOCX")
 
     btn = gr.Button("生成文章")
-    btn.click(generate_article_with_rate, [query_input, segments_input], [output_text, output_file])
+    btn.click(generate_article_with_progress, [query_input, segments_input], [output_text, output_file])
 
 # -------------------------------
-# 8. 啟動 Gradio（Hugging Face Space 適用）
+# 8. 啟動 Gradio
 # -------------------------------
 if __name__ == "__main__":
-    demo.launch(share=True)  # share=True 自動生成公開 URL
+    demo.launch()
+