Update app.py

app.py

@@ -4,150 +4,80 @@ from transformers import pipeline
 from moviepy import ImageClip, AudioFileClip 
 import os
 
-# --- 1. INITIALIZE ENGINES ---
-# Music Genre Classifier (Public & Fast)
+# --- ENGINES ---
 classifier = pipeline("audio-classification", model="dima806/music_genres_classification")
-
-# External AI Spaces (API Clients)
 mastering_client = Client("amaai-lab/SonicMaster")
 image_client = Client("black-forest-labs/FLUX.1-schnell")
 
-# --- 2. LOGIC FUNCTIONS ---
-
 def master_logic(audio_path):
     if not audio_path: return None, None, "Upload a track!", None
-    
-    # 1. Detect Genre
     results = classifier(audio_path)
     genre = results[0]['label']
     
-    # 2. AI Mastering
+    # API call: returns [audio_file, status_text]
     result = mastering_client.predict(
         handle_file(audio_path),
         f"Professional {genre} studio master.",
         42, 25, 3.5
     )
     
-    # --- THE CLEANUP FIX ---
-    raw_output = result[1] # This is "Saved: /path/to/file.wav\nElapsed: 19s"
-    
-    try:
-        # We split by "Saved: " and take the second part, 
-        # then split by "\n" and take the first part.
-        clean_path = raw_output.split("Saved: ")[1].split("\n")[0].strip()
-    except Exception:
-        # Fallback just in case the API format is different
-        clean_path = raw_output
-    
-    return clean_path, clean_path, f"Genre: {genre} | Mastered ✨", clean_path
+    # USE result[0] for the actual file
+    mastered_file = result[0]
+    return mastered_file, mastered_file, f"Genre: **{genre}** | Mastered ✨", mastered_file
 
 def art_logic(status_text, vibe):
-    if not status_text or "Ready" in status_text:
-        return None
-    
-    # Clean the genre from the status message
-    clean_genre = str(status_text).split("**")[1] if "**" in str(status_text) else "Music"
-    prompt = f"Professional album cover art, {clean_genre} style, {vibe}, 4k, ultra-detailed, no text."
+    if not status_text or "Ready" in status_text: return None
+    clean_genre = status_text.split("**")[1] if "**" in status_text else "Music"
     
-    # Flux API call
     img_result = image_client.predict(
-        prompt=prompt,
-        seed=0,
-        width=1024,
-        height=1024,
-        guidance_scale=3.5,
-        num_inference_steps=4,
+        prompt=f"Professional album cover, {clean_genre}, {vibe}, high res",
+        seed=0, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=4,
         api_name="/infer"
     )
     return img_result[0]
 
 def video_logic(audio_path, image_path):
-    if not audio_path or not image_path: 
-        return None
-    
-    # MoviePy v2.0+ Syntax (Uses 'with_' and 'subclipped')
+    if not audio_path or not image_path: return None
     audio = AudioFileClip(audio_path).subclipped(0, 30)
     img = ImageClip(image_path).with_duration(audio.duration).resized(width=1080)
-    
-    # Create 9:16 Vertical Video
-    video = img.on_color(size=(1080, 1920), color=(15, 15, 15), pos="center")
-    video = video.with_audio(audio)
-    
-    out_path = "promo_snippet.mp4"
-    video.write_videofile(out_path, fps=24, codec="libx264", audio_codec="aac")
-    return out_path
+    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
 
-# --- 3. UI LAYOUT ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML("<h1 style='text-align: center;'>🎧 AI Artist Command Center</h1>")
-    
-    # Persistent storage for audio paths
+# --- UI ---
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎵 AI Artist Studio")
     raw_storage = gr.State()
     master_storage = gr.State()
 
     with gr.Tabs():
-        # TAB 1: MASTERING & COMPARISON
-        with gr.TabItem("🎙️ 1. Master"):
+        with gr.TabItem("🎧 1. Master"):
+            in_audio = gr.Audio(label="Upload", type="filepath")
+            master_btn = gr.Button("🚀 MASTER", variant="primary")
             with gr.Row():
-                in_audio = gr.Audio(label="Step 1: Upload Raw Demo", type="filepath")
-                with gr.Column():
-                    monitor = gr.Audio(label="Step 2: Studio Monitor", interactive=False)
-                    ab_toggle = gr.Radio(
-                        choices=["Original 🔈", "Mastered ✨"], 
-                        value="Mastered ✨", 
-                        label="A/B Comparison (Instant Bypass)"
-                    )
-            
-            master_btn = gr.Button("🚀 START MASTERING", variant="primary")
-            status_display = gr.Markdown("Ready to analyze...")
-            download_wav = gr.File(label="Download High-Res WAV")
-
-        # TAB 2: ARTWORK GENERATION
-        with gr.TabItem("🖼️ 2. Album Art"):
-            with gr.Row():
-                with gr.Column():
-                    vibe_in = gr.Textbox(label="Visual Vibe", placeholder="e.g. Cyberpunk Tokyo, oil painting...")
-                    art_btn = gr.Button("🎨 GENERATE COVER")
-                art_out = gr.Image(label="Your Artwork", type="filepath")
-
-        # TAB 3: SOCIAL MEDIA EXPORT
-        with gr.TabItem("📱 3. Social Promo"):
-            promo_btn = gr.Button("🎬 CREATE TIKTOK SNIPPET", variant="primary")
-            video_out = gr.Video(label="Final 9:16 Video")
-
-    # --- 4. WIRING THE EVENTS ---
-    
-    # Mastering logic
-    master_btn.click(
-        fn=master_logic, 
-        inputs=in_audio, 
-        outputs=[monitor, download_wav, status_display, master_storage]
-    ).then(
-        fn=lambda x: x, inputs=in_audio, outputs=raw_storage
-    )
-
-    # A/B Toggle logic
-    ab_toggle.change(
-        fn=toggle_ab, 
-        inputs=[ab_toggle, raw_storage, master_storage], 
-        outputs=monitor
-    )
-
-    # Art logic (takes the genre from status + user vibe)
-    art_btn.click(
-        fn=art_logic, 
-        inputs=[status_display, vibe_in], 
-        outputs=art_out
-    )
-
-    # Video logic (takes mastered audio + generated art)
-    promo_btn.click(
-        fn=video_logic, 
-        inputs=[master_storage, art_out], 
-        outputs=video_out
+                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.TabItem("🎨 2. Art"):
+            vibe_in = gr.Textbox(label="Vibe")
+            art_btn = gr.Button("🎨 ART")
+            art_out = gr.Image()
+
+        with gr.TabItem("📱 3. Promo"):
+            promo_btn = gr.Button("🎬 VIDEO")
+            video_out = gr.Video()
+
+    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)
 
-demo.launch()
+# Theme moved to launch to fix the UserWarning
+demo.launch(theme=gr.themes.Soft())