Update app.py

app.py

@@ -13,22 +13,7 @@ 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
@@ -195,7 +180,25 @@ def process_audio(audio_file, selected_effects, isolate_vocals, preset_name, exp
         final_audio.export(output_path, format=export_format.lower())
         return output_path, show_waveform(output_path)
 
-# === Gradio Interface ===
+# === Batch Processing Function ===
+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
+
+# === Effect List ===
 effect_options = [
     "Noise Reduction",
     "Compress Dynamic Range",
@@ -208,19 +211,47 @@ effect_options = [
     "Normalize"
 ]
 
-interface = gr.Interface(
-    fn=batch_process_audio,
-    inputs=[
-        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"
-)
-
-interface.launch()
+# === Multi-Tab UI ===
+with gr.Blocks(title="AI Audio Studio") as demo:
+    gr.Markdown("## 🎧 AI Audio Studio\nUpload, edit, export — all powered by AI")
+
+    # ----- Batch Processing Tab -----
+    with gr.Tab("🔊 Batch Processing"):
+        gr.Interface(
+            fn=batch_process_audio,
+            inputs=[
+                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="Batch Audio Processor",
+            description="Upload multiple files, apply effects in bulk, and download all results in a single ZIP.",
+            allow_flagging="never",
+            submit_btn="Process All Files",
+            clear_btn=False
+        )
+
+    # ----- Single File Studio Tab -----
+    with gr.Tab("🎵 Single File Studio"):
+        gr.Interface(
+            fn=process_audio,
+            inputs=[
+                gr.Audio(label="Upload Audio", type="filepath"),
+                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.Audio(label="Processed Audio", type="filepath"),
+                gr.Image(label="Waveform Preview")
+            ],
+            title="Edit One File at a Time",
+            description="Apply effects, preview waveform, and export as MP3 or WAV",
+            allow_flagging="never"
+        )
+
+demo.launch()