Update app.py

app.py

@@ -74,43 +74,39 @@ def check_vram_availability(required_gb=3.5):
     return available_vram >= required_gb
 
 # 4) GENRE PROMPT FUNCTIONS
-def set_red_hot_chili_peppers_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_red_hot_chili_peppers_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("strong rhythmic steps" if bpm > 120 else "groovy rhythmic flow")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", groovy basslines"
     guitar = f", {guitar_style} guitar riffs" if guitar_style != "none" else ", syncopated guitar riffs"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Funk rock{bass}{guitar}{drum}{synth}{vocal}, Red Hot Chili Peppers-inspired vibe with dynamic energy and funky breakdowns, {rhythm} at {bpm} BPM."
+    return f"Instrumental funk rock{bass}{guitar}{drum}{synth}, Red Hot Chili Peppers-inspired vibe with dynamic energy and funky breakdowns, {rhythm} at {bpm} BPM."
 
-def set_nirvana_grunge_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_nirvana_grunge_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("intense rhythmic steps" if bpm > 120 else "grungy rhythmic pulse")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", melodic basslines"
     guitar = f", {guitar_style} guitar riffs" if guitar_style != "none" else ", raw distorted guitar riffs"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Grunge{bass}{guitar}{drum}{synth}{vocal}, Nirvana-inspired angst-filled sound with quiet-loud dynamics, {rhythm} at {bpm} BPM."
+    return f"Instrumental grunge{bass}{guitar}{drum}{synth}, Nirvana-inspired angst-filled sound with quiet-loud dynamics, {rhythm} at {bpm} BPM."
 
-def set_pearl_jam_grunge_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_pearl_jam_grunge_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("soulful rhythmic steps" if bpm > 120 else "driving rhythmic flow")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", deep bass"
     guitar = f", {guitar_style} guitar leads" if guitar_style != "none" else ", soulful guitar leads"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Grunge{bass}{guitar}{drum}{synth}{vocal}, Pearl Jam-inspired emotional intensity with soaring choruses, {rhythm} at {bpm} BPM."
+    return f"Instrumental grunge{bass}{guitar}{drum}{synth}, Pearl Jam-inspired emotional intensity with soaring choruses, {rhythm} at {bpm} BPM."
 
-def set_soundgarden_grunge_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_soundgarden_grunge_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("heavy rhythmic steps" if bpm > 120 else "sludgy rhythmic groove")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ""
     guitar = f", {guitar_style} guitar riffs" if guitar_style != "none" else ", heavy sludgy guitar riffs"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Grunge{bass}{guitar}{drum}{synth}{vocal}, Soundgarden-inspired dark, psychedelic edge with powerful vocals, {rhythm} at {bpm} BPM."
+    return f"Instrumental grunge{bass}{guitar}{drum}{synth}, Soundgarden-inspired dark, psychedelic edge, {rhythm} at {bpm} BPM."
 
-def set_foo_fighters_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_foo_fighters_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     styles = ["anthemic", "gritty", "melodic", "fast-paced", "driving"]
     tempos = ["upbeat", "mid-tempo", "high-energy"]
     moods = ["energetic", "introspective", "rebellious", "uplifting"]
@@ -121,90 +117,80 @@ def set_foo_fighters_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_st
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ""
-    guitar = f", {guitar_style} guitar riffs" if guitar_style != "none" else ", {style} guitar riffs"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Alternative rock{bass}{guitar}{drum}{synth}{vocal}, Foo Fighters-inspired {mood} vibe with powerful choruses, {rhythm} at {bpm} BPM."
+    guitar = f", {guitar_style} guitar riffs" if guitar_style != "none" else f", {style} guitar riffs"
+    return f"Instrumental alternative rock{bass}{guitar}{drum}{synth}, Foo Fighters-inspired {mood} vibe with powerful choruses, {rhythm} at {bpm} BPM."
 
-def set_smashing_pumpkins_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_smashing_pumpkins_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("dynamic rhythmic steps" if bpm > 120 else "dreamy rhythmic flow")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ""
     guitar = f", {guitar_style} guitar textures" if guitar_style != "none" else ", dreamy guitar textures"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Alternative rock{bass}{guitar}{drum}{synth}{vocal}, Smashing Pumpkins-inspired blend of melancholy and aggression, {rhythm} at {bpm} BPM."
+    return f"Instrumental alternative rock{bass}{guitar}{drum}{synth}, Smashing Pumpkins-inspired blend of melancholy and aggression, {rhythm} at {bpm} BPM."
 
-def set_radiohead_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_radiohead_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("complex rhythmic steps" if bpm > 120 else "intricate rhythmic pulse")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ", atmospheric synths"
     bass = f", {bass_style}" if bass_style != "none" else ""
     guitar = f", {guitar_style} guitar layers" if guitar_style != "none" else ", intricate guitar layers"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Experimental rock{bass}{guitar}{drum}{synth}{vocal}, Radiohead-inspired blend of introspective and innovative soundscapes, {rhythm} at {bpm} BPM."
+    return f"Instrumental experimental rock{bass}{guitar}{drum}{synth}, Radiohead-inspired blend of introspective and innovative soundscapes, {rhythm} at {bpm} BPM."
 
-def set_classic_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_classic_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("bluesy rhythmic steps" if bpm > 120 else "steady rhythmic groove")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", groovy bass"
     guitar = f", {guitar_style} electric guitars" if guitar_style != "none" else ", bluesy electric guitars"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Classic rock{bass}{guitar}{drum}{synth}{vocal}, Led Zeppelin-inspired raw energy with dynamic solos, {rhythm} at {bpm} BPM."
+    return f"Instrumental classic rock{bass}{guitar}{drum}{synth}, Led Zeppelin-inspired raw energy with dynamic solos, {rhythm} at {bpm} BPM."
 
-def set_alternative_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_alternative_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("quirky rhythmic steps" if bpm > 120 else "energetic rhythmic flow")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", melodic basslines"
     guitar = f", {guitar_style} guitar riffs" if guitar_style != "none" else ", distorted guitar riffs"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Alternative rock{bass}{guitar}{drum}{synth}{vocal}, Pixies-inspired quirky, energetic vibe, {rhythm} at {bpm} BPM."
+    return f"Instrumental alternative rock{bass}{guitar}{drum}{synth}, Pixies-inspired quirky, energetic vibe, {rhythm} at {bpm} BPM."
 
-def set_post_punk_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_post_punk_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("sharp rhythmic steps" if bpm > 120 else "moody rhythmic pulse")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", driving basslines"
     guitar = f", {guitar_style} guitars" if guitar_style != "none" else ", jangly guitars"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Post-punk{bass}{guitar}{drum}{synth}{vocal}, Joy Division-inspired moody, atmospheric sound, {rhythm} at {bpm} BPM."
+    return f"Instrumental post-punk{bass}{guitar}{drum}{synth}, Joy Division-inspired moody, atmospheric sound, {rhythm} at {bpm} BPM."
 
-def set_indie_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_indie_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("catchy rhythmic steps" if bpm > 120 else "jangly rhythmic flow")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ""
     guitar = f", {guitar_style} guitars" if guitar_style != "none" else ", jangly guitars"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ", heartfelt vocals"
-    return f"Indie rock{bass}{guitar}{drum}{synth}{vocal}, Arctic Monkeys-inspired blend of witty lyrics and catchy riffs, {rhythm} at {bpm} BPM."
+    return f"Instrumental indie rock{bass}{guitar}{drum}{synth}, Arctic Monkeys-inspired blend of catchy riffs, {rhythm} at {bpm} BPM."
 
-def set_funk_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_funk_rock_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("aggressive rhythmic steps" if bpm > 120 else "funky rhythmic groove")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ""
     bass = f", {bass_style}" if bass_style != "none" else ", slap bass"
     guitar = f", {guitar_style} guitar chords" if guitar_style != "none" else ", funky guitar chords"
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Funk rock{bass}{guitar}{drum}{synth}{vocal}, Rage Against the Machine-inspired mix of groove and aggression, {rhythm} at {bpm} BPM."
+    return f"Instrumental funk rock{bass}{guitar}{drum}{synth}, Rage Against the Machine-inspired mix of groove and aggression, {rhythm} at {bpm} BPM."
 
-def set_detroit_techno_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_detroit_techno_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("pulsing rhythmic steps" if bpm > 120 else "deep rhythmic groove")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ", crisp hi-hats"
     synth = f", {synthesizer} accents" if synthesizer != "none" else ", deep pulsing synths"
     bass = f", {bass_style}" if bass_style != "none" else ", driving basslines"
     guitar = f", {guitar_style} guitars" if guitar_style != "none" else ""
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ""
-    return f"Detroit techno{bass}{guitar}{drum}{synth}{vocal}, Juan Atkins-inspired rhythmic groove, {rhythm} at {bpm} BPM."
+    return f"Instrumental Detroit techno{bass}{guitar}{drum}{synth}, Juan Atkins-inspired rhythmic groove, {rhythm} at {bpm} BPM."
 
-def set_deep_house_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence):
+def set_deep_house_prompt(bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style):
     rhythm = f" with {rhythmic_steps}" if rhythmic_steps != "none" else ("soulful rhythmic steps" if bpm > 120 else "laid-back rhythmic flow")
     drum = f", {drum_beat} drums" if drum_beat != "none" else ""
     synth = f", {synthesizer} accents" if synthesizer != "none" else ", warm analog synth chords"
     bass = f", {bass_style}" if bass_style != "none" else ", deep basslines"
     guitar = f", {guitar_style} guitars" if guitar_style != "none" else ""
-    vocal = f", {vocal_presence}" if vocal_presence != "none" else ", soulful vocal chops"
-    return f"Deep house{bass}{guitar}{drum}{synth}{vocal}, Larry Heard-inspired laid-back groove, {rhythm} at {bpm} BPM."
+    return f"Instrumental deep house{bass}{guitar}{drum}{synth}, Larry Heard-inspired laid-back groove, {rhythm} at {bpm} BPM."
 
 # 5) AUDIO PROCESSING FUNCTIONS
 def apply_eq(segment):
@@ -218,13 +204,13 @@ def apply_fade(segment, fade_in_duration=1000, fade_out_duration=1000):
     return segment
 
 # 6) GENERATION & I/O FUNCTIONS
-def generate_music(instrumental_prompt: str, cfg_scale: float, total_duration: int, chunk_duration: int, crossfade_duration: int, bpm: int, drum_beat: str, synthesizer: str, rhythmic_steps: str, bass_style: str, guitar_style: str, vocal_presence: str):
+def generate_music(instrumental_prompt: str, cfg_scale: float, top_k: int, top_p: float, temperature: float, total_duration: int, chunk_duration: int, crossfade_duration: int, bpm: int, drum_beat: str, synthesizer: str, rhythmic_steps: str, bass_style: str, guitar_style: str):
     global musicgen_model
     if not instrumental_prompt.strip():
         return None, "⚠️ Please enter a valid instrumental prompt!"
     try:
         start_time = time.time()
-        total_duration = min(max(total_duration, 5), 30)
+        total_duration = total_duration  # Validated by radio button (30, 60, 90, 120)
         chunk_duration = min(max(chunk_duration, 5), 15)
         num_chunks = max(1, total_duration // chunk_duration)
         chunk_duration = total_duration / num_chunks
@@ -247,9 +233,9 @@ def generate_music(instrumental_prompt: str, cfg_scale: float, total_duration: i
             musicgen_model.set_generation_params(
                 duration=generation_duration,
                 use_sampling=True,
-                top_k=250,
-                top_p=0.9,
-                temperature=1.0,
+                top_k=top_k,
+                top_p=top_p,
+                temperature=temperature,
                 cfg_coef=cfg_scale
             )
 
@@ -312,7 +298,7 @@ def generate_music(instrumental_prompt: str, cfg_scale: float, total_duration: i
         print_resource_usage("After Final Generation")
         print(f"Total Generation Time: {time.time() - start_time:.2f} seconds")
 
-        return mp3_path, "✅ Done! Generated audio."
+        return mp3_path, "✅ Done! Generated instrumental audio."
     except Exception as e:
         return None, f"❌ Generation failed: {e}"
     finally:
@@ -323,7 +309,7 @@ def generate_music(instrumental_prompt: str, cfg_scale: float, total_duration: i
 
 # Function to clear inputs
 def clear_inputs():
-    return "", 3.0, 10, 10, 1000, 120, "none", "none", "none", "none", "none", "none"
+    return "", 3.0, 250, 0.9, 1.0, 30, 10, 1000, 120, "none", "none", "none", "none", "none"
 
 # 7) CUSTOM CSS
 css = """
@@ -376,6 +362,15 @@ p {
     padding: 10px 20px; 
     border-radius: 5px; 
 }
+.gradio-container { 
+    padding: 20px; 
+}
+.group-container { 
+    margin-bottom: 20px; 
+    padding: 15px; 
+    border: 1px solid #00FF9F; 
+    border-radius: 8px; 
+}
 @keyframes glitch-ghost { 
     0% { transform: translate(0, 0); opacity: 1; }
     20% { transform: translate(-5px, 2px); opacity: 0.8; }
@@ -397,142 +392,166 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("""
         <div class="header-container">
             <div id="ghost-logo">👻</div>
-            <h1>GhostAI Music Generator</h1>
+            <h1>GhostAI Music Generator 🎹</h1>
             <p>Summon the Sound of the Unknown</p>
         </div>
     """)
     
     with gr.Column(elem_classes="input-container"):
+        gr.Markdown("### 🎸 Prompt Settings")
         instrumental_prompt = gr.Textbox(
-            label="Instrumental Prompt",
-            placeholder="Click a genre button or type your own prompt",
+            label="Instrumental Prompt ✍️",
+            placeholder="Click a genre button or type your own instrumental prompt",
             lines=4,
             elem_classes="textbox"
         )
         with gr.Row(elem_classes="genre-buttons"):
-            rhcp_btn = gr.Button("Red Hot Chili Peppers", elem_classes="genre-btn")
-            nirvana_btn = gr.Button("Nirvana Grunge", elem_classes="genre-btn")
-            pearl_jam_btn = gr.Button("Pearl Jam Grunge", elem_classes="genre-btn")
-            soundgarden_btn = gr.Button("Soundgarden Grunge", elem_classes="genre-btn")
-            foo_fighters_btn = gr.Button("Foo Fighters", elem_classes="genre-btn")
-            smashing_pumpkins_btn = gr.Button("Smashing Pumpkins", elem_classes="genre-btn")
-            radiohead_btn = gr.Button("Radiohead", elem_classes="genre-btn")
-            classic_rock_btn = gr.Button("Classic Rock", elem_classes="genre-btn")
-            alternative_rock_btn = gr.Button("Alternative Rock", elem_classes="genre-btn")
-            post_punk_btn = gr.Button("Post-Punk", elem_classes="genre-btn")
-            indie_rock_btn = gr.Button("Indie Rock", elem_classes="genre-btn")
-            funk_rock_btn = gr.Button("Funk Rock", elem_classes="genre-btn")
-            detroit_techno_btn = gr.Button("Detroit Techno", elem_classes="genre-btn")
-            deep_house_btn = gr.Button("Deep House", elem_classes="genre-btn")
+            rhcp_btn = gr.Button("Red Hot Chili Peppers 🌶️", elem_classes="genre-btn")
+            nirvana_btn = gr.Button("Nirvana Grunge 🎸", elem_classes="genre-btn")
+            pearl_jam_btn = gr.Button("Pearl Jam Grunge 🦪", elem_classes="genre-btn")
+            soundgarden_btn = gr.Button("Soundgarden Grunge 🌑", elem_classes="genre-btn")
+            foo_fighters_btn = gr.Button("Foo Fighters 🤘", elem_classes="genre-btn")
+            smashing_pumpkins_btn = gr.Button("Smashing Pumpkins 🎃", elem_classes="genre-btn")
+            radiohead_btn = gr.Button("Radiohead 🧠", elem_classes="genre-btn")
+            classic_rock_btn = gr.Button("Classic Rock 🎸", elem_classes="genre-btn")
+            alternative_rock_btn = gr.Button("Alternative Rock 🎵", elem_classes="genre-btn")
+            post_punk_btn = gr.Button("Post-Punk 🖤", elem_classes="genre-btn")
+            indie_rock_btn = gr.Button("Indie Rock 🎤", elem_classes="genre-btn")
+            funk_rock_btn = gr.Button("Funk Rock 🕺", elem_classes="genre-btn")
+            detroit_techno_btn = gr.Button("Detroit Techno 🎛️", elem_classes="genre-btn")
+            deep_house_btn = gr.Button("Deep House 🏠", elem_classes="genre-btn")
     
     with gr.Column(elem_classes="settings-container"):
-        cfg_scale = gr.Slider(
-            label="Guidance Scale (CFG)",
-            minimum=1.0,
-            maximum=10.0,
-            value=3.0,
-            step=0.1,
-            info="Higher values make the instrumental more closely follow the prompt."
-        )
-        total_duration = gr.Slider(
-            label="Total Duration (seconds)",
-            minimum=5,
-            maximum=30,
-            value=10,
-            step=1,
-            info="Total duration of the track (5 to 30 seconds)."
-        )
-        chunk_duration = gr.Slider(
-            label="Chunk Duration (seconds)",
-            minimum=5,
-            maximum=15,
-            value=10,
-            step=1,
-            info="Duration of each chunk to render (5 to 15 seconds)."
-        )
-        crossfade_duration = gr.Slider(
-            label="Crossfade Duration (ms)",
-            minimum=100,
-            maximum=2000,
-            value=1000,
-            step=100,
-            info="Crossfade duration between chunks."
-        )
-        bpm = gr.Slider(
-            label="Tempo (BPM)",
-            minimum=60,
-            maximum=180,
-            value=120,
-            step=1,
-            info="Beats per minute to influence the track's tempo."
-        )
-        drum_beat = gr.Dropdown(
-            label="Drum Beat",
-            choices=["none", "standard rock", "funk groove", "techno kick", "jazz swing"],
-            value="none",
-            info="Select a drum beat style to influence the rhythm."
-        )
-        synthesizer = gr.Dropdown(
-            label="Synthesizer",
-            choices=["none", "analog synth", "digital pad", "arpeggiated synth"],
-            value="none",
-            info="Select a synthesizer style to add electronic accents."
-        )
-        rhythmic_steps = gr.Dropdown(
-            label="Rhythmic Steps",
-            choices=["none", "syncopated steps", "steady steps", "complex steps"],
-            value="none",
-            info="Select a rhythmic step style to enhance the beat."
-        )
-        bass_style = gr.Dropdown(
-            label="Bass Style",
-            choices=["none", "slap bass", "deep bass", "melodic bass"],
-            value="none",
-            info="Select a bass style to shape the low end."
-        )
-        guitar_style = gr.Dropdown(
-            label="Guitar Style",
-            choices=["none", "distorted", "clean", "jangle"],
-            value="none",
-            info="Select a guitar style to define the riffs."
-        )
-        vocal_presence = gr.Dropdown(
-            label="Vocal Presence",
-            choices=["none", "background vocals", "vocal chops"],
-            value="none",
-            info="Select a vocal style to add vocal elements."
-        )
+        gr.Markdown("### ⚙️ API Settings")
+        with gr.Group(elem_classes="group-container"):
+            cfg_scale = gr.Slider(
+                label="CFG Scale 🎯",
+                minimum=1.0,
+                maximum=10.0,
+                value=3.0,
+                step=0.1,
+                info="Controls how closely the music follows the prompt."
+            )
+            top_k = gr.Slider(
+                label="Top-K Sampling 🔢",
+                minimum=10,
+                maximum=500,
+                value=250,
+                step=10,
+                info="Limits sampling to the top k most likely tokens."
+            )
+            top_p = gr.Slider(
+                label="Top-P Sampling 🎰",
+                minimum=0.0,
+                maximum=1.0,
+                value=0.9,
+                step=0.05,
+                info="Keeps tokens with cumulative probability above p."
+            )
+            temperature = gr.Slider(
+                label="Temperature 🔥",
+                minimum=0.1,
+                maximum=2.0,
+                value=1.0,
+                step=0.1,
+                info="Controls randomness; higher values increase diversity."
+            )
+            total_duration = gr.Radio(
+                label="Song Length ⏳ (seconds)",
+                choices=[30, 60, 90, 120],
+                value=30,
+                info="Select the total duration of the track."
+            )
+            chunk_duration = gr.Slider(
+                label="Chunk Duration ⏱️ (seconds)",
+                minimum=5,
+                maximum=15,
+                value=10,
+                step=1,
+                info="Duration of each chunk to render (5 to 15 seconds)."
+            )
+            crossfade_duration = gr.Slider(
+                label="Crossfade Duration 🎶 (ms)",
+                minimum=100,
+                maximum=2000,
+                value=1000,
+                step=100,
+                info="Crossfade duration between chunks."
+            )
+
+        gr.Markdown("### 🎵 Musical Controls")
+        with gr.Group(elem_classes="group-container"):
+            bpm = gr.Slider(
+                label="Tempo 🎵 (BPM)",
+                minimum=60,
+                maximum=180,
+                value=120,
+                step=1,
+                info="Beats per minute to set the track's tempo."
+            )
+            drum_beat = gr.Dropdown(
+                label="Drum Beat 🥁",
+                choices=["none", "standard rock", "funk groove", "techno kick", "jazz swing"],
+                value="none",
+                info="Select a drum beat style to influence the rhythm."
+            )
+            synthesizer = gr.Dropdown(
+                label="Synthesizer 🎹",
+                choices=["none", "analog synth", "digital pad", "arpeggiated synth"],
+                value="none",
+                info="Select a synthesizer style for electronic accents."
+            )
+            rhythmic_steps = gr.Dropdown(
+                label="Rhythmic Steps 👣",
+                choices=["none", "syncopated steps", "steady steps", "complex steps"],
+                value="none",
+                info="Select a rhythmic step style to enhance the beat."
+            )
+            bass_style = gr.Dropdown(
+                label="Bass Style 🎸",
+                choices=["none", "slap bass", "deep bass", "melodic bass"],
+                value="none",
+                info="Select a bass style to shape the low end."
+            )
+            guitar_style = gr.Dropdown(
+                label="Guitar Style 🎸",
+                choices=["none", "distorted", "clean", "jangle"],
+                value="none",
+                info="Select a guitar style to define the riffs."
+            )
+
         with gr.Row(elem_classes="action-buttons"):
-            gen_btn = gr.Button("Generate Music")
-            clr_btn = gr.Button("Clear Inputs")
+            gen_btn = gr.Button("Generate Music 🚀")
+            clr_btn = gr.Button("Clear Inputs 🧹")
     
     with gr.Column(elem_classes="output-container"):
-        out_audio = gr.Audio(label="Generated Stereo Instrumental Track", type="filepath")
-        status = gr.Textbox(label="Status", interactive=False)
+        gr.Markdown("### 🎧 Output")
+        out_audio = gr.Audio(label="Generated Instrumental Track 🎵", type="filepath")
+        status = gr.Textbox(label="Status 📢", interactive=False)
     
-    rhcp_btn.click(set_red_hot_chili_peppers_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    nirvana_btn.click(set_nirvana_grunge_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    pearl_jam_btn.click(set_pearl_jam_grunge_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    soundgarden_btn.click(set_soundgarden_grunge_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    foo_fighters_btn.click(set_foo_fighters_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    smashing_pumpkins_btn.click(set_smashing_pumpkins_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    radiohead_btn.click(set_radiohead_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    classic_rock_btn.click(set_classic_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    alternative_rock_btn.click(set_alternative_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    post_punk_btn.click(set_post_punk_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    indie_rock_btn.click(set_indie_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    funk_rock_btn.click(set_funk_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    detroit_techno_btn.click(set_detroit_techno_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
-    deep_house_btn.click(set_deep_house_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence], outputs=instrumental_prompt)
+    rhcp_btn.click(set_red_hot_chili_peppers_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    nirvana_btn.click(set_nirvana_grunge_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    pearl_jam_btn.click(set_pearl_jam_grunge_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    soundgarden_btn.click(set_soundgarden_grunge_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    foo_fighters_btn.click(set_foo_fighters_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    smashing_pumpkins_btn.click(set_smashing_pumpkins_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    radiohead_btn.click(set_radiohead_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    classic_rock_btn.click(set_classic_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    alternative_rock_btn.click(set_alternative_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    post_punk_btn.click(set_post_punk_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    indie_rock_btn.click(set_indie_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    funk_rock_btn.click(set_funk_rock_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    detroit_techno_btn.click(set_detroit_techno_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
+    deep_house_btn.click(set_deep_house_prompt, inputs=[bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style], outputs=instrumental_prompt)
     gen_btn.click(
         generate_music,
-        inputs=[instrumental_prompt, cfg_scale, total_duration, chunk_duration, crossfade_duration, bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence],
+        inputs=[instrumental_prompt, cfg_scale, top_k, top_p, temperature, total_duration, chunk_duration, crossfade_duration, bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style],
         outputs=[out_audio, status]
     )
     clr_btn.click(
         clear_inputs,
         inputs=None,
-        outputs=[instrumental_prompt, cfg_scale, total_duration, chunk_duration, crossfade_duration, bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style, vocal_presence]
+        outputs=[instrumental_prompt, cfg_scale, top_k, top_p, temperature, total_duration, chunk_duration, crossfade_duration, bpm, drum_beat, synthesizer, rhythmic_steps, bass_style, guitar_style]
     )
 
 # 9) TURN OFF OPENAPI/DOCS