Update app.py

app.py

@@ -13,6 +13,8 @@ import tqdm
 from midi_synthesizer import synthesis
 import TMIDIX
 
+import matplotlib.pyplot as plt
+
 in_space = os.getenv("SYSTEM") == "spaces"
       
 #=================================================================================================
@@ -151,8 +153,6 @@ def GenerateMIDI(progress=gr.Progress()):
         itrack += 1
         
     audio = synthesis([500, output1], 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2')
-
-    midi_plot = []
     
     x = []
     y =[]
@@ -165,9 +165,15 @@ def GenerateMIDI(progress=gr.Progress()):
       y.append(s[4])
       c.append(colors[s[3]])
 
-    midi_plot.append([x, y, c])
+    fig = plt.figure(figsize=(14,5))
+    ax=plt.axes(title=fname)
+    ax.set_facecolor('black')
+    
+    plt.scatter(x,y, c=c)
+    plt.xlabel("Time")
+    plt.ylabel("Pitch")
 
-    yield [500, output1], midi_plot, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
+    yield [500, output1], fig, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
         
 #=================================================================================================
 
@@ -196,7 +202,7 @@ if __name__ == "__main__":
         run_btn = gr.Button("generate", variant="primary")
 
         output_midi_seq = gr.Variable()
-        output_plot = gr.ScatterPlot(label="output plot")
+        output_plot = gr.Plot(label="output plot")
         output_audio = gr.Audio(label="output audio", format="mp3", elem_id="midi_audio")
         output_midi = gr.File(label="output midi", file_types=[".mid"])
         run_event = run_btn.click(GenerateMIDI, [], [output_midi_seq, output_plot, output_midi, output_audio])