Update app.py

app.py

@@ -322,9 +322,10 @@ button:hover {
 }
 """
 
-import yt_dlp
+import scipy.io.wavfile
+import os
 
-def download_audio(url, download_type, cookie_file):
+def download_audio(url, cookie_file):
     ydl_opts = {
         'format': 'bestaudio/best',
         'postprocessors': [{
@@ -348,14 +349,19 @@ def download_audio(url, download_type, cookie_file):
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         try:
             info_dict = ydl.extract_info(url, download=True)
-            file_path = ydl.prepare_filename(info_dict)
-            return file_path, "Download successful", None, None, None, None
+            file_path = ydl.prepare_filename(info_dict).replace('.webm', '.wav')  # Adjust extension after extraction
+            if not os.path.exists(file_path):
+                return None, "Downloaded file not found", None
+            # Read the audio file for gr.Audio
+            sample_rate, data = scipy.io.wavfile.read(file_path)
+            audio_data = (sample_rate, data)
+            return file_path, "Download successful", audio_data
         except yt_dlp.utils.ExtractorError as e:
             if "Sign in to confirm you’re not a bot" in str(e):
-                return None, "Authentication failed. Please upload updated cookies from a logged-in browser session.", None, None, None, None
-            return None, f"Download failed: {str(e)}", None, None, None, None
+                return None, "Authentication failed. Please upload updated cookies from a logged-in browser session in the respective tab.", None
+            return None, f"Download failed: {str(e)}", None
         except Exception as e:
-            return None, f"Unexpected error: {str(e)}", None, None, None, None
+            return None, f"Unexpected error: {str(e)}", None
 
 @spaces.GPU
 def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, pitch_shift, model_dir, output_dir, out_format, norm_thresh, amp_thresh, batch_size, exclude_stems="", progress=gr.Progress(track_tqdm=True)):
@@ -486,7 +492,6 @@ def update_ensemble_models(category):
     return gr.update(choices=choices, value=[])
 
 def create_interface():
-    """Create the Gradio interface for music source separation."""
     with gr.Blocks(title="🎵 SESA Fast Separation 🎵", css=CSS, elem_id="app-container") as app:
         gr.Markdown("<h1 class='header-text'>🎵 SESA Fast Separation 🎵</h1>")
         gr.Markdown("**Note**: If YouTube downloads fail, try uploading an audio file directly or use a valid cookies file.")
@@ -554,9 +559,13 @@ def create_interface():
 
         # Event Handlers
         roformer_category.change(update_roformer_models, inputs=[roformer_category], outputs=[roformer_model])
-        download_roformer.click(fn=download_audio, inputs=[url_ro, cookies_ro], outputs=[roformer_audio])
+        download_roformer.click(
+            fn=download_audio,
+            inputs=[url_ro, cookies_ro],
+            outputs=[roformer_audio]  # Now expects (sample_rate, data)
+        )
         roformer_button.click(
-            roformer_separator,
+            fn=roformer_separator,
             inputs=[
                 roformer_audio, roformer_model, roformer_seg_size, roformer_override_seg_size,
                 roformer_overlap, roformer_pitch_shift, model_file_dir, output_dir,
@@ -565,7 +574,11 @@ def create_interface():
             outputs=[roformer_stem1, roformer_stem2]
         )
         ensemble_category.change(update_ensemble_models, inputs=[ensemble_category], outputs=[ensemble_models])
-        download_ensemble.click(fn=download_audio, inputs=[url_ensemble, cookies_ensemble], outputs=[ensemble_audio])
+        download_ensemble.click(
+            fn=download_audio,
+            inputs=[url_ensemble, cookies_ensemble],
+            outputs=[ensemble_audio]  # Now expects (sample_rate, data)
+        )
         ensemble_button.click(
             fn=auto_ensemble_process,
             inputs=[