Update app.py

app.py

@@ -2,6 +2,7 @@
 import gradio as gr
 import numpy as np
 import librosa
+import librosa.display  # Added missing import
 import soundfile as sf
 import os
 import tempfile
@@ -11,7 +12,7 @@ import matplotlib
 import matplotlib.pyplot as plt
 from scipy import signal
 from typing import Tuple, List, Any
-import shutil # Import shutil for directory cleanup
+import shutil
 
 # Use a non-interactive backend for Matplotlib
 matplotlib.use('Agg')
@@ -326,8 +327,7 @@ def slice_stem_real(
             y = librosa.util.buf_to_float(y_int, dtype=np.float32)
         else:
             # Handle case where it's a filepath (from separate_stems)
-             y, sample_rate = librosa.load(stem_audio_path, sr=None)
-
+            y, sample_rate = librosa.load(stem_audio_path, sr=None)  # Fixed indentation
 
         if y.ndim == 0:
             return [], ""
@@ -496,12 +496,6 @@ def slice_all_and_zip(
 
         with zipfile.ZipFile(zip_path, 'w') as zf:
             for name, data in valid_stems.items():
-                # Create temporary file for this stem
-                # No need to save to a temp file here, can pass the tuple directly
-                # stem_temp_dir = tempfile.mkdtemp()
-                # stem_path = os.path.join(stem_temp_dir, f"{name}.wav")
-                # sf.write(stem_path, data[1], data[0])
-
                 # Process stem
                 sliced_files, _ = slice_stem_real(
                     data, loop_choice, sensitivity, name,
@@ -515,11 +509,6 @@ def slice_all_and_zip(
                     arcname = os.path.join(file_type, os.path.basename(file_path))
                     zf.write(file_path, arcname)
 
-                # Clean up stem temp files
-                # shutil.rmtree(stem_temp_dir) # No temp dir created here anymore
-
-        # Note: The main temp_dir containing the zip file will be cleaned up by Gradio
-
         return zip_path
 
     except Exception as e:
@@ -664,24 +653,24 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             gr.Markdown("### Separated Stems")
             with gr.Row():
                 with gr.Column():
-                    # vocals_output.render() # Removed redundant render call
+                    vocals_output  # Place component in layout
                     slice_vocals_btn = gr.Button("Slice Vocals")
                 with gr.Column():
-                    # drums_output.render() # Removed redundant render call
+                    drums_output  # Place component in layout
                     slice_drums_btn = gr.Button("Slice Drums")
             with gr.Row():
                 with gr.Column():
-                    # bass_output.render() # Removed redundant render call
+                    bass_output  # Place component in layout
                     slice_bass_btn = gr.Button("Slice Bass")
                 with gr.Column():
-                    # other_output.render() # Removed redundant render call
+                    other_output  # Place component in layout
                     slice_other_btn = gr.Button("Slice Other")
             with gr.Row():
                 with gr.Column():
-                    # guitar_output.render() # Removed redundant render call
+                    guitar_output  # Place component in layout
                     slice_guitar_btn = gr.Button("Slice Guitar")
                 with gr.Column():
-                    # piano_output.render() # Removed redundant render call
+                    piano_output  # Place component in layout
                     slice_piano_btn = gr.Button("Slice Piano")
 
             # Gallery for previews
@@ -692,6 +681,9 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
                 object_fit="contain",
                 height="auto"
             )
+            
+            # Status textboxes for individual slicing
+            status_textbox = gr.Textbox(label="Status", visible=True)
 
     # --- EVENT HANDLERS ---
 
@@ -748,7 +740,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             pan_depth_slider, level_depth_slider, modulation_rate_radio, lufs_target_slider,
             attack_gain_slider, sustain_gain_slider, filter_type_radio, filter_freq_slider, filter_depth_slider
         ],
-        outputs=[loop_gallery, gr.Textbox(label="Status", visible=True)] # Added status textbox here
+        outputs=[loop_gallery, status_textbox]
     )
 
     slice_drums_btn.click(
@@ -759,7 +751,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             pan_depth_slider, level_depth_slider, modulation_rate_radio, lufs_target_slider,
             attack_gain_slider, sustain_gain_slider, filter_type_radio, filter_freq_slider, filter_depth_slider
         ],
-        outputs=[loop_gallery, gr.Textbox(label="Status", visible=True)] # Added status textbox here
+        outputs=[loop_gallery, status_textbox]
     )
 
     slice_bass_btn.click(
@@ -770,7 +762,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             pan_depth_slider, level_depth_slider, modulation_rate_radio, lufs_target_slider,
             attack_gain_slider, sustain_gain_slider, filter_type_radio, filter_freq_slider, filter_depth_slider
         ],
-        outputs=[loop_gallery, gr.Textbox(label="Status", visible=True)] # Added status textbox here
+        outputs=[loop_gallery, status_textbox]
     )
 
     slice_other_btn.click(
@@ -781,7 +773,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             pan_depth_slider, level_depth_slider, modulation_rate_radio, lufs_target_slider,
             attack_gain_slider, sustain_gain_slider, filter_type_radio, filter_freq_slider, filter_depth_slider
         ],
-        outputs=[loop_gallery, gr.Textbox(label="Status", visible=True)] # Added status textbox here
+        outputs=[loop_gallery, status_textbox]
     )
 
     slice_guitar_btn.click(
@@ -792,7 +784,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             pan_depth_slider, level_depth_slider, modulation_rate_radio, lufs_target_slider,
             attack_gain_slider, sustain_gain_slider, filter_type_radio, filter_freq_slider, filter_depth_slider
         ],
-        outputs=[loop_gallery, gr.Textbox(label="Status", visible=True)] # Added status textbox here
+        outputs=[loop_gallery, status_textbox]
     )
 
     slice_piano_btn.click(
@@ -803,10 +795,9 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
             pan_depth_slider, level_depth_slider, modulation_rate_radio, lufs_target_slider,
             attack_gain_slider, sustain_gain_slider, filter_type_radio, filter_freq_slider, filter_depth_slider
         ],
-        outputs=[loop_gallery, gr.Textbox(label="Status", visible=True)] # Added status textbox here
+        outputs=[loop_gallery, status_textbox]
     )
 
-
     # Slice all stems and create ZIP
     slice_all_btn.click(
         fn=slice_all_and_zip,
@@ -827,4 +818,4 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="red"))
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()