Update gradio_app.py

gradio_app.py

@@ -109,17 +109,30 @@ def separate_dnr_video(video_path):
     return dialog_video, effect_video, music_video
 
 @spaces.GPU()
-def separate_speakers(audio_path):
+def separate_speakers_video(video_path):
+    audio_path, video = extract_audio_from_video(video_path)
     output_files = separate_speakers_core(audio_path)
+
+    session_id = uuid.uuid4().hex[:8]
+    output_dir = os.path.join("output_sep_video", session_id)
+    os.makedirs(output_dir, exist_ok=True)
+
+    output_videos = []
+    for i, audio_file in enumerate(output_files):
+        speaker_video_path = os.path.join(output_dir, f"speaker_{i+1}_video.mp4")
+        video_with_sep_audio = attach_audio_to_video(video, audio_file, speaker_video_path)
+        output_videos.append(video_with_sep_audio)
+
     updates = []
     for i in range(MAX_SPEAKERS):
-        if i < len(output_files):
-            updates.append(gr.update(value=output_files[i], visible=True, label=f"Speaker {i+1}"))
+        if i < len(output_videos):
+            updates.append(gr.update(value=output_videos[i], visible=True, label=f"Speaker {i+1}"))
         else:
             updates.append(gr.update(value=None, visible=False))
     return updates
 
 
+
 # --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("# TIGER: Time-frequency Interleaved Gain Extraction and Reconstruction for Efficient Speech Separation")