Update app.py

app.py

@@ -139,60 +139,8 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     yield output, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio), [
         create_msg("visualizer_end", None)]
 
-
-def cancel_run(mid_seq):
-    if mid_seq is None:
-        return None, None, None
-    text_encoding = 'ISO-8859-1'
-    midi_data = TMIDIX.score2midi(mid_seq, text_encoding)
-
-    with open(f"Allegro-Music-Transformer-Music-Composition.mid", 'wb') as f:
-        f.write(midi_data)
-
-    audio = synthesis(TMIDIX.score2opus(mid_seq), 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2')
-
-    yield "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio), [
-        create_msg("visualizer_end", None)]
-
-
 # =================================================================================================
 
-def load_javascript(dir="javascript"):
-    scripts_list = glob.glob(f"{dir}/*.js")
-    javascript = ""
-    for path in scripts_list:
-        with open(path, "r", encoding="utf8") as jsfile:
-            javascript += f"\n<!-- {path} --><script>{jsfile.read()}</script>"
-    template_response_ori = gr.routes.templates.TemplateResponse
-
-    def template_response(*args, **kwargs):
-        res = template_response_ori(*args, **kwargs)
-        res.body = res.body.replace(
-            b'</head>', f'{javascript}</head>'.encode("utf8"))
-        res.init_headers()
-        return res
-
-    gr.routes.templates.TemplateResponse = template_response
-
-
-class JSMsgReceiver(gr.HTML):
-
-    def __init__(self, **kwargs):
-        super().__init__(elem_id="msg_receiver", visible=False, **kwargs)
-
-    def postprocess(self, y):
-        if y:
-            y = f"<p>{json.dumps(y)}</p>"
-        return super().postprocess(y)
-
-    def get_block_name(self) -> str:
-        return "html"
-
-
-def create_msg(name, data):
-    return {"name": name, "data": data}
-
-
 if __name__ == "__main__":
     
     PDT = timezone('US/Pacific')
@@ -205,6 +153,8 @@ if __name__ == "__main__":
     parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
     parser.add_argument("--port", type=int, default=7860, help="gradio server port")
     opt = parser.parse_args()
+
+    soundfonts = ["SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"]
     
     print('Loading model...')
 
@@ -250,21 +200,22 @@ if __name__ == "__main__":
             "(https://colab.research.google.com/github/asigalov61/Allegro-Music-Transformer/blob/main/Allegro_Music_Transformer_Composer.ipynb)"
             " for faster execution and endless generation"
         )
-        js_msg = JSMsgReceiver()
-        input_drums = gr.Checkbox(label="Add Drums", value=False, info="Add drums to the composition")
-        input_instrument = gr.Radio(
-            ["Piano", "Guitar", "Bass", "Violin", "Cello", "Harp", "Trumpet", "Sax", "Flute", "Choir", "Organ"],
-            value="Piano", label="Lead Instrument Controls", info="Desired lead instrument")
+        gr.Markdown("## Upload your MIDI")
+        
+        input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"], type="filepath")
         input_num_tokens = gr.Slider(16, 512, value=256, label="Number of Tokens", info="Number of tokens to generate")
+        
         run_btn = gr.Button("generate", variant="primary")
         interrupt_btn = gr.Button("interrupt")
 
-        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, [input_num_tokens, input_drums, input_instrument],
-                                  [output_midi_seq, output_midi, output_audio, js_msg])
-        interrupt_btn.click(cancel_run, output_midi_seq, [output_midi, output_audio, js_msg],
-                            cancels=run_event, queue=False)
+        output_midi_title = gr.Textbox(label="Output MIDI title")
+        output_midi_summary = gr.Textbox(label="Output MIDI summary")
+        output_audio = gr.Audio(label="Output MIDI audio", format="wav", elem_id="midi_audio")
+        output_plot = gr.Plot(label="Output MIDI score plot")
+        output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
+
+        
+        run_event = run_btn.click(GenerateMIDI, [input_midi, input_num_tokens],
+                                  [output_midi_title, output_midi_summary, output_audio, output_plot, output_midi])
+        
         app.queue(concurrency_count=1).launch(server_port=opt.port, share=opt.share, inbrowser=True)