Support remote image URLs in API by fetching images server-side

- Add fetch_image_if_url helper to download images if input is a URL string
- Use this helper for base_img, garment_img, and mask_img in generate function
- Allows both file uploads and remote URLs for all image fields in API calls
- No change to UI behavior

app.py

@@ -331,6 +331,19 @@ def is_nsfw_content(img: Image.Image) -> bool:
 # Main generate function
 # -----------------------------------------------------------------------------
 
+def fetch_image_if_url(img):
+    """
+    If img is a string and looks like a URL, download and return as PIL.Image.
+    Otherwise, return as-is (assume already PIL.Image).
+    """
+    if isinstance(img, str) and (img.startswith("http://") or img.startswith("https://")):
+        print(f"[FETCH] Downloading image from URL: {img}")
+        resp = requests.get(img)
+        resp.raise_for_status()
+        from PIL import Image
+        return Image.open(io.BytesIO(resp.content)).convert("RGB")
+    return img
+
 def generate(
     base_img: Image.Image, 
     garment_img: Image.Image, 
@@ -338,6 +351,11 @@ def generate(
     mask_img: Optional[Image.Image],  # NEW: Optional mask parameter  
     request: gr.Request
 ) -> Image.Image:
+    base_img = fetch_image_if_url(base_img)
+    garment_img = fetch_image_if_url(garment_img)
+    if mask_img is not None:
+        mask_img = fetch_image_if_url(mask_img)
+    
     if base_img is None or garment_img is None:
         raise gr.Error("Please provide both images.")