Update app.py

app.py

@@ -24,28 +24,32 @@ in_space = os.getenv("SYSTEM") == "spaces"
 
 #==========================================================================================================
 
-def find_midi(title, artist):
+def find_midi(search_string, search_options):
   
     print('=' * 70)
-    print('Loading MIDI file...')
+    print('Preparing to search...')
     
     #==================================================
     
-    print('Searching titles...Please wait...')
+    
     random.shuffle(AUX_DATA)
     
-    titles_index = []
+    search_data = []
     
     for A in AUX_DATA:
-      titles_index.append(A[0])
+        data = []
+        if 'Titles' in search_options:
+            data.append(A[1])
+        if 'Lyrics' in search_options:
+            data.append(A[2])
+        if 'Summaries' in search_options:
+            data.append(A[3])
+
+        search_data.append(data)
     
-    search_string = ''
+    print('Searching titles...Please wait...')
     
-    if title != '' and artist != '':
-      search_string = title + ' --- ' + artist
     
-    else:
-      search_string = title + artist
     
     search_match = process.extract(query=search_string, choices=titles_index, limit=1)
     search_index = titles_index.index(search_match[0][0])
@@ -215,7 +219,7 @@ if __name__ == "__main__":
         output_plot = gr.Plot(label="Output MIDI search sample plot")
         output_midi = gr.File(label="Output MIDI search sample MIDI", file_types=[".mid"])
         
-        run_event = submit.click(find_midi, [title, artist],
+        run_event = submit.click(find_midi, [search_string, search_options],
                                                   [output_midi_seq, output_midi, output_audio, output_plot])
         
     app.queue(1).launch(server_port=opt.port, share=opt.share, inbrowser=True)