Update app.py

app.py

@@ -39,11 +39,13 @@ def run(search_prompt, mid=None, progress=gr.Progress()):
         x.append(s[1] / mid_seq_ticks)
         y.append(s[4])
         c.append(colors[s[3]])
+
+    plot = gr.ScatterPlot(x=x, y=y, color=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), x, y, c
+    yield mdata, "output.mid", (44100, audio), plot
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
@@ -98,5 +100,6 @@ if __name__ == "__main__":
         
         run_event = search_btn.click(run, [search_prompt],
                                   [output_midi_seq, output_midi, output_audio, output_plot])
+        gr.show(output_plot)
         
     app.queue(1).launch(server_port=opt.port, share=opt.share, inbrowser=True)