Update app.py

app.py

@@ -146,7 +146,7 @@ def GenerateMIDI(progress=gr.Progress()):
           
     while itrack < len(opus):
         for event in opus[itrack]:
-            if event[0] == 'note': 
+            if event[0] == 'note_on' or event[0] == 'note_off': 
                 output1.append(o)
         itrack += 1
         
@@ -156,16 +156,6 @@ def GenerateMIDI(progress=gr.Progress()):
         
 #=================================================================================================
 
-def cancel_run(output_midi_seq):
-    if output_midi_seq is None:
-        return None, None
-    with open(f"Allegro-Music-Transformer-Music-Composition.mid", 'wb') as f:
-        f.write(TMIDIX.score2midi(output_midi_seq))
-    audio = synthesis(TMIDIX.score2opus(output_midi_seq), 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2')
-    return "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio)
-
-#=================================================================================================
-
 if __name__ == "__main__":
 
     parser = argparse.ArgumentParser()
@@ -189,13 +179,11 @@ if __name__ == "__main__":
                         )
                 
         run_btn = gr.Button("generate", variant="primary")
-        stop_btn = gr.Button("stop and output")
 
         output_midi_seq = gr.Variable()
         output_midi_visualizer = gr.HTML(elem_id="midi_visualizer_container")
         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_midi, output_audio])
-        stop_btn.click(cancel_run, output_midi_seq, [output_midi, output_audio], cancels=run_event, queue=False)
         
         app.queue(2).launch(server_port=opt.port, share=opt.share, inbrowser=True)