Update app.py

app.py

@@ -68,23 +68,36 @@ qa_chain = RetrievalQA.from_chain_type(
 )
 
 # -------------------------------
-# 5. 查詢 API 剩餘額度
+# 5. 檢查 Hugging Face Token 權限
 # -------------------------------
-def get_hf_rate_limit():
+def check_hf_token_permissions():
+    """確認 Token 是否可呼叫 Inference Endpoint"""
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     try:
-        r = requests.get("https://huggingface.co/api/whoami", headers=headers)
+        r = requests.get("https://huggingface.co/api/whoami-v2", headers=headers)
         r.raise_for_status()
         data = r.json()
-        remaining = data.get("rate_limit", {}).get("remaining", "未知")
-        return f"本小時剩餘 API 次數：約 {remaining}"
+        if "allow_inference" in data and data["allow_inference"]:
+            return True
+        return False
     except Exception:
-        return "無法取得 API 速率資訊"
+        return False
+
+token_valid = check_hf_token_permissions()
+if not token_valid:
+    print("⚠ 警告：Hugging Face API Token 權限不足，無法呼叫模型。")
+
 
 # -------------------------------
-# 6. 生成文章（可即時顯示進度）
+# 6. 生成文章（修正版，支援進度顯示）
 # -------------------------------
-def generate_article_progress(query, segments=5):
+def generate_article_with_progress(query, segments=5):
+    if not token_valid:
+        # Token 權限不足，直接返回訊息
+        yield "⚠ API Token 權限不足，請檢查 Token 是否允許呼叫 Inference Endpoint。", None
+        return
+
+    import time
     docx_file = "/tmp/generated_article.docx"
     doc = DocxDocument()
     doc.add_heading(query, level=1)
@@ -105,16 +118,17 @@ def generate_article_progress(query, segments=5):
         doc.add_paragraph(paragraph)
         prompt = f"請接續上一段生成下一段：\n{paragraph}\n\n下一段："
 
-        # yield 即時更新 Textbox
         yield "\n\n".join(all_text), None
+        time.sleep(0.1)
 
     doc.save(docx_file)
     rate_info = get_hf_rate_limit()
-    yield f"{rate_info}\n\n" + "\n\n".join(all_text), docx_file
+    final_text = f"{rate_info}\n\n" + "\n\n".join(all_text)
+    yield final_text, docx_file
 
 
 # -------------------------------
-# 7. Gradio 介面（支援進度更新）
+# 7. Gradio 介面（修正版）
 # -------------------------------
 with gr.Blocks() as demo:
     gr.Markdown("# 佛教經論 RAG 系統 (HF API)")
@@ -126,11 +140,17 @@ with gr.Blocks() as demo:
     output_file = gr.File(label="下載 DOCX")
 
     btn = gr.Button("生成文章")
-    btn.click(generate_article_progress, [query_input, segments_input], [output_text, output_file])
 
+    # 使用 .click() 搭配 generator
+    btn.click(
+        fn=generate_article_with_progress,
+        inputs=[query_input, segments_input],
+        outputs=[output_text, output_file]
+    )
 
 # -------------------------------
-# 8. 啟動 Gradio（Hugging Face Space 適用）
+# 8. 啟動 Gradio（HF Space 適用）
 # -------------------------------
 if __name__ == "__main__":
     demo.launch()
+