Update app.py

app.py

@@ -528,6 +528,10 @@ def update_ensemble_models(category):
     logger.debug(f"Updating ensemble models for category {category}: {choices}")
     return gr.update(choices=choices, value=[])
 
+def download_audio_wrapper(url, cookie_file):
+    file_path, status, audio_data = download_audio(url, cookie_file)
+    return audio_data, status  # Return audio_data for gr.Audio, status for gr.Textbox    
+
 def create_interface():
     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>")
@@ -554,6 +558,7 @@ def create_interface():
                         url_ro = gr.Textbox(label="🔗 Or Paste URL", placeholder="YouTube or audio URL", interactive=True)
                         cookies_ro = gr.File(label="🍪 Cookies File", file_types=[".txt"], interactive=True)
                         download_roformer = gr.Button("⬇️ Download", variant="secondary")
+                    roformer_download_status = gr.Textbox(label="📢 Download Status", interactive=False)  # Added status output
                     roformer_exclude_stems = gr.Textbox(label="🚫 Exclude Stems", placeholder="e.g., vocals, drums (comma-separated)", interactive=True)
                     with gr.Row():
                         roformer_category = gr.Dropdown(label="📚 Category", choices=list(ROFORMER_MODELS.keys()), value="General Purpose", interactive=True)
@@ -578,6 +583,7 @@ def create_interface():
                         url_ensemble = gr.Textbox(label="🔗 Or Paste URL", placeholder="YouTube or audio URL", interactive=True)
                         cookies_ensemble = gr.File(label="🍪 Cookies File", file_types=[".txt"], interactive=True)
                         download_ensemble = gr.Button("⬇️ Download", variant="secondary")
+                    ensemble_download_status = gr.Textbox(label="📢 Download Status", interactive=False)  # Added status output
                     ensemble_exclude_stems = gr.Textbox(label="🚫 Exclude Stems", placeholder="e.g., vocals, drums (comma-separated)", interactive=True)
                     with gr.Row():
                         ensemble_category = gr.Dropdown(label="📚 Category", choices=list(ROFORMER_MODELS.keys()), value="Instrumentals", interactive=True)
@@ -597,9 +603,9 @@ def create_interface():
         # Event Handlers
         roformer_category.change(update_roformer_models, inputs=[roformer_category], outputs=[roformer_model])
         download_roformer.click(
-            fn=download_audio,
+            fn=download_audio_wrapper,
             inputs=[url_ro, cookies_ro],
-            outputs=[roformer_audio]  # Now expects (sample_rate, data)
+            outputs=[roformer_audio, roformer_download_status]  # Updated to use wrapper and output status
         )
         roformer_button.click(
             fn=roformer_separator,
@@ -612,9 +618,9 @@ def create_interface():
         )
         ensemble_category.change(update_ensemble_models, inputs=[ensemble_category], outputs=[ensemble_models])
         download_ensemble.click(
-            fn=download_audio,
+            fn=download_audio_wrapper,
             inputs=[url_ensemble, cookies_ensemble],
-            outputs=[ensemble_audio]  # Now expects (sample_rate, data)
+            outputs=[ensemble_audio, ensemble_download_status]  # Updated to use wrapper and output status
         )
         ensemble_button.click(
             fn=auto_ensemble_process,