Update app.py

app.py

@@ -9,75 +9,127 @@ classifier = pipeline("audio-classification", model="dima806/music_genres_classi
 mastering_client = Client("amaai-lab/SonicMaster")
 image_client = Client("black-forest-labs/FLUX.1-schnell")
 
-def master_logic(audio_path):
-    if not audio_path: return None, None, "Upload a track!", None
+# --- LOGIC ---
+
+def master_logic(audio_path, brightness, loudness, reverb, steps, cfg):
+    if not audio_path: return None, None, "⚠️ Upload audio!", None
+    
+    # 1. Detect Genre
     results = classifier(audio_path)
     genre = results[0]['label']
     
-    # API call: returns [audio_file, status_text]
+    # 2. Map Sliders to AI Instructions
+    bright_txt = "increase high-end clarity and brightness" if brightness > 7 else ("warm vintage roll-off" if brightness < 3 else "balanced EQ")
+    loud_txt = "aggressive limiting for maximum loudness" if loudness > 7 else ("wide natural dynamic range" if loudness < 3 else "standard radio compression")
+    verb_txt = "spacious hall reverb" if reverb > 7 else ("dry intimate studio sound" if reverb < 3 else "natural room acoustics")
+    
+    full_prompt = f"Professional {genre} studio master, {bright_txt}, {loud_txt}, {verb_txt}."
+    
+    # 3. AI Mastering API
     result = mastering_client.predict(
         handle_file(audio_path),
-        f"Professional {genre} studio master.",
-        42, 25, 3.5
+        full_prompt,
+        42,      # Seed
+        int(steps), 
+        float(cfg)
     )
     
-    # USE result[0] for the actual file
-    mastered_file = result[0]
-    return mastered_file, mastered_file, f"Genre: **{genre}** | Mastered ✨", mastered_file
+    mastered_file = result[0] # The actual downloaded file
+    status_msg = f"✅ **{genre}** Mastered with {int(steps)} steps at {cfg} intensity."
+    
+    return mastered_file, mastered_file, status_msg, mastered_file
 
-def art_logic(status_text, vibe):
+def art_logic(status_text, vibe, detail_steps, creativity_cfg):
     if not status_text or "Ready" in status_text: return None
-    clean_genre = status_text.split("**")[1] if "**" in status_text else "Music"
+    
+    genre = status_text.split("**")[1] if "**" in status_text else "Music"
+    prompt = f"Professional album cover art, {genre} style, {vibe}, 4k, ultra-detailed, no text."
     
     img_result = image_client.predict(
-        prompt=f"Professional album cover, {clean_genre}, {vibe}, high res",
-        seed=0, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=4,
+        prompt=prompt,
+        seed=0,
+        width=1024,
+        height=1024,
+        guidance_scale=float(creativity_cfg),
+        num_inference_steps=int(detail_steps),
         api_name="/infer"
     )
     return img_result[0]
 
-def video_logic(audio_path, image_path):
+def video_logic(audio_path, image_path, volume_level):
     if not audio_path or not image_path: return None
-    audio = AudioFileClip(audio_path).subclipped(0, 30)
+    
+    # Load and adjust levels
+    audio = AudioFileClip(audio_path).subclipped(0, 30).volumex(volume_level)
     img = ImageClip(image_path).with_duration(audio.duration).resized(width=1080)
+    
+    # Composition
     video = img.on_color(size=(1080, 1920), color=(15, 15, 15), pos="center").with_audio(audio)
-    video.write_videofile("promo.mp4", fps=24, codec="libx264")
-    return "promo.mp4"
-
-def toggle_ab(choice, raw, mastered):
-    return mastered if "Mastered" in choice else raw
+    
+    out_path = "final_promo.mp4"
+    video.write_videofile(out_path, fps=24, codec="libx264", audio_codec="aac")
+    return out_path
 
-# --- UI ---
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎵 AI Artist Studio")
+# --- UI LAYOUT ---
+with gr.Blocks(theme=gr.themes.Base()) as demo:
+    gr.HTML("<h1 style='text-align: center;'>🎚️ AI Studio: Effect Console</h1>")
+    
     raw_storage = gr.State()
     master_storage = gr.State()
 
     with gr.Tabs():
-        with gr.TabItem("🎧 1. Master"):
-            in_audio = gr.Audio(label="Upload", type="filepath")
-            master_btn = gr.Button("🚀 MASTER", variant="primary")
+        # TAB 1: MASTERING
+        with gr.TabItem("🎧 Mastering Console"):
             with gr.Row():
-                monitor = gr.Audio(label="Monitor")
-                ab_toggle = gr.Radio(["Original 🔈", "Mastered ✨"], value="Mastered ✨", label="A/B")
-            status = gr.Markdown("Ready")
-            export_file = gr.File(label="WAV")
+                with gr.Column(scale=1):
+                    in_audio = gr.Audio(label="Raw Upload", type="filepath")
+                    master_btn = gr.Button("🚀 APPLY MASTERING", variant="primary")
+                
+                with gr.Column(scale=1):
+                    gr.Markdown("### FX Levels")
+                    bright_sl = gr.Slider(0, 10, value=5, label="Brightness (EQ)")
+                    loud_sl = gr.Slider(0, 10, value=5, label="Loudness (Limiters)")
+                    verb_sl = gr.Slider(0, 10, value=2, label="Space (Reverb)")
+                    
+                    with gr.Accordion("Advanced Engine Settings", open=False):
+                        m_steps = gr.Slider(10, 50, value=25, step=1, label="Refinement Steps")
+                        m_cfg = gr.Slider(1.0, 15.0, value=3.5, step=0.5, label="AI Guidance Scale")
 
-        with gr.TabItem("🎨 2. Art"):
-            vibe_in = gr.Textbox(label="Vibe")
-            art_btn = gr.Button("🎨 ART")
-            art_out = gr.Image()
+            status = gr.Markdown("Ready.")
+            with gr.Row():
+                monitor = gr.Audio(label="Studio Monitor")
+                export_wav = gr.File(label="Download Master")
 
-        with gr.TabItem("📱 3. Promo"):
-            promo_btn = gr.Button("🎬 VIDEO")
-            video_out = gr.Video()
+        # TAB 2: ARTWORK
+        with gr.TabItem("🎨 Art Direction"):
+            with gr.Row():
+                with gr.Column():
+                    vibe_in = gr.Textbox(label="Visual Theme", placeholder="e.g. Neon Cyberpunk, 70s Vinyl...")
+                    art_btn = gr.Button("🎨 GENERATE ART")
+                    
+                    gr.Markdown("### Image 'Levels'")
+                    art_steps = gr.Slider(1, 4, value=4, step=1, label="Detail Iterations")
+                    art_cfg = gr.Slider(0.0, 5.0, value=3.5, label="Prompt Strictness")
+                
+                art_out = gr.Image(label="Cover Art")
 
-    master_btn.click(master_logic, in_audio, [monitor, export_file, status, master_storage]).then(
-        lambda x: x, in_audio, raw_storage
-    )
-    ab_toggle.change(toggle_ab, [ab_toggle, raw_storage, master_storage], monitor)
-    art_btn.click(art_logic, [status, vibe_in], art_out)
-    promo_btn.click(video_logic, [master_storage, art_out], video_out)
+        # TAB 3: PROMO EXPORT
+        with gr.TabItem("🎬 Video Export"):
+            with gr.Row():
+                with gr.Column():
+                    vol_sl = gr.Slider(0.0, 2.0, value=1.0, label="Final Video Volume (Gain)")
+                    promo_btn = gr.Button("🎬 RENDER TIKTOK CLIP", variant="primary")
+                video_out = gr.Video(label="9:16 Social Preview")
+
+    # --- WIRING ---
+    master_btn.click(
+        master_logic, 
+        [in_audio, bright_sl, loud_sl, verb_sl, m_steps, m_cfg], 
+        [monitor, export_wav, status, master_storage]
+    ).then(lambda x: x, in_audio, raw_storage)
+
+    art_btn.click(art_logic, [status, vibe_in, art_steps, art_cfg], art_out)
+    
+    promo_btn.click(video_logic, [master_storage, art_out, vol_sl], video_out)
 
-# Theme moved to launch to fix the UserWarning
-demo.launch(theme=gr.themes.Soft())
+demo.launch()