Update app.py

app.py

@@ -190,9 +190,7 @@ with gr.Blocks() as demo:
                 ]
             
             mode.change(toggle_inputs, inputs=mode, outputs=[media_upload, url])
-            media_upload.change(show_preview_from_upload, inputs=media_upload, outputs=[img_out, vid_out])
-            url.submit(show_preview_from_url, inputs=url, outputs=[img_out, vid_out])
-
+           
             # Visibility logic function
             def toggle_visibility(checked):
                 return gr.update(visible=checked)
@@ -253,8 +251,19 @@ with gr.Blocks() as demo:
             json_out = gr.JSON(label="Scene JSON")
             zip_out = gr.File(label="Download Results")
 
-                
-    
+    media_upload.change(show_preview_from_upload, inputs=media_upload, outputs=[img_out, vid_out])
+    url.submit(show_preview_from_url, inputs=url, outputs=[img_out, vid_out])
+        
+     # Unified run click → switch visibility based on image or video output
+    def route_output(image_output, json_output, zip_file):
+        # Show img_out if image was returned, else show video
+        if isinstance(image_output, Image.Image):
+            return gr.update(value=image_output, visible=True), gr.update(visible=False), json_output, zip_file
+        elif isinstance(zip_file, str) and zip_file.endswith(".mp4"):
+            return gr.update(visible=False), gr.update(value=zip_file, visible=True), json_output, zip_file
+        else:
+            return gr.update(visible=False), gr.update(visible=False), json_output, zip_file
+
     
     # # Output Tabs
     # with gr.Tab("Scene JSON"):