Update app.py

app.py

@@ -28,7 +28,7 @@ in_space = os.getenv("SYSTEM") == "spaces"
 
 #==========================================================================================================
 
-def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, melody_patch):
+def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, custom_render_patch, melody_patch):
     
     print('*' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -49,6 +49,7 @@ def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, mel
     print('Render type:', render_type)
     print('Soudnfont bank', soundfont_bank)
     print('Audio render sample rate', render_sample_rate)
+    print('Custom MIDI render patch', custom_render_patch)
     print('Melody patch', melody_patch)
     
     print('=' * 70)
@@ -125,7 +126,8 @@ def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, mel
         output_score = []
 
         for c in cscore:
-            tones_chord = sorted(set([t[4] % 12 for t in c if t[3] != 9]))
+            output_score.extend(c)
+'''            tones_chord = sorted(set([t[4] % 12 for t in c if t[3] != 9]))
             drums_events = [t for t in c if t[3] == 9]
 
             if tones_chord:
@@ -144,7 +146,8 @@ def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, mel
                     output_score.extend([c[0]] + drums_events)
             else:
                 output_score.extend(c)
-                        
+'''
+
     print('Done processing!')
     print('=' * 70)
     
@@ -154,6 +157,9 @@ def render_midi(input_midi, render_type, soundfont_bank, render_sample_rate, mel
     for e in output_score:
         e[1] = e[1] * 16
         e[2] = e[2] * 16
+
+        if -1 < custom_render_patch < 128:
+            e[6] = custom_render_patch
         
     print('Done recalculating timings!')
     print('=' * 70)
@@ -279,7 +285,9 @@ if __name__ == "__main__":
 
         render_sample_rate = gr.Radio(["16000", "32000", "44100"], label="MIDI audio render sample rate", value="16000")
 
-        melody_patch = gr.Slider(0, 127, value=40, label="Melody MIDI Patch")
+        custom_render_patch = gr.Slider(-1, 127, value=-1, label="Custom MIDI render patch")
+
+        melody_patch = gr.Slider(0, 127, value=40, label="Melody MIDI patch")
 
         submit = gr.Button()
 
@@ -292,7 +300,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 = submit.click(render_midi, [input_midi, render_type, soundfont_bank, render_sample_rate, melody_patch],
+        run_event = submit.click(render_midi, [input_midi, render_type, soundfont_bank, render_sample_rate, custom_render_patch, melody_patch],
                                                 [output_midi_md5, output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
         
     app.queue(1).launch(server_port=opt.port, share=opt.share, inbrowser=True)