Update simpler_app.py

simpler_app.py

@@ -60,8 +60,6 @@ mixtral_model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
 pipe = pipeline("text-generation", model=zephyr_model, torch_dtype=torch.bfloat16, device_map="auto")
 
-
-
 def prepare_lyrics_with_llm(theme, tags, lyrics):
 
     language = "English"
@@ -121,7 +119,12 @@ def general_process(theme, tags, lyrics):
     
     gr.Info("Generating audio ref")
     audio_ref = generate_audio_ref(tags)
-    return audio_ref, lyrics_result
+
+    if lyrics_result and audio_ref:
+        gr.Info("Generating Song")
+        generated_song = infer_music(lyrics_result, audio_ref)
+        
+    return audio_ref, lyrics_result, generated_song
 
     
 with gr.Blocks() as demo:
@@ -132,13 +135,14 @@ with gr.Blocks() as demo:
         style_tags = gr.Textbox(label="Music style tags")
         lyrics = gr.Textbox(label="Lyrics optional")
         submit_btn = gr.Button("Submit")
-        song_result = gr.Audio(label="Song result")
+        audio_ref = gr.Audio(label="Audio ref used")
         generated_lyrics = gr.Textbox(label="Generated Lyrics")
+        song_result = gr.Audio(label="Your generated Song")
 
     submit_btn.click(
         fn = general_process,
         inputs = [theme_song, style_tags, lyrics],
-        outputs = [song_result, generated_lyrics]
+        outputs = [audio_ref, generated_lyrics, song_result]
     )
 
 demo.queue().launch(show_api=False, show_error=True, ssr_mode=False)