Bennie12 commited on
Commit
21569c9
·
verified ·
1 Parent(s): a84bee6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -7
app.py CHANGED
@@ -53,21 +53,26 @@ def predict_text_api(payload: dict):
53
  @api.post("/run/predict_image")
54
  async def predict_image_api(file: UploadFile = File(...), explain_mode: str = Form(...)):
55
  try:
 
 
 
 
 
56
  img_bytes = await file.read()
57
  if not img_bytes:
58
- raise ValueError("未收到圖片資料")
59
- result = analyze_image(img_bytes, explain_mode = explain_mode)
 
60
 
61
  return {
62
- "data": [
63
- result["status"],
64
- f'{result["confidence"]}%',
65
- ", ".join(result["suspicious_keywords"])
66
- ]
67
  }
68
 
69
  except Exception as e:
70
  return {"error": str(e)}
 
71
  # ✅ Gradio UI 功能
72
  def predict_text(text, mode):
73
  result = analyze_text(text=text, explain_mode=mode)
 
53
  @api.post("/run/predict_image")
54
  async def predict_image_api(file: UploadFile = File(...), explain_mode: str = Form(...)):
55
  try:
56
+ if not file:
57
+ raise ValueError("未上傳圖片")
58
+ if not explain_mode:
59
+ raise ValueError("未指定分析模式")
60
+
61
  img_bytes = await file.read()
62
  if not img_bytes:
63
+ raise ValueError("圖片內容為空")
64
+
65
+ result = analyze_image(img_bytes, explain_mode=explain_mode)
66
 
67
  return {
68
+ "status": result["status"],
69
+ "confidence": f'{result["confidence"]}%',
70
+ "suspicious_keywords": result["suspicious_keywords"]
 
 
71
  }
72
 
73
  except Exception as e:
74
  return {"error": str(e)}
75
+
76
  # ✅ Gradio UI 功能
77
  def predict_text(text, mode):
78
  result = analyze_text(text=text, explain_mode=mode)