Update app.py

app.py

@@ -472,36 +472,6 @@ def add_arpeggio_sequence(tokenizer, mid_seq, sequence, pattern):
 
 
  
- 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)
-
- 
  with gr.Tabs():
  with gr.TabItem("custom prompt") as tab1:
  input_instruments = gr.Dropdown(label="🪗instruments (auto if empty)", choices=list(patch2number.keys()),
@@ -565,6 +535,41 @@ def add_arpeggio_sequence(tokenizer, mid_seq, sequence, pattern):
  )
  undo_btn = gr.Button("undo the last continuation")
 
+
+
+ 
+ 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)
+
+
+
+ 
  tab1.select(lambda: 0, None, tab_select, queue=False)
  tab2.select(lambda: 1, None, tab_select, queue=False)
  tab3.select(lambda: 2, None, tab_select, queue=False)