Update app.py

app.py

@@ -134,19 +134,20 @@ def infer(video_in):
     wave_files = get_wav_files('./outputs/tmp')
     print(wave_files)
     plot_spectrogram(wave_files[0], 'spectrogram.png')
-    
-    return wave_files[0], 'spectrogram.png'
+    final_merged_out = merge_audio_to_video(video_in, wave_files[0])
+    return wave_files[0], 'spectrogram.png', final_merged_out
 
 with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# Video-To-Audio")
         with gr.Row():
-            video_in = gr.Video(label='Video IN')
-            submit_btn = gr.Button("Submit")
-        with gr.Row():
-            output_sound = gr.Audio(label="Audio OUT")
-            output_spectrogram = gr.Image(label='Spectrogram')
-            merged_out = gr.Video(label="Merged video + generated audio")
+            with gr.Column():
+                video_in = gr.Video(label='Video IN')
+                submit_btn = gr.Button("Submit")
+            with gr.Column():
+                output_sound = gr.Audio(label="Audio OUT")
+                output_spectrogram = gr.Image(label='Spectrogram')
+                merged_out = gr.Video(label="Merged video + generated audio")
         
     submit_btn.click(
         fn = infer,