Update app.py

app.py

@@ -2,13 +2,18 @@ import gradio as gr
 import asyncio
 import wave
 import os
+import numpy as np
+import nest_asyncio
 from google import genai
 from google.genai import types
 
+# Enable nested asyncio (required for Gradio + asyncio)
+nest_asyncio.apply()
+
+# Configure Gemini API
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 client = genai.Client(http_options={'api_version': 'v1alpha'}, api_key=GOOGLE_API_KEY)
 
-
 # Save PCM audio to WAV file
 def save_wave_file(filename, pcm, channels=1, rate=24000, sample_width=2):
     with wave.open(filename, "wb") as wf:
@@ -17,7 +22,7 @@ def save_wave_file(filename, pcm, channels=1, rate=24000, sample_width=2):
         wf.setframerate(rate)
         wf.writeframes(pcm)
 
-# Async music generation
+# Async music generation function
 async def generate_music(prompt, bpm, temperature):
     audio_chunks = []
 
@@ -45,24 +50,29 @@ async def generate_music(prompt, bpm, temperature):
             await session.pause()
 
         all_pcm = b"".join(audio_chunks)
+
+        # Save WAV file
         output_path = "generated_music.wav"
         save_wave_file(output_path, all_pcm)
-        return output_path, output_path, "Music generated successfully."
+
+        # Convert PCM to numpy array for audio playback
+        audio_np = np.frombuffer(all_pcm, dtype=np.int16)
+        return (24000, audio_np), output_path, "Music generated successfully!"
 
     except Exception as e:
-        return None, None, f"Error: {str(e)}"
+        return None, None, f"❌ Error: {str(e)}"
 
-# Wrapper for Gradio to run async function
+# Wrapper for Gradio
 def generate_music_gradio(prompt, bpm, temperature):
-    return asyncio.run(generate_music(prompt, bpm, temperature))
+    loop = asyncio.get_event_loop()
+    return loop.run_until_complete(generate_music(prompt, bpm, temperature))
 
 # Gradio UI
 with gr.Blocks(title="Gemini Lyria Music Generator") as demo:
-    gr.Markdown("## Lyria Music Generator")
+    gr.Markdown("## 🎶 Gemini Lyria Music Generator")
 
-    # Section 1: Input
     with gr.Group():
-        gr.Markdown("### Input")
+        gr.Markdown("### 🎛 Input")
         with gr.Row():
             prompt_input = gr.Textbox(
                 label="Music Style / Prompt",
@@ -71,19 +81,17 @@ with gr.Blocks(title="Gemini Lyria Music Generator") as demo:
         with gr.Row():
             bpm_input = gr.Slider(label="BPM", minimum=60, maximum=180, value=90)
             temp_input = gr.Slider(label="Temperature", minimum=0.1, maximum=2.0, step=0.1, value=1.0)
-        generate_btn = gr.Button("Generate Music")
+        generate_btn = gr.Button("🎧 Generate Music")
 
-    # Section 2: Output
     with gr.Group():
-        gr.Markdown("### Output")
+        gr.Markdown("### 🎵 Output")
         with gr.Row():
-            output_audio = gr.Audio(label="Generated Audio")
+            output_audio = gr.Audio(label="Generated Audio", type="numpy")
             download_file = gr.File(label="Download WAV")
         status_output = gr.Textbox(label="Status", interactive=False)
 
-    # Section 3: Examples
     with gr.Group():
-        gr.Markdown("### Examples")
+        gr.Markdown("### 🔁 Examples")
         examples = gr.Examples(
             examples=[
                 ["minimal techno", 125, 1.0],
@@ -95,7 +103,6 @@ with gr.Blocks(title="Gemini Lyria Music Generator") as demo:
             inputs=[prompt_input, bpm_input, temp_input]
         )
 
-    # Event binding
     generate_btn.click(
         fn=generate_music_gradio,
         inputs=[prompt_input, bpm_input, temp_input],