Update app_v2.py

app_v2.py

@@ -120,16 +120,35 @@ async def get_example(name: str, thumb: int = 0):
 
 @server.api(name="anonymize_screenshot")
 def anonymize_screenshot_api(image_path: str) -> str:
-    img = Image.open(image_path).convert("RGB")
-    ocr = ocr_image(img)
-    if not ocr["text"].strip():
-        return json.dumps({"boxes": [], "text": "", "spans": []})
-    _, spans = run_pii_analysis(ocr["text"])
-    boxes = map_spans_to_boxes(ocr["words"], spans)
-    return json.dumps({
-        "width": img.width, "height": img.height,
-        "boxes": boxes, "text": ocr["text"], "spans": spans,
-    }, ensure_ascii=False)
+    """Gradio API: accepts an image (via gradio_client.handle_file) and
+    returns detected redaction boxes as JSON.
+
+    Tolerant of both a plain path string and a FileData dict, since
+    different gradio_client versions pass each.
+    """
+    import traceback
+    try:
+        if isinstance(image_path, dict):
+            image_path = image_path.get("path") or image_path.get("url") or ""
+        if not image_path or not isinstance(image_path, str):
+            return json.dumps({"error": f"expected image path, got {type(image_path).__name__}"})
+
+        img = Image.open(image_path).convert("RGB")
+        ocr = ocr_image(img)
+        if not ocr["text"].strip():
+            return json.dumps({
+                "width": img.width, "height": img.height,
+                "boxes": [], "text": "", "spans": [],
+            })
+        _, spans = run_pii_analysis(ocr["text"])
+        boxes = map_spans_to_boxes(ocr["words"], spans)
+        return json.dumps({
+            "width": img.width, "height": img.height,
+            "boxes": boxes, "text": ocr["text"], "spans": spans,
+        }, ensure_ascii=False)
+    except Exception as e:
+        traceback.print_exc()
+        return json.dumps({"error": f"{type(e).__name__}: {e}"})
 
 
 # =====================================================================
@@ -1278,4 +1297,4 @@ function toast(msg) {
 
 
 if __name__ == "__main__":
-    server.launch(server_name="0.0.0.0", server_port=7860)
+    server.launch(server_name="0.0.0.0", server_port=7860, show_error=True)