Update app.py

app.py

@@ -42,22 +42,31 @@ def ext_from_src(src: str) -> str:
 
 def fetch_bytes(src: str, stream_threshold=STREAM_THRESHOLD, timeout=60) -> bytes:
     if is_remote(src):
-        with requests.get(src, timeout=timeout, stream=True) as r:
-            r.raise_for_status()
-            cl = r.headers.get("content-length")
+        # Try HEAD first to detect size; fallback to GET
+        try:
+            h = requests.head(src, timeout=8, allow_redirects=True)
+            h.raise_for_status()
+            cl = h.headers.get("content-length")
             if cl and int(cl) > stream_threshold:
-                fd, path = tempfile.mkstemp()
-                os.close(fd)
-                try:
-                    with open(path, "wb") as f:
-                        for chunk in r.iter_content(8192):
-                            if chunk:
-                                f.write(chunk)
-                    with open(path, "rb") as f:
-                        return f.read()
-                finally:
-                    try: os.remove(path)
-                    except Exception: pass
+                with requests.get(src, timeout=timeout, stream=True) as r:
+                    r.raise_for_status()
+                    fd, path = tempfile.mkstemp()
+                    os.close(fd)
+                    try:
+                        with open(path, "wb") as f:
+                            for chunk in r.iter_content(8192):
+                                if chunk:
+                                    f.write(chunk)
+                        with open(path, "rb") as f:
+                            return f.read()
+                    finally:
+                        try: os.remove(path)
+                        except Exception: pass
+            # small content or no content-length: simple GET
+        except Exception:
+            pass
+        with requests.get(src, timeout=timeout) as r:
+            r.raise_for_status()
             return r.content
     with open(src, "rb") as f:
         return f.read()
@@ -98,12 +107,10 @@ def extract_best_frames_bytes(media_path: str, sample_count: int = 5, timeout_pr
         duration = float(out) if out else None
     except Exception:
         duration = None
-
     if duration and duration > 0:
         timestamps = [(duration * i) / (sample_count + 1) for i in range(1, sample_count + 1)]
     else:
         timestamps = [0.5, 1.0, 2.0][:sample_count]
-
     frames = []
     for i, t in enumerate(timestamps):
         fd, tmp = tempfile.mkstemp(suffix=f"_{i}.jpg")
@@ -259,15 +266,21 @@ def load_preview(url: str):
         if ext in IMAGE_EXTS:
             try:
                 img = Image.open(url)
+                if getattr(img, "is_animated", False):
+                    img.seek(0)
                 return img.convert("RGB"), None, "Image"
             except Exception:
                 return None, None, "Preview failed"
+    # Remote: prefer HEAD, then GET fallback for image bytes
     try:
-        h = requests.head(url, timeout=8, allow_redirects=True)
+        h = requests.head(url, timeout=6, allow_redirects=True)
         ctype = (h.headers.get("content-type") or "").lower()
         if ctype.startswith("video/") or any(url.lower().split("?")[0].endswith(ext) for ext in VIDEO_EXTS):
             return None, url, "Video"
-        r = requests.get(url, timeout=20)
+    except Exception:
+        pass
+    try:
+        r = requests.get(url, timeout=15)
         r.raise_for_status()
         img = Image.open(BytesIO(r.content))
         if getattr(img, "is_animated", False):
@@ -294,7 +307,11 @@ with gr.Blocks(title="Flux Multimodal (fixed)", css=css) as demo:
         return (img if label == "Image" else None, vid if label == "Video" else None, label == "Image", label == "Video")
 
     url_input.change(fn=preview_update, inputs=[url_input], outputs=[preview_image, preview_video, preview_image, preview_video])
-    submit_btn.click(fn=process_media, inputs=[url_input, custom_prompt, api_key], outputs=[final_md])
+
+    def submit_wrapper(url, prompt, key):
+        return process_media(url, prompt, key)
+
+    submit_btn.click(fn=submit_wrapper, inputs=[url_input, custom_prompt, api_key], outputs=[final_md])
 
 if __name__ == "__main__":
     demo.queue().launch()