Update app.py

app.py

@@ -72,11 +72,14 @@ def preload_models():
     get_model("segmentation", "deeplabv3_resnet50", device="cpu")
     get_model("depth", "midas_v21_small_256", device="cpu")
 
-# Main Handler
-def handle(mode, media_upload, 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.
+    Returns: (img_out, vid_out, json_out, zip_out)
     """
     session_id = generate_session_id()
     logger.info(f"Session ID: {session_id} | Handler activated with mode: {mode}")
@@ -84,15 +87,25 @@ def handle(mode, media_upload, url, run_det, det_model, det_confidence, run_seg,
 
     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
+        return (
+            gr.update(visible=False),
+            gr.update(visible=False),
+            format_error("No valid input provided. Please check your upload or URL."),
+            None
+        )
 
     first_input = media[0]
-    
+
     # --- VIDEO PATH ---
     if isinstance(first_input, str) and first_input.lower().endswith((".mp4", ".mov", ".avi")):
         valid, err = validate_video(first_input)
         if not valid:
-            return None, format_error(err), None
+            return (
+                gr.update(visible=False),
+                gr.update(visible=False),
+                format_error(err),
+                None
+            )
         try:
             _, msg, output_video_path = process_video(
                 video_path=first_input,
@@ -105,16 +118,31 @@ def handle(mode, media_upload, url, run_det, det_model, det_confidence, run_seg,
                 depth_model=depth_model,
                 blend=blend
             )
-            return gr.update(visible=False), msg, output_video_path  # Hide img_out, show vid_out via click handler
+            return (
+                gr.update(visible=False),  # hide image
+                gr.update(value=output_video_path, visible=True),  # show video
+                msg,
+                output_video_path  # for download
+            )
         except Exception as e:
             logger.error(f"Video processing failed: {e}")
-            return None, format_error(str(e)), None
+            return (
+                gr.update(visible=False),
+                gr.update(visible=False),
+                format_error(str(e)),
+                None
+            )
 
-    # --- IMAGE INPUT ---
+    # --- IMAGE PATH ---
     elif isinstance(first_input, Image.Image):
         valid, err = validate_image(first_input)
         if not valid:
-            return None, format_error(err), None
+            return (
+                gr.update(visible=False),
+                gr.update(visible=False),
+                format_error(err),
+                None
+            )
         try:
             result_img, msg, output_zip = process_image(
                 image=first_input,
@@ -127,17 +155,38 @@ def handle(mode, media_upload, url, run_det, det_model, det_confidence, run_seg,
                 depth_model=depth_model,
                 blend=blend
             )
-            return result_img, msg, output_zip  # Show img_out only
+            return (
+                gr.update(value=result_img, visible=True),  # show image
+                gr.update(visible=False),  # hide video
+                msg,
+                output_zip
+            )
         except timeout_decorator.timeout_decorator.TimeoutError:
             logger.error("Image processing timed out.")
-            return None, format_error("Processing timed out. Try a smaller image or simpler model."), None
+            return (
+                gr.update(visible=False),
+                gr.update(visible=False),
+                format_error("Processing timed out. Try a smaller image or simpler model."),
+                None
+            )
         except Exception as e:
             logger.error(f"Image processing failed: {e}")
-            return None, format_error(str(e)), None
+            return (
+                gr.update(visible=False),
+                gr.update(visible=False),
+                format_error(str(e)),
+                None
+            )
 
     logger.warning("Unsupported media type resolved.")
     log_runtime(start_time)
-    return None, format_error("Unsupported input type."), None
+    return (
+        gr.update(visible=False),
+        gr.update(visible=False),
+        format_error("Unsupported input type."),
+        None
+    )
+
 
 def show_preview_from_upload(files):
     if not files:
@@ -289,7 +338,7 @@ with gr.Blocks() as demo:
 
     # Button Click Event
     run.click(
-    handle,
+    fn=handle,
     inputs=[
         mode, media_upload, url,
         run_det, det_model, det_confidence,
@@ -297,20 +346,14 @@ with gr.Blocks() as demo:
         run_depth, depth_model,
         blend
     ],
-    outputs=[img_out, json_out, zip_out]
+    outputs=[
+        img_out,  # will be visible only if it's an image
+        vid_out,  # will be visible only if it's a video
+        json_out,
+        zip_out
+    ]
 )
 
-    run.click(
-        handle,
-        inputs=[
-            mode, media_upload, url,
-            run_det, det_model, det_confidence,
-            run_seg, seg_model,
-            run_depth, depth_model,
-            blend
-        ],
-        outputs=[vid_out, json_out, zip_out]
-    )
 
     # Footer Section
     gr.Markdown("---")