Update app.py

app.py

@@ -5,8 +5,14 @@ import tempfile
 import shutil
 import threading
 
-def delayed_delete(path, delay=1800): # 1800 seconds = 30 minutes
-    """Function to delete a path after a set delay."""
+# 1. INITIALIZE SEPARATOR GLOBALLY (Load once, use many times)
+# This stays in RAM so requests start instantly.
+BASE_TEMP_DIR = tempfile.mkdtemp()
+global_separator = Separator(output_dir=BASE_TEMP_DIR, output_format="MP3", output_bitrate="128k")
+print("Loading model into memory...")
+global_separator.load_model(model_filename="htdemucs_6s.yaml") 
+
+def delayed_delete(path, delay=1800):
     def command():
         import time
         time.sleep(delay)
@@ -15,86 +21,65 @@ def delayed_delete(path, delay=1800): # 1800 seconds = 30 minutes
                 shutil.rmtree(path)
             elif os.path.isfile(path):
                 os.remove(path)
-            print(f"Cleanup: Deleted {path}")
         except Exception as e:
-            print(f"Error cleaning up {path}: {e}")
+            print(f"Cleanup error: {e}")
 
-    # Start a background thread so it doesn't block your app
     threading.Thread(target=command, daemon=True).start()
-    
+
 def separate_audio(input_file, model_name):
     if input_file is None:
         return [None] * 6
     
-    # Create a temporary directory for this specific session
-    temp_dir = tempfile.mkdtemp()
+    # Create a sub-folder for this specific request
+    request_id = next(tempfile._get_candidate_names())
+    session_dir = os.path.join(BASE_TEMP_DIR, request_id)
+    os.makedirs(session_dir, exist_ok=True)
     
     try:
-        # Initialize separator with the temp directory
-        separator = Separator(output_dir=temp_dir, output_format="MP3", output_bitrate="128k")
-        separator.load_model(model_filename=model_name)
-        output_files = separator.separate(input_file)
+        # Update the output directory for this specific run
+        global_separator.output_dir = session_dir
+        
+        # Perform separation (Model is already loaded!)
+        output_files = global_separator.separate(input_file)
         
         stems = ["Vocals", "Drums", "Bass", "Other", "Guitar", "Piano"]
         results = []
         
         for stem in stems:
-            match = next((os.path.join(temp_dir, f) for f in output_files if stem in f), None)
+            match = next((os.path.join(session_dir, f) for f in output_files if stem in f), None)
             results.append(match)
 
-        # Schedule cleanup once after processing is successful
-        delayed_delete(temp_dir, delay=1800)
+        delayed_delete(session_dir, delay=600) # Cleanup after 10 mins
         return tuple(results)
 
     except Exception as e:
-        # Cleanup immediately if something crashes
-        if os.path.exists(temp_dir):
-            shutil.rmtree(temp_dir)
-        print(f"Error during separation: {e}")
-        raise e
+        if os.path.exists(session_dir):
+            shutil.rmtree(session_dir)
+        raise gr.Error(f"Separation failed: {e}")
 
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
-    gr.Markdown("""
-    # 🎵 AI 6-Stem Audio Separator
-    Split your music into high-quality individual tracks using the **Demucs 6s** model.
-    """)
+    gr.Markdown("# 🎵 AI 6-Stem Audio Separator")
     
     with gr.Row():
-        with gr.Column(scale=1):
-            input_audio = gr.Audio(
-                label="Upload Audio File", 
-                type="filepath",
-                interactive=True
-            )
-            model_dropdown = gr.Dropdown(
-                choices=["htdemucs_6s.yaml"], 
-                value="htdemucs_6s.yaml",
-                label="Select Model"
-            )
-            separate_btn = gr.Button("🚀 Separate Stems", variant="primary")
+        with gr.Column():
+            input_audio = gr.Audio(label="Upload Audio File", type="filepath")
+            model_dropdown = gr.Dropdown(choices=["htdemucs_6s.yaml"], value="htdemucs_6s.yaml", label="Model")
+            separate_btn = gr.Button("🚀 Separate", variant="primary")
             
-        with gr.Column(scale=2):
-            with gr.Group():
-                gr.Markdown("### 🔊 Separation Results")
-                with gr.Row():
-                    output_vocals = gr.Audio(label="🎤 Vocals", type="filepath")
-                    output_drums = gr.Audio(label="🥁 Drums", type="filepath")
-                with gr.Row():
-                    output_bass = gr.Audio(label="🎸 Bass", type="filepath")
-                    output_guitar = gr.Audio(label="🎸 Guitar", type="filepath")
-                with gr.Row():
-                    output_piano = gr.Audio(label="🎹 Piano", type="filepath")
-                    output_other = gr.Audio(label="🎼 Other", type="filepath")
+        with gr.Column():
+            output_vocals = gr.Audio(label="🎤 Vocals")
+            output_drums = gr.Audio(label="🥁 Drums")
+            output_bass = gr.Audio(label="🎸 Bass")
+            output_guitar = gr.Audio(label="🎸 Guitar")
+            output_piano = gr.Audio(label="�� Piano")
+            output_other = gr.Audio(label="🎼 Other")
 
-    # Action binding
     separate_btn.click(
         fn=separate_audio,
         inputs=[input_audio, model_dropdown],
-        outputs=[
-            output_vocals, output_drums, output_bass, 
-            output_other, output_guitar, output_piano
-        ]
+        outputs=[output_vocals, output_drums, output_bass, output_other, output_guitar, output_piano],
+        api_name="separate_audio" # This is important for your JS client
     )
 
     gr.Markdown("---")