Update app.py

app.py

@@ -172,7 +172,8 @@ def load_midi(input_midi, melody_patch=-1, use_nth_note=1):
 #==================================================================================
 
 @spaces.GPU
-def Generate_Accompaniment(input_midi, 
+def Generate_Accompaniment(input_midi,
+                           input_melody,
                            generation_type,
                            melody_patch,
                            use_nth_note,
@@ -266,6 +267,7 @@ def Generate_Accompaniment(input_midi,
     print('Requested settings:')
     print('=' * 70)
     print('Input MIDI file name:', fn)
+    print('Input sample melody:', input_melody)
     print('Generation type:', generation_type)
     print('Source melody patch:', melody_patch)
     print('Use nth melody note:', use_nth_note)
@@ -275,7 +277,12 @@ def Generate_Accompaniment(input_midi,
 
     #==================================================================
 
-    score, score_list = load_midi(input_midi.name, melody_patch, use_nth_note)
+    if input_melody == 'Custom MIDI':
+        score, score_list = load_midi(input_midi.name, melody_patch, use_nth_note)
+
+    else:
+        score_list = popular_hook_melodies[[m[0] for m in popular_hook_melodies].index(input_melody)][1]
+        score = TMIDIX.flatten(score)
 
     print('Sample score events', score[:12])
 
@@ -462,9 +469,11 @@ with gr.Blocks() as demo:
     
     #==================================================================================
     
-    gr.Markdown("## Upload source melody MIDI or select an example MIDI below")
+    gr.Markdown("## Upload source melody MIDI or select a sample melody below")
     
     input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
+
+    input_melody = gr.Dropdown([m[0] for m in popular_hook_melodies], value="Custom MIDI", label="Popular melodies database")
     
     gr.Markdown("## Generation options")
     
@@ -482,7 +491,8 @@ with gr.Blocks() as demo:
     output_midi = gr.File(label="MIDI file", file_types=[".mid"])
 
     generate_btn.click(Generate_Accompaniment, 
-                       [input_midi, 
+                       [input_midi,
+                        input_melody,
                         generation_type,
                         melody_patch,
                         use_nth_note,
@@ -495,11 +505,11 @@ with gr.Blocks() as demo:
                       )
 
     gr.Examples(
-                [["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]
+                [["USSR-National-Anthem-Seed-Melody.mid", "Custom MIDI", "Guided", -1, 1, 0.9],
+                 ["Sparks-Fly-Seed-Melody.mid", "Custom MIDI", "Guided", -1, 1, 0.9]
                 ],
-                [input_midi, 
+                [input_midi,
+                 input_melody,
                  generation_type,
                  melody_patch,
                  use_nth_note,