Update app.py

app.py

@@ -11,6 +11,8 @@ import json
 import MIDI
 from midi_synthesizer import synthesis
 
+import matplotlib.pyplot as plt
+
 in_space = os.getenv("SYSTEM") == "spaces"
 
 def run(search_prompt, mid=None, progress=gr.Progress()):
@@ -40,12 +42,19 @@ def run(search_prompt, mid=None, progress=gr.Progress()):
         y.append(s[4])
         c.append(colors[s[3]])
 
-    plot = gr.ScatterPlot(x=x, y=y, color=c)
+    plt.close()
+    plt.figure(figsize=(14,5))
+    ax=plt.axes(title='Euterpe X Composition')
+    ax.set_facecolor('black')
+    
+    plt.scatter(x,y, c=c)
+    plt.xlabel("Time")
+    plt.ylabel("Pitch")
  
     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), plot
+    yield mdata, "output.mid", (44100, audio), plt
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
@@ -96,7 +105,7 @@ 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(label="output midi metadata")
-        output_plot = gr.ScatterPlot(label="output midi plot")
+        output_plot = gr.Plot(label="output midi plot")
         
         run_event = search_btn.click(run, [search_prompt],
                                   [output_midi_seq, output_midi, output_audio, output_plot])