Update app.py

app.py

@@ -7,14 +7,14 @@ import nest_asyncio
 from google import genai
 from google.genai import types
 
-# Enable nested asyncio (required for Gradio + asyncio)
+# Enable nested async for environments like Gradio/Jupyter
 nest_asyncio.apply()
 
-# Configure Gemini API
+# Load Google API key from environment
 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
+# Helper to save PCM as .wav
 def save_wave_file(filename, pcm, channels=1, rate=24000, sample_width=2):
     with wave.open(filename, "wb") as wf:
         wf.setnchannels(channels)
@@ -22,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 function
+# Async music generation logic
 async def generate_music(prompt, bpm, temperature):
     audio_chunks = []
 
@@ -46,37 +46,37 @@ async def generate_music(prompt, bpm, temperature):
             )
 
             await session.play()
-            await asyncio.sleep(5)  # Streaming duration
+            await asyncio.sleep(5)  # Generation duration
             await session.pause()
 
         all_pcm = b"".join(audio_chunks)
 
-        # Save WAV file
+        # Save to WAV file for download
         output_path = "generated_music.wav"
         save_wave_file(output_path, all_pcm)
 
-        # Convert PCM to numpy array for audio playback
+        # Return numpy array for playback (gr.Audio)
         audio_np = np.frombuffer(all_pcm, dtype=np.int16)
-        return (24000, audio_np), output_path, "Music generated successfully!"
+        return (24000, audio_np), output_path, "✅ Music generated successfully!"
 
     except Exception as e:
         return None, None, f"❌ Error: {str(e)}"
 
-# Wrapper for Gradio
+# Sync wrapper for Gradio (uses event loop)
 def generate_music_gradio(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:
+with gr.Blocks(title="🎵 Gemini Lyria Music Generator") as demo:
     gr.Markdown("## 🎶 Gemini Lyria Music Generator")
 
     with gr.Group():
-        gr.Markdown("### 🎛 Input")
+        gr.Markdown("### 🎛 Prompt & Controls")
         with gr.Row():
             prompt_input = gr.Textbox(
                 label="Music Style / Prompt",
-                placeholder="e.g., ambient synth, minimal techno, classical piano"
+                placeholder="e.g., ambient synth, classical piano, lo-fi chill"
             )
         with gr.Row():
             bpm_input = gr.Slider(label="BPM", minimum=60, maximum=180, value=90)
@@ -84,21 +84,21 @@ with gr.Blocks(title="Gemini Lyria Music Generator") as demo:
         generate_btn = gr.Button("🎧 Generate Music")
 
     with gr.Group():
-        gr.Markdown("### 🎵 Output")
+        gr.Markdown("### 🎼 Output")
         with gr.Row():
             output_audio = gr.Audio(label="Generated Audio", type="numpy")
             download_file = gr.File(label="Download WAV")
         status_output = gr.Textbox(label="Status", interactive=False)
 
     with gr.Group():
-        gr.Markdown("### 🔁 Examples")
+        gr.Markdown("### 🔁 Try These Examples")
         examples = gr.Examples(
             examples=[
-                ["minimal techno", 125, 1.0],
-                ["classical piano in a rainy mood", 70, 0.9],
-                ["ambient space drone", 90, 1.5],
-                ["lo-fi chill beats", 80, 1.0],
-                ["orchestral epic music", 110, 1.2],
+                ["ambient synth in a forest", 100, 1.2],
+                ["classical piano with raindrops", 70, 0.8],
+                ["lo-fi chillhop vibe", 85, 1.0],
+                ["orchestral film music", 110, 1.5],
+                ["spacey techno beats", 125, 1.3],
             ],
             inputs=[prompt_input, bpm_input, temp_input]
         )