Update app.py

app.py

@@ -1,12 +1,29 @@
 import gradio as gr
 from bert_explainer import analyze_text, analyze_image
 from fastapi import FastAPI
-# 🧠 預測函式：供 UI 和 API 共用
+from fastapi.middleware.cors import CORSMiddleware
+import uvicorn
+
+# ✅ 初始化 FastAPI
 api = FastAPI()
 
-@api.get("/test")
+# ✅ 開放 CORS（避免跨網域錯誤）
+api.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# ✅ API 路由：健康檢查
+@api.get("/health")
+def health_check():
+    return {"status": "ok"}
+
+# ✅ API 路由：測試文字分析 GET 方法
+@api.get("/run/predict_text")
 def test_predict_text():
-    # 測試文字直接寫死，不需輸入
     result = analyze_text("這是測試訊息", explain_mode="cnn")
     return {
         "status": result["status"],
@@ -14,6 +31,25 @@ def test_predict_text():
         "suspicious_keywords": result["suspicious_keywords"]
     }
 
+# ✅ API 路由：正式 POST 方法
+@api.post("/run/predict_text")
+def predict_text_api(payload: dict):
+    try:
+        text, mode = payload["data"]
+        result = analyze_text(text=text, explain_mode=mode)
+        return {
+            "data": [
+                result["status"],
+                f'{result["confidence"]}%',
+                ", ".join(result["suspicious_keywords"])
+            ]
+        }
+    except Exception as e:
+        return {"error": str(e)}
+
+# ✅ API 路由：圖片分析（POST），這裡可延伸實作 predict_image 版本
+
+# ✅ Gradio UI 功能
 def predict_text(text, mode):
     result = analyze_text(text=text, explain_mode=mode)
     return result["status"], f"{result['confidence']}%", ", ".join(result["suspicious_keywords"])
@@ -23,7 +59,6 @@ def predict_image(file_path, mode):
         result = analyze_image(f.read(), explain_mode=mode)
     return result["status"], f"{result['confidence']}%", ", ".join(result["suspicious_keywords"])
 
-# 🧩 Gradio UI：兩個 Tab（文字 & 圖片）
 with gr.Blocks() as demo:
     with gr.Tab("文字模式"):
         text_input = gr.Textbox(lines=3, label="輸入文字")
@@ -43,6 +78,8 @@ with gr.Blocks() as demo:
         image_output3 = gr.Textbox(label="可疑詞彙")
         image_btn.click(fn=predict_image, inputs=[image_input, image_mode], outputs=[image_output1, image_output2, image_output3])
 
-# ✅ 讓 Gradio 支援 API 路由 + 互動功能
+# ✅ 啟用 Gradio + FastAPI 整合
+app = gr.mount_gradio_app(api, demo)
+
 if __name__ == "__main__":
-    demo.queue().launch()
+    uvicorn.run(app, host="0.0.0.0", port=7860)