Update app.py

app.py

@@ -25,7 +25,7 @@ musicgen_processor = None
 # ---------------------------------------------------------------------
 # Load Llama 3 Model with Zero GPU (Lazy Loading)
 # ---------------------------------------------------------------------
-@spaces.GPU(duration=300)  # Increased duration to 300 seconds
+@spaces.GPU(duration=300)
 def load_llama_pipeline_zero_gpu(model_id: str, token: str):
     global llama_pipeline
     if llama_pipeline is None:
@@ -101,38 +101,25 @@ def generate_audio(prompt: str, audio_length: int):
 # ---------------------------------------------------------------------
 # Gradio Interface
 # ---------------------------------------------------------------------
-def radio_imaging_app(user_prompt, llama_model_id, audio_length):
+def radio_imaging_app(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 generate_script(user_prompt, pipeline_llama)
+
 
-    # Generate Audio
-    audio_data = generate_audio(script, audio_length)
-    return script, audio_data
+def generate_audio_from_script(script, audio_length):
+    return generate_audio(script, audio_length)
 
 
 # ---------------------------------------------------------------------
 # Interface
 # ---------------------------------------------------------------------
-with gr.Blocks(css="""
-#app-title {
-    text-align: center;
-    font-size: 2rem;
-    font-weight: bold;
-    color: #4CAF50;
-}
-#subsection {
-    margin: 20px 0;
-    font-size: 1.2rem;
-    color: #333;
-    text-align: center;
-}
-""") as demo:
-    gr.Markdown('<div id="app-title">🎧 AI Radio Imaging with Llama 3 + MusicGen (Zero GPU)</div>')
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎧 AI Radio Imaging with Llama 3 + MusicGen (Zero GPU)")
 
     with gr.Tab("Step 1: Generate Promo Script"):
         with gr.Row():
@@ -149,8 +136,8 @@ with gr.Blocks(css="""
 
         generate_script_button.click(
             fn=radio_imaging_app,
-            inputs=[user_prompt, llama_model_id, gr.State(0)],
-            outputs=[script_output, None],
+            inputs=[user_prompt, llama_model_id],
+            outputs=script_output,
         )
 
     with gr.Tab("Step 2: Generate Audio"):
@@ -166,7 +153,7 @@ with gr.Blocks(css="""
             audio_output = gr.Audio(label="Generated Audio", type="filepath")
 
         generate_audio_button.click(
-            fn=generate_audio,
+            fn=generate_audio_from_script,
             inputs=[script_output, audio_length],
             outputs=audio_output,
         )