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)