Update app.py

app.py

@@ -98,32 +98,37 @@ def run_shield_check(image: Image.Image) -> dict:
 
 @app.post("/moderate-image")
 def moderate_image(request: ImageRequest):
-    image = load_image(str(request.image_url))
-
-    nsfw_score = run_nsfw_check(image)
-    shield_scores = run_shield_check(image)
-
-    reasons = []
-    if nsfw_score >= THRESHOLD:
-        reasons.append(
-            f"NSFW classifier flagged the image as explicit ({nsfw_score:.0%} confidence)"
-        )
-    for policy, score in shield_scores.items():
-        if score >= THRESHOLD:
+    try:
+        image = load_image(str(request.image_url))
+
+        nsfw_score = run_nsfw_check(image)
+        shield_scores = run_shield_check(image)
+    
+        reasons = []
+        if nsfw_score >= THRESHOLD:
             reasons.append(
-                f"ShieldGemma 2 flagged {POLICIES[policy]} ({score:.0%} confidence)"
+                f"NSFW classifier flagged the image as explicit ({nsfw_score:.0%} confidence)"
             )
-
-    is_safe = not reasons
-    return {
-        "rating": "safe" if is_safe else "not safe",
-        "reason": (
-            "No policy violations detected by either model."
-            if is_safe
-            else "; ".join(reasons)
-        ),
-        "scores": {
-            "nsfw": round(nsfw_score, 4),
-            **{k: round(v, 4) for k, v in shield_scores.items()},
-        },
-    }
+        for policy, score in shield_scores.items():
+            if score >= THRESHOLD:
+                reasons.append(
+                    f"ShieldGemma 2 flagged {POLICIES[policy]} ({score:.0%} confidence)"
+                )
+    
+        is_safe = not reasons
+        return {
+            "rating": "safe" if is_safe else "not safe",
+            "reason": (
+                "No policy violations detected by either model."
+                if is_safe
+                else "; ".join(reasons)
+            ),
+            "scores": {
+                "nsfw": round(nsfw_score, 4),
+                **{k: round(v, 4) for k, v in shield_scores.items()},
+            },
+        }
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"{type(e).__name__}: {e}\n{traceback.format_exc()}")