Update app.py

app.py

@@ -107,7 +107,7 @@ print('=' * 70)
 
 #==================================================================================
 
-def load_midi(input_midi, melody_patch=-1):
+def load_midi(input_midi, melody_patch=-1, use_nth_note):
 
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi)
 
@@ -128,7 +128,7 @@ def load_midi(input_midi, melody_patch=-1):
         else:
             zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
     
-    cscore = TMIDIX.chordify_score([1000, zscore])
+    cscore = TMIDIX.chordify_score([1000, zscore])[:use_nth_note]
     
     score = []
     
@@ -160,6 +160,7 @@ def load_midi(input_midi, melody_patch=-1):
 def Generate_Accompaniment(input_midi, 
                            generation_type,
                            melody_patch,
+                           use_nth_note,
                            model_temperature
                           ):
 
@@ -252,13 +253,14 @@ def Generate_Accompaniment(input_midi,
     print('Input MIDI file name:', fn)
     print('Generation type:', generation_type)
     print('Source melody patch:', melody_patch)
+    print('Use nth melody note:', use_nth_note)
     print('Model temperature:', model_temperature)
    
     print('=' * 70)
 
     #==================================================================
 
-    score, score_list = load_midi(input_midi.name)
+    score, score_list = load_midi(input_midi.name, melody_patch, use_nth_note)
 
     print('Sample score events', score[:12])
 
@@ -453,6 +455,7 @@ with gr.Blocks() as demo:
     
     generation_type = gr.Radio(["Guided", "Freestyle"], value="Guided", label="Generation type")
     melody_patch = gr.Slider(-1, 127, value=-1, step=1, label="Source melody MIDI patch")
+    use_nth_note = gr.Slider(1, 8, value=1, step=1, label="Use each nth melody note")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     
     generate_btn = gr.Button("Generate", variant="primary")
@@ -467,6 +470,7 @@ with gr.Blocks() as demo:
                        [input_midi, 
                         generation_type,
                         melody_patch,
+                        use_nth_note,
                         model_temperature
                        ], 
                        [output_audio,
@@ -476,13 +480,14 @@ with gr.Blocks() as demo:
                       )
 
     gr.Examples(
-                [["USSR-National-Anthem-Seed-Melody.mid", "Guided", -1, 0.9],
-                 ["Hotel-California-Seed-Melody.mid", "Guided", -1, 0.9],
-                 ["Sparks-Fly-Seed-Melody.mid", "Guided", -1, 0.9]
+                [["USSR-National-Anthem-Seed-Melody.mid", "Guided", -1, 1, 0.9],
+                 ["Hotel-California-Seed-Melody.mid", "Guided", -1, 1, 0.9],
+                 ["Sparks-Fly-Seed-Melody.mid", "Guided", -1, 1, 0.9]
                 ],
                 [input_midi, 
                  generation_type,
                  melody_patch,
+                 use_nth_note,
                  model_temperature
                 ],
                 [output_audio,