add transcribe button

app.py

@@ -2,7 +2,7 @@ import os
 
 import gradio as gr
 
-from gradio_components.prediction import predict
+from gradio_components.prediction import predict, transcribe
 
 theme = gr.themes.Glass(
 	primary_hue="fuchsia",
@@ -136,13 +136,17 @@ def UI():
 							)
 					with gr.Row():
 						submit = gr.Button("Generate Music")
-						output = gr.Audio("listen to the generated music", type="filepath")
+						output_audio = gr.Audio("listen to the generated music", type="filepath")
+					with gr.Row():
+						transcribe_button = gr.Button("Transcribe")
+						d = gr.DownloadButton("Download the file", visible=False)
+						transcribe_button.click(transcribe, inputs=[output_audio], outputs=d)
 
 			submit.click(
 				fn=predict,
 				inputs=[model_path, prompt, melody, duration, topk, topp, temperature,
 				        sample_rate],
-				outputs=output
+				outputs=output_audio
 				)
 
 		gr.Examples(
@@ -204,7 +208,7 @@ def UI():
 				],
 			inputs=[melody, difficulty, sample_rate, duration],
 			label="Audio Examples",
-			outputs=[output],
+			outputs=[output_audio],
 			# cache_examples=True,
 			)
 	demo.queue().launch()

gradio_components/prediction.py

@@ -9,6 +9,9 @@ from audiocraft.models import MusicGen
 from tempfile import NamedTemporaryFile
 from pathlib import Path
 from transformers import AutoModelForSeq2SeqLM
+import basic_pitch
+import basic_pitch.inference
+from basic_pitch import ICASSP_2022_MODEL_PATH
 
 
 def load_model(version='facebook/musicgen-melody'):
@@ -103,4 +106,25 @@ def predict(model_path, text, melody, duration, topk, topp, temperature, target_
         top_p=topp,
         temperature=temperature,
         gradio_progress=progress)
-    return wavs[0]
+    return wavs[0]
+
+
+def transcribe(audio_path):
+    # model_output, midi_data, note_events = predict("generated_0.wav")
+    model_output, midi_data, note_events = basic_pitch.inference.predict(
+        audio_path=audio_path,
+        model_or_model_path=ICASSP_2022_MODEL_PATH,
+        )
+
+    with NamedTemporaryFile("wb", suffix=".mid", delete=False) as file:
+        try:
+            midi_data.write(file)
+            print(f"midi file saved to {file.name}")
+        except Exception as e:
+            print(f"Error while writing midi file: {e}")
+            raise e
+
+    return gr.DownloadButton(
+        value=file.name,
+        label=f"Download MIDI file {file.name}",
+        visible=True)