Update app.py

app.py

@@ -1,53 +1,31 @@
 import gradio as gr
 from bert_explainer import analyze_text, analyze_image
+import fastapi
+import uvicorn
 
+# 定義函式
 def predict_text(text, mode):
     result = analyze_text(text=text, explain_mode=mode)
-    status = result['status']
-    confidence = f"{result['confidence']}%"
-    keywords = ', '.join(result['suspicious_keywords'])
-    return status, confidence, keywords
-
+    return result['status'], f"{result['confidence']}%", ', '.join(result['suspicious_keywords'])
 
 def predict_image(file, mode):
-    result = analyze_image(file.read(), explain_mode= mode)
-    status = result['status']
-    confidence = f"{result['confidence']}%"
-    keywords = ', '.join(result['suspicious_keywords'])
-    return status, confidence, keywords
+    result = analyze_image(file.read(), explain_mode=mode)
+    return result['status'], f"{result['confidence']}%", ', '.join(result['suspicious_keywords'])
 
-text_interface = gr.Interface(
-    fn=predict_text,
-    inputs=[
-        gr.Textbox(label="輸入訊息"),
-        gr.Radio(choices=["cnn", "bert", "both"], label="分析模式", value="cnn")
-    ],
-    outputs=[
-        gr.Textbox(label="判斷結果"),
-        gr.Textbox(label="可疑分數"),
-        gr.Textbox(label="可疑詞彙")
-    ],
-    allow_flagging="never"
-)
+# 定義 Gradio 接口
+text_api = gr.Interface(fn=predict_text,
+    inputs=[gr.Textbox(), gr.Radio(["cnn", "bert", "both"], value="cnn")],
+    outputs=["text", "text", "text"])
 
-image_interface = gr.Interface(
-    fn=predict_image,
-    inputs=[
-        gr.Image(type="filepath", label="上傳圖片"),
-        gr.Radio(choices=["cnn", "bert", "both"], label="分析模式", value="cnn")
-    ],
-    outputs=[
-        gr.Textbox(label="判斷結果"),
-        gr.Textbox(label="可疑分數"),
-        gr.Textbox(label="可疑詞彙")
-    ],
-    allow_flagging="never"
-)
+image_api = gr.Interface(fn=predict_image,
+    inputs=[gr.Image(type="filepath"), gr.Radio(["cnn", "bert", "both"], value="cnn")],
+    outputs=["text", "text", "text"])
 
-# 整合成 Tab 介面
-demo = gr.TabbedInterface(
-    interface_list=[text_interface, image_interface],
-    tab_names=["文字預測", "圖片預測"]
-)
+# FastAPI 設定（取代 launch）
+app = fastapi.FastAPI()
+app = gr.mount_gradio_app(app, text_api, path="/run/predict_text")
+app = gr.mount_gradio_app(app, image_api, path="/run/predict_image")
 
-demo.launch()
+# 本地測試時使用
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)