Update app.py

app.py

@@ -157,7 +157,6 @@ with gr.Blocks() as demo:
             sound_design_output = gr.Textbox(label="Sound Design", lines=3)
             music_suggestion_output = gr.Textbox(label="Music Suggestions", lines=3)
 
-            # Ensure the .click is inside this Blocks context
             generate_script_button.click(
                 fn=lambda user_prompt, model_id, duration: generate_script(user_prompt, model_id, hf_token, duration),
                 inputs=[user_prompt, llama_model_id, duration],
@@ -172,7 +171,6 @@ with gr.Blocks() as demo:
             generate_voice_button = gr.Button("Generate Voice")
             voice_output = gr.Audio(label="Generated Voice", type="filepath")
 
-            # Ensure the .click is inside this Blocks context
             generate_voice_button.click(
                 fn=generate_voice,
                 inputs=[script_output, speaker],
@@ -183,13 +181,24 @@ with gr.Blocks() as demo:
         with gr.Tab("Step 3: Generate Music"):
             with gr.Row():
                 audio_length = gr.Slider(label="Music Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
+                music_prompt_display = gr.Textbox(
+                    label="Music Suggestions from Step 1", 
+                    lines=3, 
+                    interactive=False
+                )
 
             generate_music_button = gr.Button("Generate Music")
             music_output = gr.Audio(label="Generated Music", type="filepath")
 
-            # Ensure the .click is inside this Blocks context
+            # Pass music suggestions from Step 1 to Step 3
+            generate_script_button.click(
+                fn=lambda _, __, ___: None,  # Dummy function to update the display
+                inputs=[],
+                outputs=[music_suggestion_output],
+            )
+
             generate_music_button.click(
-                fn=generate_music,
+                fn=lambda prompt, audio_length: generate_music(prompt, audio_length),
                 inputs=[music_suggestion_output, audio_length],
                 outputs=[music_output],
             )
@@ -202,7 +211,6 @@ with gr.Blocks() as demo:
             blend_button = gr.Button("Blend Audio")
             final_output = gr.Audio(label="Final Promo Audio", type="filepath")
 
-            # Ensure the .click is inside this Blocks context
             blend_button.click(
                 fn=blend_audio,
                 inputs=[voice_output, music_output, ducking],
@@ -216,4 +224,4 @@ with gr.Blocks() as demo:
         </p>
     """)
 
-demo.launch(debug=True)
+demo.launch(debug=True)