Update app.py

app.py

@@ -1,25 +1,38 @@
-import gradio as gr
-from bert_explainer import analyze_text
-
-def predict_fn(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
-
-iface =gr.Interface(
-    fn=predict_fn,
-    inputs = [
-        gr.TextArea(label="輸入訊息"),
-        gr.Radio(choices=["cnn", "bert", "both"], label="分析模式", value="cnn")
-    ],
-    outputs = [
-        gr.Textbox(label = "判斷結果"),
-        gr.Textbox(label = "可疑分數"),
-        gr.Textbox(label = "可疑詞彙")
-    ],
-    title= "預判詐騙訊息",
-    description="輸入訊息，AI 將判定是否為詐騙並標記可疑詞  "
-)
-iface.launch()
+import gradio as gr
+from bert_explainer import analyze_text, analyze_image
+
+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
+
+
+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
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 📩 預測詐騙訊息 (文字 + 圖片)")
+    
+    with gr.Tab("文字預測"):
+        text_inputs = gr.TextArea(label="輸入訊息"),
+        mode_radio_1 = gr.Radio(choices=["cnn", "bert", "both"], label="分析模式", value="cnn")
+        text_buotton = gr.Button("開始分析")
+        text_result =[gr.Textbox(label = "判斷結果"),gr.Textbox(label = "可疑分數"),gr.Textbox(label = "可疑詞彙")]
+        text_buotton.click(fn=predict_text, inputs=[text_inputs, mode_radio_1], outputs=text_result)
+    
+    with gr.Tab("圖片預測"):
+        image_inputs = gr.TextArea(label="上傳圖片"),
+        mode_radio_2 = gr.Radio(choices=["cnn", "bert", "both"], label="分析模式", value="cnn")
+        image_buotton = gr.Button("開始分析")
+        image_result =[gr.Textbox(label = "判斷結果"),gr.Textbox(label = "可疑分數"),gr.Textbox(label = "可疑詞彙")]
+        image_buotton.click(fn=predict_image, inputs=[image_inputs, mode_radio_2], outputs=image_result)
+
+
+
+demo.launch()