Create app.py

app.py

@@ -0,0 +1,64 @@
+import gradio as gr
+import tempfile
+import os
+import requests
+
+from basic_pitch.inference import predict
+from basic_pitch import ICASSP_2022_MODEL_PATH
+import pretty_midi
+
+def audio_to_midi(audio_url: str, instrument: str):
+    try:
+        with tempfile.TemporaryDirectory() as tmp:
+            audio_path = os.path.join(tmp, "input.wav")
+
+            # Download audio
+            r = requests.get(audio_url, timeout=30)
+            r.raise_for_status()
+            with open(audio_path, "wb") as f:
+                f.write(r.content)
+
+            # Run Basic Pitch
+            _, midi_data, _ = predict(
+                audio_path,
+                model_path=ICASSP_2022_MODEL_PATH
+            )
+
+            notes = []
+            for inst in midi_data.instruments:
+                for n in inst.notes:
+                    notes.append({
+                        "pitch": int(n.pitch),
+                        "startTime": round(n.start, 4),
+                        "duration": round(n.end - n.start, 4),
+                        "velocity": int(n.velocity)
+                    })
+
+            return {
+                "instrument": instrument,
+                "notes": notes
+            }
+
+    except Exception as e:
+        return {
+            "error": str(e),
+            "notes": []
+        }
+
+with gr.Blocks() as demo:
+    gr.Markdown("## Audio → MIDI API")
+    audio_input = gr.Textbox(label="Audio URL")
+    instr_input = gr.Textbox(label="Instrument")
+    output = gr.JSON()
+    
+    btn = gr.Button("Convert")
+    btn.click(fn=audio_to_midi, inputs=[audio_input, instr_input], outputs=output)
+
+if __name__ == "__main__":
+    demo.launch(
+        ssr_mode=False,
+        theme=gr.themes.Soft(
+            primary_hue="indigo",
+            secondary_hue="blue",
+        ),
+    )