Update app.py

app.py

@@ -3,29 +3,51 @@ 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)
     return result['status'], f"{result['confidence']}%", ', '.join(result['suspicious_keywords'])
 
-def predict_image(file, mode):
-    result = analyze_image(file.read(), explain_mode=mode)
+# ⛳ 分析圖片
+def predict_image(file_path, mode):
+    with open(file_path, "rb") as f:
+        result = analyze_image(f.read(), explain_mode=mode)
     return result['status'], f"{result['confidence']}%", ', '.join(result['suspicious_keywords'])
 
-# 定義 Gradio 接口
-text_api = gr.Interface(fn=predict_text,
-    inputs=[gr.Textbox(), gr.Radio(["cnn", "bert", "both"], value="cnn")],
-    outputs=["text", "text", "text"])
+# 🚀 建立文字 API
+text_api = gr.Interface(
+    fn=predict_text,
+    inputs=[
+        gr.Textbox(lines=3, placeholder="請輸入訊息", label="輸入文字"),
+        gr.Radio(["cnn", "bert", "both"], value="cnn", label="分析模式")
+    ],
+    outputs=[
+        gr.Textbox(label="判斷結果"),
+        gr.Textbox(label="置信度"),
+        gr.Textbox(label="可疑詞彙")
+    ],
+    title="詐騙文字分析 API",
+    flagging_mode="never"
+)
 
-image_api = gr.Interface(fn=predict_image,
-    inputs=[gr.Image(type="filepath"), gr.Radio(["cnn", "bert", "both"], value="cnn")],
-    outputs=["text", "text", "text"])
+# 🚀 建立圖片 API
+image_api = gr.Interface(
+    fn=predict_image,
+    inputs=[
+        gr.Image(type="filepath", label="上傳圖片"),
+        gr.Radio(["cnn", "bert", "both"], value="cnn", label="分析模式")
+    ],
+    outputs=[
+        gr.Textbox(label="判斷結果"),
+        gr.Textbox(label="置信度"),
+        gr.Textbox(label="可疑詞彙")
+    ],
+    title="詐騙圖片分析 API",
+    flagging_mode="never"
+)
 
-# FastAPI 設定（取代 launch）
+# 🌐 FastAPI
 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")
 
-# 本地測試時使用
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)