Update app.py

app.py

@@ -156,6 +156,23 @@ def _ffprobe_streams(path: str) -> Optional[dict]:
     except Exception:
         return None
 
+def determine_media_type(src: str, progress=None) -> Tuple[bool, bool]:
+    is_image = is_video = False
+    ext = ext_from_src(src)
+    if ext in IMAGE_EXTS: 
+        is_image = True
+    if ext in VIDEO_EXTS: 
+        is_video = True
+    if is_remote(src):
+        head = safe_head(src)
+        if head:
+            ctype = (head.headers.get("content-type") or "").lower()
+            if ctype.startswith("image/"): 
+                is_image, is_video = True, False
+            elif ctype.startswith("video/"): 
+                is_video, is_image = True, False
+    return is_image, is_video
+
 def worker(url: str, prompt: str, key: str, progress=gr.Progress()):
     try:
         if not url: return ("error", "**Error:** No URL provided.", "")