Update app.py

app.py

@@ -23,7 +23,7 @@ in_space = os.getenv("SYSTEM") == "spaces"
 # =================================================================================================
                        
 @spaces.GPU
-def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type):
+def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type, input_strip_notes):
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     start_time = reqtime.time()
@@ -75,6 +75,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type)
     print('Input file name:', fn)
     print('Req num toks:', input_num_tokens)
     print('Conditioning type:', input_conditioning_type)
+    print('Strip notes:', input_strip_notes)
     print('-' * 70)
 
     #===============================================================================
@@ -185,7 +186,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type)
         
                 if delta_time != 0:
                     melody_chords.extend([delta_time, dur+128, ptc+256])
-                    if strip_notes_from_composition:
+                    if input_strip_notes:
                       if len(c) > 1:
                         times.append(delta_time)
                         durs.append(dur+128)
@@ -376,6 +377,7 @@ if __name__ == "__main__":
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
         input_num_tokens = gr.Slider(4, 128, value=32, step=1, label="Number of composition chords to generate progression for")
         input_conditioning_type = gr.Radio(["Chords", "Chords-Times", "Chords-Times-Durations"], label="Conditioning type")
+        input_strip_notes = gr.Checkbox(label="Strip notes from the composition")
         
         run_btn = gr.Button("generate", variant="primary")
 
@@ -388,24 +390,24 @@ if __name__ == "__main__":
         output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])
 
 
-        run_event = run_btn.click(GenerateAccompaniment, [input_midi, input_num_tokens, input_conditioning_type],
+        run_event = run_btn.click(GenerateAccompaniment, [input_midi, input_num_tokens, input_conditioning_type, input_strip_notes],
                                   [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot])
 
         gr.Examples(
-            [["Chords-Progressions-Transformer-Piano-Seed-1.mid", 128, "Chords"], 
-             ["Chords-Progressions-Transformer-Piano-Seed-2.mid", 128, "Chords"], 
-             ["Chords-Progressions-Transformer-Piano-Seed-3.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-Piano-Seed-4.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-Piano-Seed-5.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-Piano-Seed-6.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-MI-Seed-1.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-MI-Seed-2.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-MI-Seed-3.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-MI-Seed-4.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-MI-Seed-5.mid", 128, "Chords"],
-             ["Chords-Progressions-Transformer-MI-Seed-6.mid", 128, "Chords"]
+            [["Chords-Progressions-Transformer-Piano-Seed-1.mid", 128, "Chords", False], 
+             ["Chords-Progressions-Transformer-Piano-Seed-2.mid", 128, "Chords-Times", False], 
+             ["Chords-Progressions-Transformer-Piano-Seed-3.mid", 128, "Chords-Times-Druations", False],
+             ["Chords-Progressions-Transformer-Piano-Seed-4.mid", 128, "Chords", False],
+             ["Chords-Progressions-Transformer-Piano-Seed-5.mid", 128, "Chords-Times", False],
+             ["Chords-Progressions-Transformer-Piano-Seed-6.mid", 128, "Chords-Times-Durations", False],
+             ["Chords-Progressions-Transformer-MI-Seed-1.mid", 128, "Chords", False],
+             ["Chords-Progressions-Transformer-MI-Seed-2.mid", 128, "Chords-Times", False],
+             ["Chords-Progressions-Transformer-MI-Seed-3.mid", 128, "Chords-Times-Durations", False],
+             ["Chords-Progressions-Transformer-MI-Seed-4.mid", 128, "Chords-Times", False],
+             ["Chords-Progressions-Transformer-MI-Seed-5.mid", 128, "Chords", False],
+             ["Chords-Progressions-Transformer-MI-Seed-6.mid", 128, "Chords-Times-Durations", False]
             ],
-            [input_midi, input_num_tokens, input_conditioning_type],
+            [input_midi, input_num_tokens, input_conditioning_type, input_strip_notes],
             [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot],
             GenerateAccompaniment,
             cache_examples=True,