Update app.py

app.py

@@ -45,8 +45,8 @@ def generate_script(user_prompt: str, model_id: str, token: str, duration: int):
 
         # Extract the script and music suggestion
         generated_text = result[0]["generated_text"]
-        if "Music suggestion:" in generated_text:
-            parts = generated_text.split("Music suggestion:", 1)
+        if "Music:" in generated_text:
+            parts = generated_text.split("Music:", 1)
             script = parts[0].strip()
             music_suggestion = parts[1].strip()
             return script, music_suggestion
@@ -137,7 +137,8 @@ with gr.Blocks() as demo:
         ducking = gr.Checkbox(label="Enable Ducking", value=True)
 
     generate_script_button = gr.Button("Generate Script")
-    script_output = gr.Textbox(label="Generated Script and Music Suggestion")
+    script_output = gr.Textbox(label="Generated Script")
+    music_suggestion_output = gr.Textbox(label="Music Suggestion")
     generate_voice_button = gr.Button("Generate Voice")
     voice_output = gr.Audio(label="Generated Voice", type="filepath")
     generate_music_button = gr.Button("Generate Music")
@@ -160,7 +161,7 @@ with gr.Blocks() as demo:
     generate_script_button.click(
         fn=step_generate_script,
         inputs=[user_prompt, llama_model_id, duration],
-        outputs=[script_output],
+        outputs=[script_output, music_suggestion_output],
     )
 
     generate_voice_button.click(
@@ -171,7 +172,7 @@ with gr.Blocks() as demo:
 
     generate_music_button.click(
         fn=step_generate_music,
-        inputs=[script_output, audio_length],
+        inputs=[music_suggestion_output, audio_length],
         outputs=[music_output],
     )