Update app.py

app.py

@@ -7,7 +7,7 @@ import zipfile
 import tempfile
 import soundfile as sf
 import traceback
-import subprocess # Necessary for running Spleeter
+import subprocess
 from typing import Tuple, List
 
 # --- Configuration ---
@@ -296,4 +296,100 @@ def create_market_ready_pack(
         with open(license_filepath, 'w') as f:
             f.write(license_content.strip())
             
-        # Create the final ZIP file
+        # Create the final ZIP file
+        zip_filename = os.path.join(temp_dir, f"{OUTPUT_FOLDER_NAME}_{key_mode_name}.zip")
+        
+        with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zf:
+            for root, dirs, files in os.walk(output_root):
+                for file in files:
+                    full_path = os.path.join(root, file)
+                    relative_path = os.path.relpath(full_path, temp_dir)
+                    zf.write(full_path, relative_path)
+                    
+        progress(1.0, desc="Packaging Complete!")
+        
+        shutil.rmtree(output_root)
+
+        return zip_filename, f"✅ Success! Your market-ready '{os.path.basename(zip_filename)}' is ready. Key/BPM: {key_mode_name}. Stems Processed: {', '.join(display_stems)}. Download below."
+
+    except Exception as e:
+        error_message = f"Critical Error: {e}"
+        print(f"Full Traceback: {traceback.format_exc()}")
+        
+        if temp_dir and os.path.exists(temp_dir):
+            shutil.rmtree(temp_dir)
+            gr.Warning("Cleaned up temporary files after failure.")
+            
+        return None, f"❌ Processing failed. {error_message}. If Spleeter failed, ensure it is installed correctly."
+
+
+# --- Gradio Interface Definition ---
+
+with gr.Blocks(title="Market-Ready Loop Pack Generator") as demo:
+    gr.Markdown(
+        """
+        # 🎧 Professional Loop Pack Automation Tool
+        
+        Upload a full music track, select your stem separation model, and generate a 
+        complete, royalty-free sample pack including time-aligned loops and transient-detected one-shots.
+        """
+    )
+
+    with gr.Row():
+        audio_input = gr.Audio(
+            type="filepath", 
+            sources=["upload"], 
+            label="1. Upload Full Mix Audio File (WAV/MP3/FLAC)",
+        )
+        
+        stem_model_input = gr.Dropdown(
+            label="2. Select Stem Separation Model",
+            choices=list(STEM_MODELS.keys()),
+            value='4-Stems (Drums, Bass, Vocals, Other)',
+            allow_custom_value=False,
+            info="Choose the number and type of stems to split the audio into (requires Spleeter installation)."
+        )
+
+    with gr.Row():
+        sensitivity_slider = gr.Slider(
+            minimum=1, 
+            maximum=10, 
+            step=1, 
+            value=6, 
+            label="3. One-Shot Sensitivity (1=Few/Loud, 10=Many/Quiet)",
+            info="Controls the transient detection threshold for one-shot slicing."
+        )
+
+        generate_button = gr.Button("🚀 Generate Loop Pack", variant="primary")
+
+    with gr.Column(scale=1):
+        status_output = gr.Textbox(label="Status / Feedback", interactive=False)
+        zip_output = gr.File(label="4. Download Final Loop Pack ZIP")
+
+    # Define the core process action
+    generate_button.click(
+        fn=create_market_ready_pack,
+        inputs=[audio_input, sensitivity_slider, stem_model_input],
+        outputs=[zip_output, status_output]
+    )
+    
+    gr.Markdown(
+        """
+        ---
+        **Final Pack Structure (Example):**
+        - `PRO_LOOP_PACK_128BPM_CMinor.zip`
+          - `License.txt`
+          - `LOOPS/`
+            - `Drums/` (e.g., `128BPM_CMinor_Drums_4Bar_01.wav`)
+            - `Bass/` 
+            - `Vocals/` 
+            - ... (based on model selected)
+          - `ONESHOTS/`
+            - `Drums/` (e.g., `128BPM_CMinor_Drums_OneShot_001.wav`)
+            - `Bass/`
+            - ...
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch(enable_queue=True)