Update app.py

app.py

@@ -137,25 +137,25 @@ def match_loudness(audio_path, target_lufs=-14.0):
 # === Auto-EQ per Genre – With New Genres Added ===
 def auto_eq(audio, genre="Pop"):
     eq_map = {
-        "Pop": [(200, 500, -3), (2000, 4000, +4)],  # Cut muddiness, boost vocals
-        "EDM": [(60, 250, +6), (8000, 12000, +3)],  # Maximize bass & sparkle
-        "Rock": [(1000, 3000, +4), (7000, 10000, -3)],  # Punchy mids, reduce sibilance
-        "Hip-Hop": [(20, 100, +6), (7000, 10000, -4)],  # Deep lows, smooth highs
-        "Acoustic": [(100, 300, -3), (4000, 8000, +2)],  # Natural tone
-        "Metal": [(100, 500, -4), (2000, 5000, +6), (7000, 12000, -3)],  # Clear low-mids, crisp highs
-        "Trap": [(80, 120, +6), (3000, 6000, -4)],  # Sub-bass boost, cut harsh highs
-        "LoFi": [(20, 200, +3), (1000, 3000, -2)],  # Warmth, soft mids
-        "Jazz": [(100, 400, +2), (1500, 3000, +1)],  # Smooth midrange
-        "Classical": [(200, 1000, +1), (3000, 6000, +2)],  # Balanced orchestral EQ
-        "Chillhop": [(50, 200, +3), (2000, 5000, +1)],  # Laid-back warmth
-        "Ambient": [(100, 500, +4), (6000, 12000, +2)],  # Spacey atmosphere
-        "Jazz Piano": [(100, 1000, +3), (2000, 5000, +2)],  # Rich piano tone
-        "Trap EDM": [(60, 120, +6), (2000, 5000, -3)],  # Heavy sub + clean highs
-        "Indie Rock": [(150, 400, +2), (2000, 5000, +3)],  # Crisp guitars
-        "Lo-Fi Jazz": [(80, 200, +3), (2000, 4000, +1)],  # Cozy jazz warmth
-        "R&B": [(100, 300, +4), (2000, 4000, +3)],  # Full vocals
-        "Soul": [(80, 200, +3), (1500, 3500, +4)],  # Emotive vocal clarity
-        "Funk": [(80, 200, +5), (1000, 3000, +3)],  # Tight low end
+        "Pop": [(200, 500, -3), (2000, 4000, +4)],
+        "EDM": [(60, 250, +6), (8000, 12000, +3)],
+        "Rock": [(1000, 3000, +4), (7000, 10000, -3)],
+        "Hip-Hop": [(20, 100, +6), (7000, 10000, -4)],
+        "Acoustic": [(100, 300, -3), (4000, 8000, +2)],
+        "Metal": [(100, 500, -4), (2000, 5000, +6), (7000, 12000, -3)],
+        "Trap": [(80, 120, +6), (3000, 6000, -4)],
+        "LoFi": [(20, 200, +3), (1000, 3000, -2)],
+        "Jazz": [(100, 400, +2), (1500, 3000, +1)],
+        "Classical": [(200, 1000, +1), (3000, 6000, +2)],
+        "Chillhop": [(50, 200, +3), (2000, 5000, +1)],
+        "Ambient": [(100, 500, +4), (6000, 12000, +2)],
+        "Jazz Piano": [(100, 1000, +3), (2000, 5000, +2)],
+        "Trap EDM": [(60, 120, +6), (2000, 5000, -3)],
+        "Indie Rock": [(150, 400, +2), (2000, 5000, +3)],
+        "Lo-Fi Jazz": [(80, 200, +3), (2000, 4000, -1)],
+        "R&B": [(100, 300, +4), (2000, 4000, +3)],
+        "Soul": [(80, 200, +3), (1500, 3500, +4)],
+        "Funk": [(80, 200, +5), (1000, 3000, +3)],
         "Default": []
     }
 
@@ -274,7 +274,6 @@ def process_audio(audio_file, selected_effects, isolate_vocals, preset_name, exp
             "Normalize": apply_normalize,
             "Noise Gate": lambda x: apply_noise_gate(x, threshold=-50.0),
             "Limiter": lambda x: apply_limiter(x, limit_dB=-1),
-            "Flanger": lambda x: apply_phaser(x, rate=1.2, depth=0.9, mix=0.7),
             "Bitcrusher": lambda x: apply_bitcrush(x, bit_depth=8),
             "Auto Gain": lambda x: apply_auto_gain(x, target_dB=-20),
             "Vocal Distortion": lambda x: apply_vocal_distortion(x),
@@ -362,7 +361,7 @@ preset_choices = {
     "🌫 ASMR Vocal": ["Auto Gain", "Low-Pass Filter (3000Hz)", "Noise Gate"],
     "🎼 Stage Mode": ["Reverb", "Bass Boost", "Limiter"],
     "🎵 Auto-Tune Style": ["Pitch Shift (+1 semitone)", "Normalize", "Treble Boost"],
-    "🎷 Jazz Vocal": ["Bass Boost (-200-400Hz)", "Treble Boost (2000-4000Hz)", "Normalize"],
+    "🎷 Jazz Vocal": ["Bass Boost (-200-400Hz)", "Treble Boost (-3000Hz)", "Normalize"],
     "🎹 Jazz Piano": ["Treble Boost (4000-6000Hz)", "Normalize", "Stereo Widening"],
     "🎻 Classical Strings": ["Bass Boost (100-500Hz)", "Treble Boost (3000-6000Hz)", "Reverb"],
     "☕ Chillhop": ["Noise Gate", "Treble Boost (-3000Hz)", "Reverb"],
@@ -371,7 +370,7 @@ preset_choices = {
     "💃 Soul Vocal": ["Noise Reduction", "Bass Boost (80-200Hz)", "Treble Boost (1500-3500Hz)"],
     "🕺 Funk Groove": ["Bass Boost (80-200Hz)", "Treble Boost (1000-3000Hz)", "Stereo Widening"],
     "🎹 Jazz Piano Solo": ["Treble Boost (2000-5000Hz)", "Normalize", "Stage Mode"],
-    "🎶 Trap EDM": ["Bass Boost (60-120Hz)", "Treble Boost (2000-5000Hz)", "Limiter"],
+    "🎸 Trap EDM": ["Bass Boost (60-120Hz)", "Treble Boost (2000-5000Hz)", "Limiter"],
     "🎸 Indie Rock": ["Bass Boost (150-400Hz)", "Treble Boost (2000-5000Hz)", "Compress Dynamic Range"]
 }
 
@@ -427,9 +426,9 @@ def visualize_spectrum(audio_path):
     buf.seek(0)
     return Image.open(buf)
 
-# === Main UI ===
-with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
-    gr.HTML('<div class="studio-header"><img src="logo.png" width="400" /></div>')
+# === Main UI – With Studio Pulse AI Branding ===
+with gr.Blocks(title="Studio Pulse AI", css="style.css") as demo:
+    gr.HTML('<div class="studio-header"><img src="logo.png" width="400" /><h3>Where Your Audio Meets Intelligence</h3></div>')
     gr.Markdown("### Upload, edit, export — powered by AI!")
 
     with gr.Tab("🎵 Single File Studio"):
@@ -463,7 +462,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
                 gr.Dropdown(choices=["Pop", "EDM", "Rock", "Hip-Hop", "Acoustic", "Metal", "Trap", "LoFi",
                                  "Jazz", "Classical", "Chillhop", "Ambient", "Jazz Piano", "Trap EDM",
                                  "Indie Rock", "Lo-Fi Jazz", "R&B", "Soul", "Funk"],
-                       label="Genre", value="Pop"),
+                      label="Genre", value="Pop"),
                 gr.Slider(minimum=-24, maximum=-6, value=-14, label="Target LUFS")
             ],
             outputs=gr.Audio(label="Mastered Output", type="filepath"),
@@ -489,7 +488,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             clear_btn=None
         )
 
-    # --- Harmonic Saturation / Exciter ===
+    # --- Harmonic Saturation / Exciter – Now Included ===
     with gr.Tab("🧬 Harmonic Saturation"):
         gr.Interface(
             fn=harmonic_saturation,
@@ -586,16 +585,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
     def load_project(project_file):
         with open(project_file.name, "rb") as f:
             data = pickle.load(f)
-        return (
-            array_to_audiosegment(data["audio"], 44100),
-            array_to_audiosegment(data["audio"], 44100),
-            array_to_audiosegment(data["audio"], 44100),
-            array_to_audiosegment(data["audio"], 44100),
-            data["preset"],
-            data["effects"],
-            data["effects"],
-            data["effects"]
-        )
+        return data["preset"], data["effects"]
 
     with gr.Tab("📁 Save/Load Project"):
         gr.Interface(
@@ -615,13 +605,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             fn=load_project,
             inputs=gr.File(label="Upload .aiproj File"),
             outputs=[
-                gr.File(label="Loaded Vocals"),
-                gr.File(label="Loaded Drums"),
-                gr.File(label="Loaded Bass"),
-                gr.File(label="Loaded Other"),
                 gr.Dropdown(choices=preset_names, label="Loaded Preset"),
-                gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects"),
-                gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects"),
                 gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects")
             ],
             title="Resume Last Project",
@@ -640,23 +624,19 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             ("images/stage_mode_card.png", "Stage Mode"),
             ("images/vocal_distortion_card.png", "Vocal Distortion"),
             ("images/tube_saturation_card.png", "Tube Saturation"),
-            ("images/jazz_card.png", "Jazz"),
-            ("images/classical_card.png", "Classical"),
-            ("images/chillhop_card.png", "Chillhop"),
-            ("images/ambient_card.png", "Ambient"),
             ("images/rnb_card.png", "R&B"),
             ("images/soul_card.png", "Soul"),
-            ("images/funk_card.png", "Funk")
+            ("images/funk_card.png", "Funk"),
+            ("images/jazz_card.png", "Jazz")
         ], 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")
+        preset_effects_out = gr.CheckboxGroup(choices=list(preset_choices["Default"]), label="Effects")
 
         def load_preset_by_card(evt: gr.SelectData):
             index = evt.index % len(preset_names)
             name = preset_names[index]
-            effects = preset_choices[name]
-            return name, effects
+            return name, preset_choices[name]
 
         preset_gallery.select(fn=load_preset_by_card, inputs=[], outputs=[preset_name_out, preset_effects_out])