Update app.py

app.py

@@ -93,7 +93,7 @@ def generate_MIDI_images(num_images):
     
 # =================================================================================================
 
-def Generate_POP_Medley(input_num_medley_comps):
+def Generate_POP_Medley(input_num_medley_comps, input_melody_patch):
     
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -101,8 +101,11 @@ def Generate_POP_Medley(input_num_medley_comps):
     print('=' * 70)
     
     print('Req number of medley compositions:', input_num_medley_comps)
+    print('Req melody MIDI patch number:', input_melody_patch)
     print('=' * 70)
     
+    #===============================================================================
+    # MIDI Images generation function
     #===============================================================================
 
     imgs_array = generate_MIDI_images(input_num_medley_comps)
@@ -120,6 +123,9 @@ def Generate_POP_Medley(input_num_medley_comps):
     
         score = TMIDIX.binary_matrix_to_original_escore_notes(bmatrix)
 
+        if input_melody_patch > -1:
+            score = TMIDIX.add_melody_to_enhanced_score_notes(score, melody_patch=input_melody_patch)
+
         medley_compositions_escores.append(score)
 
     print('Done!')
@@ -212,6 +218,7 @@ if __name__ == "__main__":
         gr.Markdown("Choose medley settings")
         
         input_num_medley_comps = gr.Slider(1, 12, value=8, step=1, label="Number of medley compositions")
+        input_melody_patch = gr.Slider(-1, 127, value=40, step=1, label="Medley melody MIDI patch number")
 
         run_btn = gr.Button("Generate POP Medley", variant="primary")
 
@@ -223,7 +230,7 @@ if __name__ == "__main__":
         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(Generate_POP_Medley, [input_num_medley_comps],
+        run_event = run_btn.click(Generate_POP_Medley, [input_num_medley_comps, input_melody_patch],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         app.queue().launch()