Update app.py

app.py

@@ -82,10 +82,9 @@ def get_hf_rate_limit():
         return "無法取得 API 速率資訊"
 
 # -------------------------------
-# 6. 生成文章（加入進度顯示）
+# 6. 生成文章（進度條 + 文字更新）
 # -------------------------------
-def generate_article_with_progress(query, segments=5):
-    import time
+def generate_article_with_progress_bar(query, segments=5):
     docx_file = "/tmp/generated_article.docx"
     doc = DocxDocument()
     doc.add_heading(query, level=1)
@@ -94,9 +93,9 @@ def generate_article_with_progress(query, segments=5):
     prompt = f"請依據下列主題生成段落：{query}\n\n每段約150-200字。"
     rate_info = get_hf_rate_limit()
 
-    # 初始化回傳
-    yield gr.Textbox.update(value=f"{rate_info}\n\n開始生成文章...\n")
-    
+    # 初始化
+    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})
@@ -105,22 +104,23 @@ def generate_article_with_progress(query, segments=5):
                 paragraph = "（本段生成失敗，請稍後再試。）"
         except Exception as e:
             paragraph = f"（本段生成失敗：{e}）"
-        
+
         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 gr.Textbox.update(value=f"{rate_info}\n\n{current_text}\n\n正在生成第 {i+1} 段 / {segments} ...")
-    
+        yield progress, gr.Textbox.update(value=f"{rate_info}\n\n{current_text}\n\n正在生成第 {i+1} 段 / {segments} ..."), None
+
     # 保存 DOCX
     doc.save(docx_file)
     full_text = "\n\n".join(all_text)
-    yield gr.Textbox.update(value=f"{rate_info}\n\n{full_text}"), docx_file
+    yield 100, gr.Textbox.update(value=f"{rate_info}\n\n{full_text}"), docx_file
 
 # -------------------------------
-# 7. Gradio 介面（更新按鈕綁定 generator）
+# 7. Gradio 介面（進度條 + 文字 + 文件）
 # -------------------------------
 with gr.Blocks() as demo:
     gr.Markdown("# 佛教經論 RAG 系統 (HF API)")
@@ -128,11 +128,16 @@ 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, [query_input, segments_input], [output_text, output_file])
+    btn.click(
+        generate_article_with_progress_bar,
+        [query_input, segments_input],
+        [progress_bar, output_text, output_file]
+    )
 
 # -------------------------------
 # 8. 啟動 Gradio
@@ -140,3 +145,4 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     demo.launch()
 
+