Update app.py

app.py

@@ -206,7 +206,7 @@ def validate_video(path):
         return False, str(e)
 
 # Input Resolution
-def resolve_input(mode, uploaded_files, url):
+def resolve_input(mode, media_upload, url):
     """
     Resolves the media input based on selected mode.
     - If mode is 'Upload', accepts either:
@@ -216,7 +216,7 @@ def resolve_input(mode, uploaded_files, url):
     
     Args:
         mode (str): 'Upload' or 'URL'
-        uploaded_files (List[Union[PIL.Image.Image, str]]): Uploaded media
+        media_upload (List[Union[PIL.Image.Image, str]]): Uploaded media
         url (str): URL to image or video
 
     Returns:
@@ -226,12 +226,12 @@ def resolve_input(mode, uploaded_files, url):
         logger.info(f"Resolving input for mode: {mode}")
 
         if mode == "Upload":
-            if not uploaded_files:
+            if not media_upload:
                 logger.warning("No upload detected.")
                 return None
 
-            image_files = [f for f in uploaded_files if isinstance(f, Image.Image)]
-            video_files = [f for f in uploaded_files if isinstance(f, str) and f.lower().endswith((".mp4", ".mov", ".avi"))]
+            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"))]
 
             if image_files and video_files:
                 logger.warning("Mixed media upload not supported (images + video).")
@@ -393,7 +393,7 @@ def process_image(
         return None, {"error": str(e)}, None
 
 # Main Handler
-def handle(mode, uploaded_files, url, run_det, det_model, det_confidence, run_seg, seg_model, run_depth, depth_model, blend):
+def handle(mode, media_upload, url, run_det, det_model, det_confidence, run_seg, seg_model, run_depth, depth_model, blend):
     """
     Master handler for resolving input and processing.
     Returns outputs for Gradio interface.
@@ -402,7 +402,7 @@ def handle(mode, uploaded_files, url, run_det, det_model, det_confidence, run_se
     logger.info(f"Session ID: {session_id} | Handler activated with mode: {mode}")
     start_time = time.time()
 
-    media = resolve_input(mode, uploaded_files, url)
+    media = resolve_input(mode, media_upload, url)
     if not media:
         return None, format_error("No valid input provided. Please check your upload or URL."), None
 
@@ -495,7 +495,7 @@ with gr.Blocks() as demo:
     # Button Click Event
     run.click(
         handle,
-        inputs=[mode, uploaded_files, url, run_det, det_model, det_confidence, run_seg, seg_model, run_depth, depth_model, blend],
+        inputs=[mode, media_upload, url, run_det, det_model, det_confidence, run_seg, seg_model, run_depth, depth_model, blend],
         outputs=[img_out, json_out, zip_out]
     )