Update app.py

app.py

@@ -93,6 +93,7 @@ def infer_video_depth(
         os.makedirs(output_dir)
 
     # Save the preprocessed (RGB) video and the generated depth visualization.
+    # Still process the video, but we won't display it in the UI
     processed_video_path = os.path.join(output_dir, os.path.splitext(video_name)[0] + '_src.mp4')
     depth_vis_path = os.path.join(output_dir, os.path.splitext(video_name)[0] + '_vis.mp4')
     save_video(frames, processed_video_path, fps=fps)
@@ -163,8 +164,8 @@ def infer_video_depth(
     gc.collect()
     torch.cuda.empty_cache()
 
-    # Return the preprocessed RGB video, depth visualization, and (if created) the stitched video.
-    return [processed_video_path, depth_vis_path, stitched_video_path]
+    # Only return the depth visualization and stitched video (not the preprocessed video)
+    return [depth_vis_path, stitched_video_path]
 
 def construct_demo():
     with gr.Blocks(analytics_enabled=False) as demo:
@@ -178,8 +179,7 @@ def construct_demo():
                 input_video = gr.Video(label="Input Video")
             with gr.Column(scale=2):
                 with gr.Row(equal_height=True):
-                    # Removed unsupported parameters for Gradio 5.x
-                    processed_video = gr.Video(label="Preprocessed Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
+                    # Removed the processed_video component from the UI
                     depth_vis_video = gr.Video(label="Generated Depth Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
                     stitched_video = gr.Video(label="Stitched RGBD Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
                     
@@ -202,7 +202,7 @@ def construct_demo():
         generate_btn.click(
             fn=infer_video_depth,
             inputs=[input_video, max_len, target_fps, max_res, stitch_option, grayscale_option, convert_from_color_option, blur_slider],
-            outputs=[processed_video, depth_vis_video, stitched_video],
+            outputs=[depth_vis_video, stitched_video],  # Changed to only include the outputs we want to show
         )
     
     return demo