Update app.py

app.py

@@ -24,7 +24,6 @@ hf_token = os.getenv("HF_TOKEN")
 # Script Generation Function
 # ---------------------------------------------------------------------
 @spaces.GPU(duration=300)
-@spaces.GPU(duration=300)
 def generate_script(user_prompt: str, model_id: str, token: str, duration: int):
     try:
         tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
@@ -145,12 +144,13 @@ with gr.Blocks() as demo:
 
             generate_script_button = gr.Button("Generate Script")
             script_output = gr.Textbox(label="Generated Script")
+            sound_design_output = gr.Textbox(label="Sound Design")
             music_suggestion_output = gr.Textbox(label="Music Suggestion")
 
             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],
-                outputs=[script_output, music_suggestion_output],
+                fn=generate_script,
+                inputs=[user_prompt, llama_model_id, hf_token, duration],
+                outputs=[script_output, sound_design_output, music_suggestion_output],
             )
 
         with gr.Tab("Step 2: Generate Voice"):
@@ -161,7 +161,7 @@ with gr.Blocks() as demo:
             voice_output = gr.Audio(label="Generated Voice", type="filepath")
 
             generate_voice_button.click(
-                fn=lambda script, speaker: generate_voice(script, speaker),
+                fn=generate_voice,
                 inputs=[script_output, speaker],
                 outputs=[voice_output],
             )
@@ -174,7 +174,7 @@ with gr.Blocks() as demo:
             music_output = gr.Audio(label="Generated Music", type="filepath")
 
             generate_music_button.click(
-                fn=lambda music_suggestion, audio_length: generate_music(music_suggestion, audio_length),
+                fn=generate_music,
                 inputs=[music_suggestion_output, audio_length],
                 outputs=[music_output],
             )
@@ -187,7 +187,7 @@ with gr.Blocks() as demo:
             final_output = gr.Audio(label="Final Promo Audio", type="filepath")
 
             blend_button.click(
-                fn=lambda voice_path, music_path, ducking: blend_audio(voice_path, music_path, ducking),
+                fn=blend_audio,
                 inputs=[voice_output, music_output, ducking],
                 outputs=[final_output],
             )
@@ -199,4 +199,4 @@ with gr.Blocks() as demo:
         </p>
     """)
 
-demo.launch(debug=True)
+demo.launch(debug=True)