Update app.py

app.py

@@ -1,19 +1,18 @@
 import gradio as gr
 from musicgen import generate_music
 from tts_singing import generate_singing
-from merge_audio import merge_music_and_vocals
+from merge_audio import merge_audio_tracks
 
-def create_song(prompt):
-    music_path = generate_music(prompt)
-    singing_path = generate_singing(prompt)
-    final_song = merge_music_and_vocals(music_path, singing_path)
+def text_to_song(prompt):
+    instrumental = generate_music(prompt)
+    vocals = generate_singing(prompt)
+    final_song = merge_audio_tracks(vocals, instrumental)
     return final_song
 
-iface = gr.Interface(
-    fn=create_song,
-    inputs=gr.Textbox(label="Enter your song prompt (e.g., 'A happy love song')"),
+gr.Interface(
+    fn=text_to_song,
+    inputs=gr.Textbox(label="Enter your song prompt (e.g. 'A happy birthday song')"),
     outputs=gr.Audio(label="Generated Song"),
-    title="🎵 Songify AI (Text-to-Song Generator)"
-)
-
-iface.launch()
+    title="🎶 Text-to-Song Generator",
+    description="Enter any song idea and get music + vocals (CPU only)",
+).launch()