feat: implement fallback to video loading if image processing fails in file handling logic

app.py

@@ -224,7 +224,13 @@ def predict(
             h_content = []
             if turn_files:
                 for f_path in turn_files:
-                    h_content.append({"type": "image", "image": Image.open(f_path).convert("RGB")})
+                    try:
+                        img = Image.open(f_path).convert("RGB")
+                        h_content.append({"type": "image", "image": img})
+                    except Exception:
+                        v_frames = load_video(f_path, max_frames=max_frames)
+                        if v_frames:
+                            h_content.append({"type": "video", "video": v_frames})
             
             if user_text:
                 h_content.append({"type": "text", "text": user_text})
@@ -238,22 +244,17 @@ def predict(
     if files:
         for f in files:
             file_path = f["path"]
-            # Detect if video or image
-            is_video = False
             try:
-                # Basic extension check instead of opening every file
-                ext = os.path.splitext(file_path)[1].lower()
-                if ext in ('.mp4', '.mkv', '.mov', '.avi', '.webm'):
-                    is_video = True
-            except:
-                is_video = False
-
-            if is_video:
-                frames = load_video(file_path, max_frames=max_frames)
-                if frames:
-                    content.append({"type": "video", "video": frames})
-            else:
-                content.append({"type": "image", "image": Image.open(file_path).convert("RGB")})
+                # Try opening as image first
+                img = Image.open(file_path).convert("RGB")
+                content.append({"type": "image", "image": img})
+            except Exception:
+                # If image fails, try as video
+                v_frames = load_video(file_path, max_frames=max_frames)
+                if v_frames:
+                    content.append({"type": "video", "video": v_frames})
+                else:
+                    print(f"Failed to identify file type for: {file_path}")
     
     if message:
         content.append({"type": "text", "text": message})