api output fix

gradio_app.py

@@ -249,6 +249,12 @@ def process_media_unified(media_input, pupil_selection, tv_model, blink_detectio
         tuple: (result_image, result_text)
     """
     try:
+        # Handle None input
+        if media_input is None:
+            from PIL import Image
+            error_img = Image.new('RGB', (400, 200), 'white')
+            return error_img, "No media input provided. Please upload an image or video."
+
         # Check if input is an image or video
         if hasattr(media_input, 'name'):
             # It's a file object (video)
@@ -262,9 +268,20 @@ def process_media_unified(media_input, pupil_selection, tv_model, blink_detectio
                 from PIL import Image
                 image = Image.open(file_path)
                 return process_image_gradio(image, pupil_selection, tv_model, blink_detection)
+            else:
+                # Unknown file type
+                from PIL import Image
+                error_img = Image.new('RGB', (400, 200), 'white')
+                return error_img, f"Unsupported file type: {file_path}"
         else:
-            # It's a PIL Image
-            return process_image_gradio(media_input, pupil_selection, tv_model, blink_detection)
+            # It's a PIL Image or other format
+            if media_input is not None:
+                return process_image_gradio(media_input, pupil_selection, tv_model, blink_detection)
+            else:
+                # Fallback for any other None case
+                from PIL import Image
+                error_img = Image.new('RGB', (400, 200), 'white')
+                return error_img, "Invalid media input format."
 
     except Exception as e:
         error_msg = f"Error processing media: {str(e)}"