Update app.py

app.py

@@ -17,7 +17,6 @@ Please refer to our [paper](https://arxiv.org/abs/2501.12375), [project page](ht
 def stitch_rgbd_videos(
     processed_video: str,
     depth_vis_video: str,
-    audio_video: str,
     max_len: int = -1,
     target_fps: int = -1,
     max_res: int = 1280,
@@ -122,7 +121,7 @@ def stitch_rgbd_videos(
             "ffmpeg",
             "-y",
             "-i", stitched_video_path,
-            "-i", audio_video,
+            "-i", processed_video,
             "-c:v", "copy",
             "-c:a", "aac",
             "-map", "0:v:0",
@@ -145,10 +144,11 @@ def construct_demo():
         with gr.Row(equal_height=True):
             with gr.Column(scale=1):
                 # Video input component for file upload.
-                processed_video = gr.Video(label="Input Video")
-                depth_vis_video = gr.Video(label="Depth Video") 
+                processed_video = gr.Video(label="Input Video with Audio")
+                
             with gr.Column(scale=1):
-                audio_video = gr.Video(label="Audio Source")
+                depth_vis_video = gr.Video(label="Depth Video") 
+                
             with gr.Column(scale=2):
                 with gr.Row(equal_height=True):
                     stitched_video = gr.Video(label="Stitched RGBD Video", interactive=False, autoplay=True, loop=True, show_share_button=True, scale=5)
@@ -169,7 +169,7 @@ def construct_demo():
         
         generate_btn.click(
             fn=stitch_rgbd_videos,
-            inputs=[processed_video, depth_vis_video, audio_video, max_len, target_fps, max_res, stitch_option, grayscale_option, convert_from_color_option, blur_slider],
+            inputs=[processed_video, depth_vis_video, max_len, target_fps, max_res, stitch_option, grayscale_option, convert_from_color_option, blur_slider],
             outputs=stitched_video,
         )