Update app.py

app.py

@@ -1,48 +1,64 @@
-import gradio as gr
+import streamlit as st
 import numpy as np
 import pretty_midi
 from accompaniment_generator.generator.base import Generator
 import os
 import uuid
-from midi2audio import FluidSynth
 import time
+from midi2audio import FluidSynth
 from scipy.io import wavfile
 
 
 def inference(audio, num_epoch):
     generator = Generator()
-    input_midi_data = pretty_midi.PrettyMIDI(audio.name)
-    output_midi_data = generator(audio.name, num_epoch=int(num_epoch))
+    output_midi_data = generator(audio, num_epoch=int(num_epoch))
     name = uuid.uuid4()
     output_midi_data.write(f'{name}.mid')
     fs = FluidSynth("font.sf2")
     fs.midi_to_audio(f'{name}.mid', f'{name}.wav')
-    fs.midi_to_audio(audio.name, f'{name}-init.wav')
-    time.sleep(2)
-    print([
-            wavfile.read(f'{name}-init.wav'),
-            wavfile.read(f'{name}.wav'),
-            ])
-    return [
-            wavfile.read(f'{name}-init.wav'),
-            wavfile.read(f'{name}.wav'),
-            ]
-
-title = "Accompaniment Generator"
-description = "Gradio demo for MIDI-DDSP: Detailed Control of Musical Performance via Hierarchical Modeling. To use it, simply upload your midi file, or click one of the examples to load them. Read more at the links below."
+    fs.midi_to_audio(audio, f'{name}-init.wav')
+    # time.sleep(2)
+    print([f'{name}-init.wav', f'{name}.wav'])
+    return f'{name}-init.wav', f'{name}.wav'
+
+
+st.title("Accompaniment Generator")
+st.description = "Gradio demo for MIDI-DDSP: Detailed Control of Musical Performance via Hierarchical Modeling. To use it, simply upload your midi file, or click one of the examples to load them. Read more at the links below."
 
 article = "<p style='text-align: center'>" \
           "<a href='https://github.com/AlekseyKorshuk/accompaniment-generator' target='_blank'>Github Repo</a>" \
           "</p>"
 
-examples = [['barbiegirl_mono.mid', 10]]
-
-gr.Interface(
-    inference,
-    [gr.inputs.File(type="file", label="Input"), gr.inputs.Number(label="Number of epoch", default=10)],
-    [gr.outputs.Audio(type="auto", label="Before"), gr.outputs.Audio(type="auto", label="After")],
-    title=title,
-    description=description,
-    article=article,
-    examples=examples,
-).launch(debug=True)
+from os import listdir
+from os.path import isfile, join
+
+onlyfiles = [f for f in listdir("./examples") if isfile(join("./examples", f))]
+
+model_name = st.selectbox(
+    'Select example MIDI file:',
+    onlyfiles
+)
+
+uploaded_file = st.file_uploader(
+    'Upload MIDI file:'
+)
+
+num_epoch = st.number_input("Number of epochs:",
+                            min_value=1,
+                            max_value=1000,
+                            step=1,
+                            value=10,
+                            )
+
+generate_image_button = st.button("Generate")
+
+if generate_image_button:
+    input_file = f"./examples/{model_name}"
+    if uploaded_file is not None:
+        input_file = uploaded_file.name
+    with st.spinner(text=f"Generating, this may take some time..."):
+        before, after = inference(input_file, num_epoch)
+    st.markdown("Before:")
+    st.audio(before)
+    st.markdown("After:")
+    st.audio(after)