Update app.py

app.py

@@ -24,11 +24,28 @@ def run(search_prompt, mid=None, progress=gr.Progress()):
        
     elif mid is not None:
         mid_seq = MIDI.midi2score(mid)
+
+
+    x = []
+    y = []
+    c = []
+    
+    colors = ['red', 'yellow', 'green', 'cyan',
+            'blue', 'pink', 'orange', 'purple',
+            'gray', 'white', 'gold', 'silver',
+            'lightgreen', 'indigo', 'maroon', 'turquoise']
+    
+    for s in [m for m in mid_seq if m[0] == 'note']:
+        x.append(s[1] / mid_seq_ticks)
+        y.append(s[4])
+        c.append(colors[s[3]])
+
+    plot = [x, y, c]
  
     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 mdata, "output.mid", (44100, audio)
+    yield mdata, "output.mid", (44100, audio), plot
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
@@ -78,9 +95,10 @@ if __name__ == "__main__":
         
         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()
+        output_midi_seq = gr.Textbox(label="output midi metadata")
+        output_plot = gr.ScatterPlot(label="output midi plot")
         
         run_event = search_btn.click(run, [search_prompt],
-                                  [output_midi_seq, output_midi, output_audio])
+                                  [output_midi_seq, output_midi, output_audio, output_plot])
         
     app.queue(1).launch(server_port=opt.port, share=opt.share, inbrowser=True)