Update app.py

app.py

@@ -412,11 +412,18 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
         progress(0.7, desc="Separating audio...")
         separation = separator.separate(audio)
         stems = [os.path.join(output_dir, file_name) for file_name in separation]
+        file_list = []
         if exclude_stems.strip():
             excluded = [s.strip().lower() for s in exclude_stems.split(',')]
             filtered_stems = [stem for stem in stems if not any(ex in os.path.basename(stem).lower() for ex in excluded)]
-            return filtered_stems[0] if filtered_stems else None, filtered_stems[1] if len(filtered_stems) > 1 else None
-        return stems[0], stems[1] if len(stems) > 1 else None
+            file_list = filtered_stems
+            stem1 = filtered_stems[0] if filtered_stems else None
+            stem2 = filtered_stems[1] if len(filtered_stems) > 1 else None
+        else:
+            file_list = stems
+            stem1 = stems[0]
+            stem2 = stems[1] if len(stems) > 1 else None
+        return stem1, stem2, file_list
     except Exception as e:
         logger.error(f"Separation error: {e}")
         raise RuntimeError(f"Separation error: {e}")
@@ -606,7 +613,15 @@ def auto_ensemble_process(audio, model_keys, seg_size=64, overlap=0.1, out_forma
             progress(1.0, desc="Ensemble completed")
             elapsed = time.time() - start_time
             logger.info(f"Total processing time: {elapsed:.2f}s")
-            return output_file, f"Ensemble completed with {ensemble_method}, excluded: {exclude_stems if exclude_stems else 'None'}, {len(model_keys)} models in {elapsed:.2f}s"
+            # Prepare file list for download
+            file_list = [output_file] + all_stems
+            # Create status message with download links
+            status = f"Ensemble completed with {ensemble_method}, excluded: {exclude_stems if exclude_stems else 'None'}, {len(model_keys)} models in {elapsed:.2f}s<br>Download files:<ul>"
+            for file in file_list:
+                file_name = os.path.basename(file)
+                status += f"<li><a href='file={file}' download>{file_name}</a></li>"
+            status += "</ul>"
+            return output_file, status, file_list
         except Exception as e:
             logger.error(f"Ensemble processing error: {e}")
             if "numpy" in str(e).lower() or "copy" in str(e).lower():
@@ -683,6 +698,7 @@ def create_interface():
                     with gr.Row():
                         roformer_stem1 = gr.Audio(label="🎸 Stem 1", type="filepath", interactive=False)
                         roformer_stem2 = gr.Audio(label="🥁 Stem 2", type="filepath", interactive=False)
+                    roformer_files = gr.File(label="📥 Download Stems", interactive=False)
             with gr.Tab("🎚️ Auto Ensemble"):
                 with gr.Group(elem_classes="dubbing-theme"):
                     gr.Markdown("### Ensemble Processing")
@@ -705,7 +721,8 @@ def create_interface():
                     ensemble_weights = gr.Textbox(label="⚖️ Weights", placeholder="e.g., 1.0, 1.0, 1.0 (comma-separated)", interactive=True)
                     ensemble_button = gr.Button("🎛️ Run Ensemble!", variant="primary")
                     ensemble_output = gr.Audio(label="🎶 Ensemble Result", type="filepath", interactive=False)
-                    ensemble_status = gr.Textbox(label="📢 Status", interactive=False)
+                    ensemble_status = gr.Textbox(label="📢 Status", interactive=False, render_markdown=True)
+                    ensemble_files = gr.File(label="📥 Download Ensemble and Stems", interactive=False)
         gr.HTML("<div class='footer'>Powered by Audio-Separator 🌟🎶 | Made with ❤️</div>")
         roformer_category.change(update_roformer_models, inputs=[roformer_category], outputs=[roformer_model])
         download_roformer.click(
@@ -720,7 +737,7 @@ def create_interface():
                 roformer_overlap, roformer_pitch_shift, model_file_dir, output_dir,
                 output_format, norm_threshold, amp_threshold, batch_size, roformer_exclude_stems
             ],
-            outputs=[roformer_stem1, roformer_stem2]
+            outputs=[roformer_stem1, roformer_stem2, roformer_files]
         )
         ensemble_category.change(update_ensemble_models, inputs=[ensemble_category], outputs=[ensemble_models])
         download_ensemble.click(
@@ -732,11 +749,11 @@ def create_interface():
             fn=auto_ensemble_process,
             inputs=[
                 ensemble_audio, ensemble_models, ensemble_seg_size, ensemble_overlap,
-                output_format, ensemble_use_tta, model_file_dir, output_dir,
+                output_format, ensemble_use_tta, model_dir, output_dir,
                 norm_threshold, amp_threshold, batch_size, ensemble_method,
                 ensemble_exclude_stems, ensemble_weights
             ],
-            outputs=[ensemble_output, ensemble_status]
+            outputs=[ensemble_output, ensemble_status, ensemble_files]
         )
     return app