Update app.py

app.py

@@ -28,7 +28,7 @@ in_space = os.getenv("SYSTEM") == "spaces"
 
 #==========================================================================================================
 
-def render_midi(input_midi, render_options):
+def render_midi(input_midi, render_type):
     
     print('*' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -46,7 +46,7 @@ def render_midi(input_midi, render_options):
     print('=' * 70)
     print('Input MIDI file name:', fn)
     print('Input MIDI md5 hash', input_midi_md5hash)
-    print('Render options:', render_options)
+    print('Render type:', render_type)
     
     print('=' * 70)
     print('Processing MIDI...Please wait...')
@@ -80,17 +80,18 @@ def render_midi(input_midi, render_options):
     print('=' * 70)
     print('Processing...Please wait...')
 
-    if not render_options:
-        output_score = escore
+    if render_type == "Render as-is" or not render_type:
+        output_score = copy.deepcopy(escore)
 
-    elif "Render as-is" in render_options:
-        output_score = escore
-
-    elif "Extract melody" in render_options:
-        output_score = [c[0] for c in cscore if c[0][3] != 9]
+    elif render_type == "Extract melody":
+        output_score = copy.deepcopy([c[0] for c in cscore if c[0][3] != 9])
+        
         for e in output_score:
             e[3] = 0
             e[6] = 40
+
+            if e[4] < 60:
+                e[4] = (e[4] % 12) + 60
         
     print('Done processing!')
     print('=' * 70)
@@ -207,8 +208,8 @@ if __name__ == "__main__":
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"], type="filepath")
 
         gr.Markdown("## Select desired render options")
-        
-        render_options = gr.CheckboxGroup(["Render as-is", "Extract melody", "Transform"], value="Render as-is")
+
+        render_type = gr.Radio(["Render as-is", "Extract melody", "Transform"], label="Render type", value="Render as-is")
 
         submit = gr.Button()
 
@@ -221,7 +222,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_options],
+        run_event = submit.click(render_midi, [input_midi, render_type],
                                                 [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)