Update app.py

app.py

@@ -7,16 +7,6 @@ import pretty_midi
 import subprocess
 import os
 from openai import OpenAI
-# Audio support flag
-try:
-    import pygame
-    pygame.mixer.init()
-    AUDIO_AVAILABLE = True
-except Exception as e:
-    print(f"[WARN] pygame audio init failed: {e}")
-    AUDIO_AVAILABLE = False
-
-midi_path_global = None
 
 # === LLM APIs ===
 def query_llm(prompt, model_name=None):
@@ -25,22 +15,24 @@ def query_llm(prompt, model_name=None):
         response = requests.post("http://localhost:11434/api/generate", json={"model": model_name, "prompt": prompt, "stream": False})
         return response.json().get("response", "")
     else:
-        # Replace or load from environment
         client = OpenAI(
-    base_url="https://api.studio.nebius.com/v1/",
-    api_key=os.environ.get("NEBIUS_API_KEY")
-)
+            base_url="https://api.studio.nebius.com/v1/",
+            api_key=os.environ.get("NEBIUS_API_KEY")
+        )
         response = client.chat.completions.create(
-    model="Qwen/Qwen3-30B-A3B",
-    max_tokens=512,
-    temperature=0.6,
-    top_p=0.9,
-    extra_body={
-        "top_k": 50
-    },
-    messages=[]
-)
-    return response["choices"][0]["message"]["content"]
+            model="Qwen/Qwen3-30B-A3B",
+            messages=[
+                {
+                    "role": "system",
+                    "content": "You are a helpful assistant."
+                },
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ]
+        )
+        return response.choices[0].message.content
 
 # === Step 1: Parse intent ===
 def get_intent_from_prompt(prompt, model_name):
@@ -117,46 +109,17 @@ def midi_from_plan(melody, tempo):
     midi.instruments.append(instrument)
     return midi
 
+# === Gradio function ===
 def generate_midi_from_prompt(prompt, model_name):
-    global midi_path_global
     intent = get_intent_from_prompt(prompt, model_name)
     melody = get_melody_from_intent(intent, model_name)
     midi = midi_from_plan(melody, intent.get("tempo", 120))
 
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mid") as tmp:
         midi.write(tmp.name)
-        midi_path_global = tmp.name
-    return midi_path_global
-
-def play_audio():
-    if not AUDIO_AVAILABLE:
-        return "Audio playback not available."
-    if midi_path_global is None:
-        return "No MIDI file loaded yet."
-    try:
-        pygame.mixer.music.load(midi_path_global)
-        pygame.mixer.music.play()
-        return "Playing audio..."
-    except Exception as e:
-        return f"Error playing audio: {e}"
-
-def pause_audio():
-    if not AUDIO_AVAILABLE:
-        return "Audio playback not available."
-    pygame.mixer.music.pause()
-    return "Paused."
-
-def unpause_audio():
-    if not AUDIO_AVAILABLE:
-        return "Audio playback not available."
-    pygame.mixer.music.unpause()
-    return "Unpaused."
+        midi_path = tmp.name
 
-def stop_audio():
-    if not AUDIO_AVAILABLE:
-        return "Audio playback not available."
-    pygame.mixer.music.stop()
-    return "Stopped."
+    return midi_path
 
 # === Get Ollama models ===
 def get_ollama_models():
@@ -169,7 +132,6 @@ def get_ollama_models():
 
 # === Gradio UI ===
 models = get_ollama_models()
-
 demo = gr.Interface(
     fn=generate_midi_from_prompt,
     inputs=[
@@ -180,19 +142,7 @@ demo = gr.Interface(
         gr.File(label="🎵 Download MIDI File")
     ],
     title="🎼 Music Command Prompt (MCP Agent)",
-    description="Describe your music idea and download a generated MIDI file. Choose from local or OpenAI LLMs."
+    description="Describe your music idea and download a generated MIDI file. Choose from local or Nebius/OpenAI LLMs."
 )
 
-with gr.Row():
-    with gr.Column(scale=1):
-        play_btn = gr.Button("▶ Play")
-        pause_btn = gr.Button("⏸ Pause")
-        unpause_btn = gr.Button("▶ Resume")
-        stop_btn = gr.Button("⏹ Stop")
-    with gr.Column(scale=3):
-        status = gr.Textbox(label="Audio Status", interactive=False)
-
-play_btn.click(play_audio, None, status)
-pause_btn.click(pause_audio, None, status)
-unpause_btn.click(unpause_audio, None, status)
-stop_btn.click(stop_audio, None, status)
+demo.launch(mcp_server=True)