Update app.py

app.py

@@ -109,18 +109,20 @@ def generate_audio(prompt: str, audio_length: int):
 # ---------------------------------------------------------------------
 # Gradio Interface
 # ---------------------------------------------------------------------
-def radio_imaging_app(user_prompt, llama_model_id, audio_length):
+def generate_script_interface(user_prompt, llama_model_id):
     # Load Llama 3 Pipeline with Zero GPU
     pipeline_llama = load_llama_pipeline_zero_gpu(llama_model_id, hf_token)
     if isinstance(pipeline_llama, str):
-        return pipeline_llama, None
+        return pipeline_llama
 
     # Generate Script
     script = generate_script(user_prompt, pipeline_llama)
+    return script
 
+def generate_audio_interface(script, audio_length):
     # Generate Audio
     audio_data = generate_audio(script, audio_length)
-    return script, audio_data
+    return audio_data
 
 # ---------------------------------------------------------------------
 # Interface
@@ -133,16 +135,27 @@ with gr.Blocks() as demo:
         llama_model_id = gr.Textbox(label="Llama 3 Model ID", value="meta-llama/Meta-Llama-3-70B")
         audio_length = gr.Slider(label="Audio Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
 
-    generate_button = gr.Button("Generate Promo Script and Audio")
-    script_output = gr.Textbox(label="Generated Script")
-    audio_output = gr.Audio(label="Generated Audio", type="filepath")
+    with gr.Row():
+        generate_script_button = gr.Button("Generate Promo Script")
+        script_output = gr.Textbox(label="Generated Script", interactive=False)
+
+    with gr.Row():
+        generate_audio_button = gr.Button("Generate Audio")
+        audio_output = gr.Audio(label="Generated Audio", type="filepath")
 
-    generate_button.click(radio_imaging_app, 
-                          inputs=[user_prompt, llama_model_id, audio_length], 
-                          outputs=[script_output, audio_output])
+    generate_script_button.click(
+        generate_script_interface, 
+        inputs=[user_prompt, llama_model_id], 
+        outputs=script_output
+    )
+
+    generate_audio_button.click(
+        generate_audio_interface, 
+        inputs=[script_output, audio_length], 
+        outputs=audio_output
+    )
 
 # ---------------------------------------------------------------------
 # Launch App
 # ---------------------------------------------------------------------
 demo.launch(debug=True)
-