Update app.py

app.py

@@ -13,7 +13,22 @@ from pathlib import Path
 import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image
+import zipfile
 
+def batch_process_audio(files, selected_effects, isolate_vocals, preset_name, export_format):
+    output_dir = tempfile.mkdtemp()
+    results = []
+
+    for file in files:
+        processed_path = process_audio(file.name, selected_effects, isolate_vocals, preset_name, export_format)
+        results.append(processed_path[0])  # Take just the file path
+
+    zip_path = os.path.join(output_dir, "batch_output.zip")
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for i, res in enumerate(results):
+            zipf.write(res, f"processed_{i}.{export_format.lower()}")
+
+    return zip_path
 # === Helper Functions ===
 def audiosegment_to_array(audio):
     return np.array(audio.get_array_of_samples()), audio.frame_rate
@@ -194,17 +209,24 @@ effect_options = [
 ]
 
 interface = gr.Interface(
-    fn=process_audio,
+    fn=batch_process_audio,
     inputs=[
-        gr.Audio(label="Upload Audio", type="filepath"),
+        gr.File(label="Upload Multiple Audio Files", file_count="multiple"),
         gr.CheckboxGroup(choices=effect_options, label="Apply Effects in Order"),
         gr.Checkbox(label="Isolate Vocals After Effects"),
         gr.Dropdown(choices=preset_names, label="Select Preset", value=preset_names[0] if preset_names else None),
         gr.Dropdown(choices=["MP3", "WAV"], label="Export Format", value="MP3")
     ],
+    outputs=gr.File(label="Download ZIP of All Processed Files"),
+    title="AI Audio Studio - Batch Edition",
+    description="Upload multiple files, apply effects in bulk, and download all results in a single ZIP.",
+    allow_flagging="never"
+)
+],
     outputs=[
         gr.Audio(label="Processed Audio", type="filepath"),
         gr.Image(label="Waveform Preview")
+        gr.File(label="Download All Processed Files (ZIP)")
     ],
     title="AI Audio Studio - Pro Edition",
     description="Apply multiple effects, isolate vocals, and export polished tracks — all powered by AI!",