Update app.py

app.py

@@ -18,9 +18,8 @@ import librosa
 import warnings
 from faster_whisper import WhisperModel
 from TTS.api import TTS
-import base64
 import pickle
-import json
+import base64
 
 # Suppress warnings
 warnings.filterwarnings("ignore")
@@ -303,23 +302,22 @@ preset_choices = {
 preset_names = list(preset_choices.keys())
 
 # === Preset Cards Gallery ===
-def get_preset_cards():
-    card_paths = []
-    for name in preset_names:
-        card_paths.append(f"https://via.placeholder.com/150x100?text={name}")
-    return card_paths
-
-# === Load Preset by Name === 
-def load_preset_by_card(evt: gr.SelectData):
-    index = evt.index % len(preset_names)
-    name = preset_names[index]
-    return name, preset_choices[name]
+preset_cards = [
+    ("images/pop_card.png", "Pop"),
+    ("images/edm_card.png", "EDM"),
+    ("images/rock_card.png", "Rock"),
+    ("images/hiphop_card.png", "Hip-Hop"),
+    ("images/acoustic_card.png", "Acoustic"),
+    ("images/stage_mode_card.png", "Stage Mode"),
+    ("images/vocal_distortion_card.png", "Vocal Distortion"),
+    ("images/tube_saturation_card.png", "Tube Saturation")
+]
 
 # === Logo Embedding (Base64 or file) ===
 def get_logo():
-    return "https://via.placeholder.com/400x100?text=AI+Audio+Studio"
+    return "logo.png"
 
-# === Main UI === 
+# === Main UI ===
 with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
     gr.HTML(f'<div class="studio-header"><img src="{get_logo()}" width="400" /></div>')
     gr.Markdown("### Upload, edit, export — powered by AI!")
@@ -378,25 +376,18 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
     # --- Preset Cards Gallery ===
     with gr.Tab("🎛 Preset Gallery"):
         gr.Markdown("### Select a preset visually")
-
-        preset_images = [
-            ("https://via.placeholder.com/150x100?text=Pop", "Pop"),
-            ("https://via.placeholder.com/150x100?text=EDM", "EDM"),
-            ("https://via.placeholder.com/150x100?text=Rock", "Rock"),
-            ("https://via.placeholder.com/150x100?text=Hip-Hop", "Hip-Hop"),
-            ("https://via.placeholder.com/150x100?text=Acoustic", "Acoustic"),
-            ("https://via.placeholder.com/150x100?text=Tube", "Tube"),
-            ("https://via.placeholder.com/150x100?text=Stage+Mode", "Stage Mode"),
-            ("https://via.placeholder.com/150x100?text=Vocal+Distortion", "Vocal Distortion")
-        ]
-
-        preset_gallery = gr.Gallery(value=preset_images, label="Preset Cards", columns=4, height="auto")
+        preset_gallery = gr.Gallery(value=preset_cards, label="Preset Cards", columns=4, height="auto")
         preset_name_out = gr.Dropdown(choices=preset_names, label="Selected Preset")
         preset_effects_out = gr.CheckboxGroup(choices=list(preset_choices.keys())[0:], label="Effects")
 
+        def load_preset_by_card(evt: gr.SelectData):
+            index = evt.index % len(preset_names)
+            name = preset_names[index]
+            return name, preset_choices[name]
+
         preset_gallery.select(fn=load_preset_by_card, inputs=[], outputs=[preset_name_out, preset_effects_out])
 
-    # --- Vocal Doubler / Harmonizer === 
+    # --- Vocal Doubler / Harmonizer ===
     with gr.Tab("🎧 Vocal Doubler / Harmonizer"):
         gr.Interface(
             fn=lambda x: apply_harmony(x),
@@ -429,7 +420,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             fn=batch_process_audio,
             inputs=[
                 gr.File(label="Upload Multiple Files", file_count="multiple"),
-                gr.CheckboxGroup(choices=list(preset_choices["Default"]), label="Apply Effects in Order"),
+                gr.CheckboxGroup(choices=preset_choices["Default"], label="Apply Effects in Order"),
                 gr.Checkbox(label="Isolate Vocals After Effects"),
                 gr.Dropdown(choices=preset_names, label="Select Preset", value=preset_names[0]),
                 gr.Dropdown(choices=["MP3", "WAV"], label="Export Format", value="MP3")
@@ -498,41 +489,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             description="Ensure consistent volume using EBU R128 standard"
         )
 
-    # --- Stereo Imaging Tool ===
-    def stereo_imaging(audio, mid_side_balance=0.5, stereo_spread=1.0):
-        samples, sr = audiosegment_to_array(AudioSegment.from_file(audio))
-        return array_to_audiosegment(samples, sr)
-
-    with gr.Tab("🎚 Stereo Imaging"):
-        gr.Interface(
-            fn=stereo_imaging,
-            inputs=[
-                gr.Audio(label="Upload Track", type="filepath"),
-                gr.Slider(minimum=0.0, maximum=1.0, value=0.5, label="Mid-Side Balance"),
-                gr.Slider(minimum=0.0, maximum=2.0, value=1.0, label="Stereo Spread")
-            ],
-            outputs=gr.Audio(label="Imaged Output", type="filepath"),
-            title="Adjust Stereo Field",
-            description="Control mid-side balance and widen stereo spread."
-        )
-
     # --- Save/Load Mix Session (.aiproj) ===
-    def save_project(audio, preset, effects):
-        project_data = {
-            "audio": AudioSegment.from_file(audio).raw_data,
-            "preset": preset,
-            "effects": effects
-        }
-        out_path = os.path.join(tempfile.gettempdir(), "project.aiproj")
-        with open(out_path, "wb") as f:
-            pickle.dump(project_data, f)
-        return out_path
-
-    def load_project(project_file):
-        with open(project_file.name, "rb") as f:
-            data = pickle.load(f)
-        return data["preset"], data["effects"]
-
     with gr.Tab("📁 Save/Load Project"):
         gr.Interface(
             fn=save_project,