Update app.py

app.py

@@ -14,10 +14,9 @@ load_dotenv()
 MUSICGEN_API_URL = os.getenv("MUSICGEN_API_URL")
 VEROVIO_API_URL = os.getenv("VEROVIO_API_URL")
 
-# Ensure output directory exists
 Path("output").mkdir(exist_ok=True)
 
-# === Tool 1: Convert WAV to MusicXML ===
+# === Tool 1 ===
 def wav_to_musicxml(wav_path: str, timestamp: str) -> str:
     output_dir = Path("output")
     for f in output_dir.glob("*_basic_pitch.mid"):
@@ -41,24 +40,20 @@ def wav_to_musicxml(wav_path: str, timestamp: str) -> str:
     score = converter.parse(midi_path)
     musicxml_path = output_dir / f"generated_{timestamp}.musicxml"
     score.write("musicxml", fp=musicxml_path)
-
     return str(musicxml_path)
 
-# === Tool 2: Render MusicXML to HTML via Verovio ===
+# === Tool 2 ===
 def render_musicxml_via_verovio_api(musicxml_path: str) -> str:
     if not VEROVIO_API_URL:
         return "❌ VEROVIO_API_URL is not configured"
-
     try:
         with open(musicxml_path, "rb") as f:
             files = {'file': f}
             response = requests.post(VEROVIO_API_URL, files=files)
     except Exception as e:
         return f"❌ Verovio API call failed: {e}"
-
     if response.status_code != 200:
         return f"❌ Verovio API error {response.status_code}: {response.text}"
-
     try:
         svg = response.json()["svg"]
         svg_b64 = base64.b64encode(svg.encode("utf-8")).decode("utf-8")
@@ -71,7 +66,7 @@ def render_musicxml_via_verovio_api(musicxml_path: str) -> str:
     except Exception as e:
         return f"⚠️ Failed to parse SVG: {e}"
 
-# === Tool 3: Melody + Prompt => Music + Score ===
+# === Tool 3 ===
 def generate_music_and_score(melody_file, prompt):
     if not MUSICGEN_API_URL:
         return None, "❌ MUSICGEN_API_URL is not configured. Please check your .env file"
@@ -101,40 +96,42 @@ def generate_music_and_score(melody_file, prompt):
     html = render_musicxml_via_verovio_api(musicxml_path)
     return wav_out_path, html
 
-# === MCP Tool App ===
+# === UI with Tabs ===
 with gr.Blocks(title="🎵 Vibe Jamming Tools") as demo:
-    
-    # Tool 1: WAV to MusicXML
-    gr.Interface(
-        fn=wav_to_musicxml,
-        inputs=[
-            gr.Audio(type="filepath", label="🎤 WAV Audio File"),
-            gr.Textbox(label="Timestamp (e.g. 20250609_123000)")
-        ],
-        outputs=gr.Textbox(label="📄 MusicXML File Path"),
-        title="Tool 1: Convert WAV to MusicXML"
-    ).render()
-
-    # Tool 2: Render MusicXML to HTML
-    gr.Interface(
-        fn=render_musicxml_via_verovio_api,
-        inputs=gr.Textbox(label="📄 MusicXML File Path"),
-        outputs=gr.HTML(label="🎼 Rendered Score Preview"),
-        title="Tool 2: Render MusicXML via Verovio"
-    ).render()
-
-    # Tool 3: Melody + Prompt → Music + Score
-    gr.Interface(
-        fn=generate_music_and_score,
-        inputs=[
-            gr.Audio(type="filepath", label="Upload your humming (.wav)"),
-            gr.Textbox(label="Describe your desired music style (prompt)")
-        ],
-        outputs=[
-            gr.Audio(type="filepath", label="🎵 Generated Music"),
-            gr.HTML(label="🎼 Score Preview (Rendered by Verovio)")
-        ],
-        title="Tool 3: Melody2Music+Score"
-    ).render()
+    gr.Markdown("## 🎹 Vibe Jamming - Multi-tool AI Music Studio")
+
+    with gr.Tabs():
+        with gr.TabItem("1️⃣ WAV ➜ MusicXML"):
+            gr.Interface(
+                fn=wav_to_musicxml,
+                inputs=[
+                    gr.Audio(type="filepath", label="🎤 WAV Audio File"),
+                    gr.Textbox(label="Timestamp (e.g. 20250609_123000)")
+                ],
+                outputs=gr.Textbox(label="📄 MusicXML File Path"),
+                live=False
+            ).render()
+
+        with gr.TabItem("2️⃣ MusicXML ➜ SVG Preview"):
+            gr.Interface(
+                fn=render_musicxml_via_verovio_api,
+                inputs=gr.Textbox(label="📄 MusicXML File Path"),
+                outputs=gr.HTML(label="🎼 Rendered Score"),
+                live=False
+            ).render()
+
+        with gr.TabItem("3️⃣ Melody2Music+Score"):
+            gr.Interface(
+                fn=generate_music_and_score,
+                inputs=[
+                    gr.Audio(type="filepath", label="Upload your humming (.wav)"),
+                    gr.Textbox(label="Describe your desired music style (prompt)")
+                ],
+                outputs=[
+                    gr.Audio(type="filepath", label="🎵 Generated Music"),
+                    gr.HTML(label="🎼 Score Preview (Rendered by Verovio)")
+                ],
+                live=False
+            ).render()
 
 demo.launch(mcp_server=True)