Update app.py

app.py

@@ -109,20 +109,18 @@ def generate_audio(prompt: str, audio_length: int):
 # ---------------------------------------------------------------------
 # Gradio Interface
 # ---------------------------------------------------------------------
-def radio_imaging_script(user_prompt, llama_model_id):
+def radio_imaging_app(user_prompt, llama_model_id, audio_length):
     # 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
+        return pipeline_llama, None
 
     # Generate Script
     script = generate_script(user_prompt, pipeline_llama)
-    return script
 
-def radio_imaging_audio(script, audio_length):
     # Generate Audio
     audio_data = generate_audio(script, audio_length)
-    return audio_data
+    return script, audio_data
 
 # ---------------------------------------------------------------------
 # Interface
@@ -130,31 +128,20 @@ def radio_imaging_audio(script, audio_length):
 with gr.Blocks() as demo:
     gr.Markdown("# 🎧 AI Radio Imaging with Llama 3 + MusicGen (Zero GPU)")
 
-    # Script Generation Section
     with gr.Row():
-        gr.Markdown("## Step 1: Generate the Promo Script")
         user_prompt = gr.Textbox(label="Enter your promo idea", placeholder="E.g., A 15-second hype jingle for a morning talk show.")
         llama_model_id = gr.Textbox(label="Llama 3 Model ID", value="meta-llama/Meta-Llama-3-70B")
-        generate_script_button = gr.Button("Generate Promo Script")
-        script_output = gr.Textbox(label="Generated Script", interactive=False)
-
-        generate_script_button.click(
-            fn=radio_imaging_script,
-            inputs=[user_prompt, llama_model_id],
-            outputs=script_output
-        )
-
-    # Audio Generation Section
-    with gr.Row():
-        gr.Markdown("## Step 2: Generate the Sound")
         audio_length = gr.Slider(label="Audio Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
-        generate_audio_button = gr.Button("Generate Sound from Script")
-        audio_output = gr.Audio(label="Generated Audio", type="filepath")
 
-        generate_audio_button.click(
-            fn=radio_imaging_audio,
-            inputs=[script_output, audio_length],
-            outputs=audio_output
-        )
+    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")
+
+    generate_button.click(radio_imaging_app, 
+                          inputs=[user_prompt, llama_model_id, audio_length], 
+                          outputs=[script_output, audio_output])
 
+# ---------------------------------------------------------------------
+# Launch App
+# ---------------------------------------------------------------------
 demo.launch(debug=True)