Update app.py

app.py

@@ -65,7 +65,7 @@ def generate_audio(prompt: str, audio_length: int, mg_model, mg_processor):
         normalized_audio = (audio_data / max(abs(audio_data)) * 32767).astype("int16")
         output_file = "radio_jingle.wav"
         wav.write(output_file, rate=sr, data=normalized_audio)
-        return output_file
+        return sr, normalized_audio
     except Exception as e:
         return str(e)
 
@@ -87,11 +87,11 @@ def radio_imaging_app(user_prompt, llama_model_id, hf_token, audio_length):
         return script, mg_processor
 
     # Generate Audio
-    audio_file = generate_audio(script, audio_length, mg_model, mg_processor)
-    if isinstance(audio_file, str) and audio_file.startswith("Error"):
-        return script, audio_file
+    audio_data = generate_audio(script, audio_length, mg_model, mg_processor)
+    if isinstance(audio_data, str):
+        return script, audio_data
 
-    return script, audio_file
+    return script, audio_data
 
 # ---------------------------------------------------------------------
 # Interface
@@ -106,7 +106,7 @@ with gr.Blocks() as demo:
 
     generate_button = gr.Button("Generate Promo Script and Audio")
     script_output = gr.Textbox(label="Generated Script")
-    audio_output = gr.Audio(label="Generated Audio", type="file")
+    audio_output = gr.Audio(label="Generated Audio", type="numpy")
 
     generate_button.click(radio_imaging_app, 
                           inputs=[user_prompt, llama_model_id, hf_token, audio_length],