Update app.py

延伸實作 predict_image 版本

app.py

@@ -47,8 +47,29 @@ def predict_text_api(payload: dict):
     except Exception as e:
         return {"error": str(e)}
 
-# ✅ API 路由：圖片分析（POST），這裡可延伸實作 predict_image 版本
+# ✅ API 路由：圖片分析（POST）
+@api.post("/run/predict_image")
+async def predict_image_api(file: UploadFile = File(...), explain_mode: str = Form("cnn")):
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+            temp_file.write(await file.read())
+            temp_file_path = temp_file.name
+
+        with open(temp_file_path, "rb") as f:
+            result = analyze_image(f.read(), explain_mode=explain_mode)
 
+        os.remove(temp_file_path)
+
+        return {
+            "data": [
+                result["status"],
+                f'{result["confidence"]}%',
+                ", ".join(result["suspicious_keywords"])
+            ]
+        }
+
+    except Exception as e:
+        return {"error": str(e)}
 # ✅ Gradio UI 功能
 def predict_text(text, mode):
     result = analyze_text(text=text, explain_mode=mode)