fix: catch all inference exceptions and OSError in SSRF check to surface real 500 cause

app/main.py

@@ -82,10 +82,15 @@ def _validate_url(url: str) -> str:
     try:
         for info in socket.getaddrinfo(hostname, None):
             addr = info[4][0]
-            ip = ipaddress.ip_address(addr)
-            if ip.is_private or ip.is_loopback or ip.is_link_local:
-                raise HTTPException(status_code=400, detail="URL resolves to a private address")
-    except socket.gaierror as exc:
+            try:
+                ip = ipaddress.ip_address(addr)
+                if ip.is_private or ip.is_loopback or ip.is_link_local:
+                    raise HTTPException(status_code=400, detail="URL resolves to a private address")
+            except ValueError:
+                pass  # skip unparseable addresses (e.g. scoped IPv6)
+    except HTTPException:
+        raise
+    except OSError as exc:
         raise HTTPException(status_code=400, detail="Cannot resolve hostname") from exc
     return url
 
@@ -93,6 +98,9 @@ def _validate_url(url: str) -> str:
 @app.post("/analyze", response_model=PredictResponse, dependencies=[Depends(_require_api_key)])
 def analyze(body: AnalyzeRequest) -> PredictResponse:
     """Accept a public image URL, download it, and run inference."""
+    import logging
+    logger = logging.getLogger(__name__)
+
     _validate_url(body.image_url)
     try:
         resp = http_requests.get(body.image_url, timeout=30)
@@ -105,5 +113,9 @@ def analyze(body: AnalyzeRequest) -> PredictResponse:
     except UnidentifiedImageError as exc:
         raise HTTPException(status_code=400, detail="URL did not return a valid image") from exc
 
-    result = model.predict(image)
-    return PredictResponse(**result)
+    try:
+        result = model.predict(image)
+        return PredictResponse(**result)
+    except Exception as exc:
+        logger.exception("Model inference failed")
+        raise HTTPException(status_code=500, detail=f"Model inference error: {type(exc).__name__}: {exc}") from exc