Update app.py

app.py

@@ -1,18 +1,17 @@
 import gradio as gr
 from gradio_client import Client, handle_file
 from transformers import pipeline
-from moviepy import ImageClip, AudioFileClip # Modern v2.x Import
+from moviepy import ImageClip, AudioFileClip 
 import os
 
-# --- INITIALIZE AI ENGINES ---
-# Identifies the genre (Rock, Pop, etc.)
-classifier = pipeline("audio-classification", model="MIT/ast-finetuned-audioset")
-# The "Pro" Mastering Engine
+# --- 1. INITIALIZE ENGINES ---
+# CORRECTED: Using a dedicated music genre classifier to avoid 401 errors
+classifier = pipeline("audio-classification", model="dima806/music_genres_classification")
+
 mastering_client = Client("amaai-lab/SonicMaster")
-# The Artwork Engine
 image_client = Client("stabilityai/stable-diffusion-xl-base-1.0")
 
-# --- LOGIC FUNCTIONS ---
+# --- 2. LOGIC FUNCTIONS ---
 
 def master_logic(audio_path):
     if not audio_path: return None, None, "Upload a track!", None
@@ -22,10 +21,9 @@ def master_logic(audio_path):
     genre = results[0]['label']
     
     # 2. AI Mastering
-    # We send the audio + a smart prompt based on the genre
     result = mastering_client.predict(
         audio=handle_file(audio_path),
-        prompt=f"Professional {genre} studio master, high-end clarity.",
+        prompt=f"Professional {genre} studio master.",
         api_name="/predict"
     )
     mastered_path = result[1]
@@ -33,17 +31,16 @@ def master_logic(audio_path):
     return mastered_path, mastered_path, f"Genre: {genre} | Mastered ✨", mastered_path
 
 def art_logic(genre_text, vibe):
-    prompt = f"Professional album cover art, {genre_text} style, {vibe}, 4k, artistic, no text."
+    # If genre is still a full status string, clean it
+    clean_genre = str(genre_text).split("|")[0].replace("Genre:", "").strip()
+    prompt = f"Professional album cover art, {clean_genre} style, {vibe}, 4k, no text."
     return image_client.predict(prompt=prompt, api_name="/predict")
 
 def video_logic(audio_path, image_path):
     if not audio_path or not image_path: return None
-    
-    # MoviePy v2.x Syntax: Use 'with_' and 'subclipped'
+    # MoviePy v2.x Syntax
     audio = AudioFileClip(audio_path).subclipped(0, 30)
     img = ImageClip(image_path).with_duration(audio.duration).resized(width=1080)
-    
-    # Create 9:16 video for TikTok
     video = img.on_color(size=(1080, 1920), color=(15, 15, 15), pos="center")
     video = video.with_audio(audio)
     
@@ -51,60 +48,37 @@ def video_logic(audio_path, image_path):
     video.write_videofile(out_path, fps=24, codec="libx264")
     return out_path
 
-# --- THE INTERFACE ---
-
+# --- 3. UI LAYOUT ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🎚️ The Ultimate AI Artist Studio")
+    gr.Markdown("# 🎵 AI Artist Suite")
     
-    # Memory for the app
+    # States
     raw_storage = gr.State()
     master_storage = gr.State()
     genre_name = gr.State()
 
     with gr.Tabs():
-        # TAB 1: MASTERING
-        with gr.TabItem("🎧 1. Mastering"):
-            with gr.Row():
-                in_audio = gr.Audio(label="Upload Song", type="filepath")
-                with gr.Column():
-                    out_audio = gr.Audio(label="Studio Monitor", interactive=False)
-                    ab_toggle = gr.Radio(["Original 🔈", "Mastered ✨"], value="Mastered ✨", label="A/B Comparison")
-            
-            master_btn = gr.Button("🚀 MASTER TRACK", variant="primary")
-            status = gr.Markdown("Ready to process...")
-            download_wav = gr.File(label="Download Master (.wav)")
+        with gr.TabItem("🎧 Mastering"):
+            in_audio = gr.Audio(label="Raw Track", type="filepath")
+            master_btn = gr.Button("🚀 MASTER")
+            out_audio = gr.Audio(label="Output")
+            status = gr.Markdown()
+            export_file = gr.File(label="Download WAV")
 
-        # TAB 2: ARTWORK
-        with gr.TabItem("🎨 2. Album Art"):
-            vibe_input = gr.Textbox(label="Visual Vibe", placeholder="e.g. A dystopian sunset, watercolor style...")
-            art_btn = gr.Button("🎨 Generate Cover Art")
-            art_output = gr.Image(label="Your Artwork")
+        with gr.TabItem("🎨 Art"):
+            vibe_in = gr.Textbox(label="Visual Vibe")
+            art_btn = gr.Button("🎨 GENERATE ART")
+            art_out = gr.Image()
 
-        # TAB 3: PROMO
-        with gr.TabItem("📱 3. Social Export"):
-            promo_btn = gr.Button("🎬 Create TikTok Snippet")
-            promo_video = gr.Video(label="9:16 Promo Video")
+        with gr.TabItem("📱 Promo"):
+            promo_btn = gr.Button("🎬 CREATE VIDEO")
+            video_out = gr.Video()
 
-    # --- WIRING IT TOGETHER ---
-    
-    # Mastering Click
-    master_btn.click(
-        master_logic, in_audio, [out_audio, download_wav, status, master_storage]
-    ).then(lambda x: x, in_audio, raw_storage).then(
-        # Extract genre name from status string for the next steps
-        lambda x: x.split(":")[1].split("|")[0].strip(), status, genre_name
+    # Wiring
+    master_btn.click(master_logic, in_audio, [out_audio, export_file, status, master_storage]).then(
+        lambda x: x, in_audio, raw_storage
     )
-
-    # A/B Toggle logic
-    ab_toggle.change(
-        lambda choice, raw, mastered: mastered if "Mastered" in choice else raw,
-        [ab_toggle, raw_storage, master_storage], out_audio
-    )
-
-    # Art Click
-    art_btn.click(art_logic, [genre_name, vibe_input], art_output)
-
-    # Video Click
-    promo_btn.click(video_logic, [master_storage, art_output], promo_video)
+    art_btn.click(art_logic, [status, vibe_in], art_out)
+    promo_btn.click(video_logic, [master_storage, art_out], video_out)
 
 demo.launch()