Update app.py

app.py

@@ -13,15 +13,17 @@ from midi_synthesizer import synthesis
 
 in_space = os.getenv("SYSTEM") == "spaces"
 
-def run(search_prompt, mid=None):
-    mid_seq = []
+def run(search_prompt, mid=None, progress=gr.Progress()):
+    
 
     if mid == None:
 
-        for m in meta_data:
+        for m in progress.tqdm(meta_data):
+            mid_seq = []
             mid_seq.extend(m[1][17:])
             mid_seq_ticks = m[1][16][1]
-            break
+            mdata = m[1][:17]
+            # break
        
     elif mid is not None:
         mid_seq = MIDI.midi2score(mid)
@@ -29,7 +31,7 @@ def run(search_prompt, mid=None):
     with open(f"output.mid", 'wb') as f:
         f.write(MIDI.score2midi([mid_seq_ticks, mid_seq]))
     audio = synthesis(MIDI.score2opus([mid_seq_ticks, mid_seq]), soundfont_path)
-    yield mid_seq, "output.mid", (44100, audio)
+    yield mdata, "output.mid", (44100, audio)
 
 
 def cancel_run(mid_seq):
@@ -87,9 +89,11 @@ if __name__ == "__main__":
             
         search_btn = gr.Button("search", variant="primary")
         stop_btn = gr.Button("stop and output")
-        output_midi_seq = gr.Textbox()
+        
         output_audio = gr.Audio(label="output audio", format="mp3", elem_id="midi_audio")
         output_midi = gr.File(label="output midi", file_types=[".mid"])
+        output_midi_seq = gr.Textbox()
+        
         run_event = search_btn.click(run, [search_prompt],
                                   [output_midi_seq, output_midi, output_audio])
         stop_btn.click(cancel_run, output_midi_seq, [output_midi, output_audio], cancels=run_event, queue=False)