Update core/input_handler.py

core/input_handler.py

@@ -139,15 +139,30 @@ def resolve_input(mode, media_upload, url):
     """
     try:
         logger.info(f"Resolving input for mode: {mode}")
+logger.info(f"Raw uploaded input: {media_upload}")
 
         if mode == "Upload":
             if not media_upload:
                 logger.warning("No upload detected.")
                 return None
 
-            image_files = [f for f in media_upload if isinstance(f, Image.Image)]
-            video_files = [f for f in media_upload if isinstance(f, str) and f.lower().endswith((".mp4", ".mov", ".avi"))]
-
+            # Gradio gives file paths, so open and classify them
+            image_files = []
+            video_files = []
+
+            for file in media_upload:
+                if isinstance(file, str):
+                    ext = file.lower().split('.')[-1]
+                    if ext in ['jpg', 'jpeg', 'png']:
+                        try:
+                            img = Image.open(file).convert("RGB")
+                            image_files.append(img)
+                        except Exception as e:
+                            logger.warning(f"Failed to open image: {file} - {e}")
+                    elif ext in ['mp4', 'avi', 'mov']:
+                        video_files.append(file)
+
+            # Only one type of input allowed
             if image_files and video_files:
                 logger.warning("Mixed media upload not supported (images + video).")
                 return None