Update app.py

app.py

@@ -444,6 +444,27 @@ def decode_b64_image(b64_str: str) -> Image.Image:
     raw = base64.b64decode(b64_str)
     return Image.open(io.BytesIO(raw)).convert("RGB")
 
+
+def fetch_image_from_url(url: str, timeout: int = 20) -> Image.Image:
+    """Download an image from a URL and return it as a PIL Image."""
+    resp = requests.get(url, timeout=timeout)
+    resp.raise_for_status()
+    content_type = resp.headers.get("Content-Type", "")
+    if content_type and not content_type.startswith("image/"):
+        raise ValueError(f"URL did not return an image (Content-Type: {content_type})")
+    return Image.open(io.BytesIO(resp.content)).convert("RGB")
+
+
+def load_image(image_url: str) -> Image.Image:
+    """
+    Load an image from either a URL (http/https) or a base64 string / data-URI.
+    This is the single entry point for all image loading in the inspect route.
+    """
+    stripped = image_url.strip()
+    if stripped.startswith("http://") or stripped.startswith("https://"):
+        return fetch_image_from_url(stripped)
+    return decode_b64_image(stripped)
+
 # ──────────────────────────────────────────────────────────────────────────────
 # JSON EXTRACTION
 # ──────────────────────────────────────────────────────────────────────────────
@@ -749,13 +770,16 @@ class ImageInput(BaseModel):
         description="One of: front_right, front_left, rear_right, rear_left, inside, door",
         example="front_left",
     )
-    image_base64: str = Field(
+    image_url: str = Field(
         ...,
         description=(
-            "Base64-encoded image. Can be raw base64 or a data URI "
+            "Image source — either a public/signed HTTPS URL "
+            "(e.g. a Firebase Storage download URL) "
+            "OR a base64-encoded string / data-URI "
             "(e.g. 'data:image/jpeg;base64,...'). "
             "Supported formats: JPEG, PNG, WEBP."
         ),
+        example="https://firebasestorage.googleapis.com/...",
     )
 
 
@@ -766,12 +790,12 @@ class InspectRequest(BaseModel):
         max_length=6,
         description="List of labeled images. Each label may appear at most once.",
         example=[
-            {"label": "front_left",  "image_base64": "<base64>"},
-            {"label": "front_right", "image_base64": "<base64>"},
-            {"label": "rear_left",   "image_base64": "<base64>"},
-            {"label": "rear_right",  "image_base64": "<base64>"},
-            {"label": "inside",      "image_base64": "<base64>"},
-            {"label": "door",        "image_base64": "<base64>"},
+            {"label": "front_left",  "image_url": "https://firebasestorage.googleapis.com/..."},
+            {"label": "front_right", "image_url": "https://firebasestorage.googleapis.com/..."},
+            {"label": "rear_left",   "image_url": "https://firebasestorage.googleapis.com/..."},
+            {"label": "rear_right",  "image_url": "https://firebasestorage.googleapis.com/..."},
+            {"label": "inside",      "image_url": "https://firebasestorage.googleapis.com/..."},
+            {"label": "door",        "image_url": "https://firebasestorage.googleapis.com/..."},
         ],
     )
 
@@ -803,7 +827,7 @@ def inspect(request: InspectRequest):
     """
     Run full trailer inspection on all submitted images in parallel.
 
-    **Input:** Up to 6 labeled base64 images.
+    **Input:** Up to 6 labeled images — each as a signed/public HTTPS URL or base64 string.
     **Output:** Per-label report with component detection results.
 
     Labels accepted: `front_right`, `front_left`, `rear_right`, `rear_left`, `inside`, `door`
@@ -832,17 +856,17 @@ def inspect(request: InspectRequest):
                 status_code=422,
                 detail=f"Duplicate label '{item.label}'. Each label may only appear once.",
             )
-        seen_labels[item.label] = item.image_base64
+        seen_labels[item.label] = item.image_url
 
-    # Decode all images
+    # Load all images (URL download or base64 decode)
     decoded: dict[str, Image.Image] = {}
-    for label, b64 in seen_labels.items():
+    for label, image_url in seen_labels.items():
         try:
-            decoded[label] = decode_b64_image(b64)
+            decoded[label] = load_image(image_url)
         except Exception as e:
             raise HTTPException(
                 status_code=422,
-                detail=f"Could not decode image for label '{label}': {e}",
+                detail=f"Could not load image for label '{label}': {e}",
             )
 
     # ── Run label analyses in small batches ─────────────────────────────────