Update app.py

app.py

@@ -1,4 +1,4 @@
-# MusicGen + Gradio + GPT Demo App (CPU-Optimized for Hugging Face Spaces)
+# MusicGen + Gradio + GPT Demo App (CPU-Optimized with MCP Server)
 
 import gradio as gr
 import os
@@ -33,7 +33,6 @@ def refine_prompt(user_input):
 # Generate music (shorter tokens for CPU speed)
 def generate_music(prompt, max_new_tokens: int = 128):
     inputs = processor(text=[prompt], return_tensors="pt").to(device)
-    # Warning: Generation on CPU may be slow
     audio_values = model.generate(**inputs, max_new_tokens=max_new_tokens)
     sampling_rate = model.config.audio_encoder.sampling_rate
     audio = audio_values[0].cpu().numpy()
@@ -42,10 +41,14 @@ def generate_music(prompt, max_new_tokens: int = 128):
     audio = audio / np.max(np.abs(audio))
     audio = audio.astype(np.float32)
 
-    # Save as .wav file (in /tmp for Spaces)
+    # Prepare int16 version and ensure 1D for WAV
     int_audio = (audio * 32767).astype(np.int16)
-    scipy.io.wavfile.write("/tmp/output.wav", sampling_rate, int_audio)
+    int_audio = np.squeeze(int_audio)
+    if int_audio.ndim > 1:
+        int_audio = int_audio[:, 0]
 
+    # Save as .wav file (in /tmp for Spaces)
+    scipy.io.wavfile.write("/tmp/output.wav", sampling_rate, int_audio)
     return sampling_rate, audio
 
 # Combined Gradio function
@@ -55,7 +58,8 @@ def main(user_input, max_new_tokens):
     return detailed_prompt, (sampling_rate, audio), "/tmp/output.wav"
 
 # Build Gradio UI
-with gr.Blocks() as demo:
+demo = gr.Blocks()
+with demo:
     gr.Markdown("""# 🎵 AI Music Generator  
 Enter a music idea or mood and get a short AI-generated track. (CPU mode)""")
 
@@ -73,5 +77,9 @@ Enter a music idea or mood and get a short AI-generated track. (CPU mode)""")
         outputs=[refined_output, audio_output, download_wav]
     )
 
-# Launch in SSR mode with queue enabled
-demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+# Launch with Gradio MCP Server
+from gradio.mcp_server import MCPServer
+
+if __name__ == "__main__":
+    server = MCPServer(demo, host="0.0.0.0", port=7860)
+    server.run()