Update app_onnx.py

app_onnx.py

@@ -494,17 +494,7 @@ if __name__ == "__main__":
  load_javascript()
  app = gr.Blocks(theme=gr.themes.Soft())
  with app:
- gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Midi Composer</h1>")
- gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.midi-composer&style=flat)\n\n"
- "Midi event transformer for symbolic music generation\n\n"
- "Demo for [SkyTNT/midi-model](https://github.com/SkyTNT/midi-model)\n\n"
- "[Open In Colab]"
- "(https://colab.research.google.com/github/SkyTNT/midi-model/blob/main/demo.ipynb)"
- " or [download windows app](https://github.com/SkyTNT/midi-model/releases)"
- " for unlimited generation\n\n"
- "**Update v1.3**: MIDITokenizerV2 and new MidiVisualizer\n\n"
- "The current **best** model: generic pretrain model (tv2o-medium) by skytnt"
- )
+
  js_msg = gr.Textbox(elem_id="msg_receiver", visible=False)
  js_msg.change(None, [js_msg], [], js="""
  (msg_json) =>{
@@ -595,12 +585,23 @@ if __name__ == "__main__":
  audio_outputs = []
  with gr.Tabs(elem_id="output_tabs"):
  for i in range(OUTPUT_BATCH_SIZE):
+ 
+ with gr.Row():
+ arpeggio_intro = gr.Button("🎵 Intro Arpeggio", variant="primary")
+ arpeggio_verse = gr.Button("🎸 Verse Arpeggio", variant="primary")
+ arpeggio_chorus = gr.Button("🎹 Chorus Arpeggio", variant="primary")
+ arpeggio_outro = gr.Button("🎷 Outro Arpeggio", variant="primary")
+ 
  with gr.TabItem(f"output {i + 1}") as tab1:
  output_midi_visualizer = gr.HTML(elem_id=f"midi_visualizer_container_{i}")
  output_audio = gr.Audio(label="output audio", format="mp3", elem_id=f"midi_audio_{i}")
  output_midi = gr.File(label="output midi", file_types=[".mid"])
+ 
  midi_outputs.append(output_midi)
  audio_outputs.append(output_audio)
+
+
+ 
  run_event = run_btn.click(run, [input_model, tab_select, output_midi_seq, output_continuation_state,
  input_continuation_select, input_instruments, input_drum_kit, input_bpm,
  input_time_sig, input_key_sig, input_midi, input_midi_events,
@@ -620,6 +621,40 @@ if __name__ == "__main__":
  queue=False)
  # stop_btn.click(None, [], [], cancels=run_event,
  # queue=False)
+
+
+
+ def add_intro_arpeggio(model_name, mid_seq):
+ tokenizer = models[model_name].tokenizer
+ sequence = ['C', 'D', 'Am', 'G']
+ pattern = [0, 1, 2, 1] # Root, Third, Fifth, Third
+ return add_arpeggio_sequence(tokenizer, mid_seq, sequence, pattern)
+
+ def add_verse_arpeggio(model_name, mid_seq):
+ tokenizer = models[model_name].tokenizer
+ sequence = ['D', 'C', 'Am', 'G']
+ pattern = [0, 2, 1, 2] # Root, Fifth, Third, Fifth
+ return add_arpeggio_sequence(tokenizer, mid_seq, sequence, pattern)
+
+ def add_chorus_arpeggio(model_name, mid_seq):
+ tokenizer = models[model_name].tokenizer
+ sequence = ['G', 'D', 'Am', 'C']
+ pattern = [0, 1, 2, 1, 0, 2] # Root, Third, Fifth, Third, Root, Fifth
+ return add_arpeggio_sequence(tokenizer, mid_seq, sequence, pattern)
+
+ def add_outro_arpeggio(model_name, mid_seq):
+ tokenizer = models[model_name].tokenizer
+ sequence = ['Am', 'G', 'D', 'C']
+ pattern = [2, 1, 0, 1] # Fifth, Third, Root, Third
+ return add_arpeggio_sequence(tokenizer, mid_seq, sequence, pattern)
+
+ arpeggio_intro.click(add_intro_arpeggio, [input_model, output_midi_seq], output_midi_seq)
+ arpeggio_verse.click(add_verse_arpeggio, [input_model, output_midi_seq], output_midi_seq)
+ arpeggio_chorus.click(add_chorus_arpeggio, [input_model, output_midi_seq], output_midi_seq)
+ arpeggio_outro.click(add_outro_arpeggio, [input_model, output_midi_seq], output_midi_seq)
+
+
+ 
  undo_btn.click(undo_continuation, [input_model, output_midi_seq, output_continuation_state],
  [output_midi_seq, output_continuation_state, js_msg], queue=False)
  app.queue().launch(server_port=opt.port, share=opt.share, inbrowser=True, ssr_mode=False)