Update app.py

app.py

@@ -249,45 +249,48 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             gr.Markdown("### Sliced Loops / Samples (Preview)")
             loop_gallery = gr.Gallery(label="Generated Loops Preview", columns=8, object_fit="contain", height="auto", preview=True)
 
-    # --- Define Event Listeners ---
-    def slice_and_display(stem_data, loop_choice, sensitivity, stem_name):
-        # Wrapper to handle progress display for single slice buttons
-        log_history = f"Slicing {stem_name}...\n"
-        yield { status_log: log_history, progress_bar: gr.update(visible=True) }
-        
-        def update_single_progress(p, desc=""):
-             progress_bar.update(value=p, label=desc, visible=True) # Cannot yield from nested func
-
-        files, temp_dir = slice_stem_real(stem_data, loop_choice, sensitivity, stem_name, progress_fn=update_single_progress)
-        
-        if temp_dir and os.path.exists(temp_dir):
-             shutil.rmtree(temp_dir) # Clean up temp dir after preview
-        
-        yield { 
-            loop_gallery: gr.update(value=files), 
-            status_log: log_history + f"✅ Sliced {stem_name} into {len(files) if files else 0} pieces.",
-            progress_bar: gr.update(visible=False)
-        }
+  # --- Define Event Listeners ---
+def slice_and_display(stem_data, loop_choice, sensitivity):
+    # This wrapper remains the same
+    log_history = f"Slicing...\n"
+    yield { status_log: log_history, progress_bar: gr.update(visible=True) } # Show progress bar
 
+    # Define how slice_stem_real updates the progress bar
+    def update_single_progress(p, desc=""):
+         progress_bar.update(value=p, label=desc, visible=True) 
 
-    submit_event = submit_button.click(
-        fn=separate_stems, 
-        inputs=[audio_input, stem_options], 
-        outputs=[vocals_output, drums_output, bass_output, other_output, status_log, progress_bar]
-    )
+    files, temp_dir = slice_stem_real(stem_data, loop_choice, sensitivity, "stem", progress_fn=update_single_progress) # Pass stem_name placeholder
     
-    stem_options.change(fn=update_output_visibility, inputs=stem_options, outputs=[vocals_output, drums_output, bass_output, other_output])
+    if temp_dir and os.path.exists(temp_dir):
+         shutil.rmtree(temp_dir) 
     
-    slice_vocals_btn.click(fn=slice_and_display, inputs=[vocals_output, loop_options_radio, sensitivity_slider, gr.Textbox("vocals", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
-    slice_drums_btn.click(fn=slice_and_display, inputs=[drums_output, loop_options_radio, sensitivity_slider, gr.Textbox("drums", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
-    slice_bass_btn.click(fn=slice_and_display, inputs=[bass_output, loop_options_radio, sensitivity_slider, gr.Textbox("bass", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
-    slice_other_btn.click(fn=slice_and_display, inputs=[other_output, loop_options_radio, sensitivity_slider, gr.Textbox("other", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
-    
-    slice_all_event = slice_all_button.click(
-        fn=slice_all_and_zip_real, 
-        inputs=[vocals_output, drums_output, bass_output, other_output, loop_options_radio, sensitivity_slider], 
-        outputs=[download_zip_file, download_zip_file, status_log, progress_bar] # Added status log and progress bar outputs
-    )
+    yield { 
+        loop_gallery: gr.update(value=files), 
+        status_log: log_history + f"✅ Sliced into {len(files) if files else 0} pieces.",
+        progress_bar: gr.update(visible=False) # Hide progress bar when done
+    }
+
+
+submit_event = submit_button.click(
+    fn=separate_stems, 
+    inputs=[audio_input, stem_options], 
+    # Corrected outputs: progress_bar removed
+    outputs=[vocals_output, drums_output, bass_output, other_output, status_log] 
+)
+
+stem_options.change(fn=update_output_visibility, inputs=stem_options, outputs=[vocals_output, drums_output, bass_output, other_output])
+
+slice_vocals_btn.click(fn=slice_and_display, inputs=[vocals_output, loop_options_radio, sensitivity_slider, gr.Textbox("vocals", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
+slice_drums_btn.click(fn=slice_and_display, inputs=[drums_output, loop_options_radio, sensitivity_slider, gr.Textbox("drums", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
+slice_bass_btn.click(fn=slice_and_display, inputs=[bass_output, loop_options_radio, sensitivity_slider, gr.Textbox("bass", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
+slice_other_btn.click(fn=slice_and_display, inputs=[other_output, loop_options_radio, sensitivity_slider, gr.Textbox("other", visible=False)], outputs=[loop_gallery, status_log, progress_bar])
+
+slice_all_event = slice_all_button.click(
+    fn=slice_all_and_zip_real, 
+    inputs=[vocals_output, drums_output, bass_output, other_output, loop_options_radio, sensitivity_slider], 
+    # Corrected outputs: progress_bar removed
+    outputs=[download_zip_file, download_zip_file, status_log] 
+)
 
 # --- Launch the UI ---
 demo.launch()