Update app.py

app.py

@@ -52,12 +52,7 @@ def find_midi(input_search_string):
     closest_index = np.argmax(similarities)
     closest_index_match_ratio = max(similarities[0].tolist())
 
-    best_corpus_match = all_MIDI_files_names[closest_index]
-
-    top_ten_matches = ''
-
-    for t in top_ten_matches_idxs:
-        top_ten_matches += str(all_MIDI_files_names[t][0]).title() + '\n'
+    best_corpus_match = mc_dataset['train'][closest_index]
 
     print('Done!')
     print('=' * 70)
@@ -69,9 +64,9 @@ def find_midi(input_search_string):
     print('Done!')
     print('=' * 70)
 
-    song_artist = best_corpus_match[0]
-    song_artist_title = str(song_artist).title()
-    zlib_file_name = best_corpus_match[1]
+    LAKH_hash = best_corpus_match['location'].split('/')[-1].split('.mid')[0]
+    LAKH_caption = str(best_corpus_match['caption'])
+    zlib_file_name = all_MIDI_files_names[MIDI_files_names.index(LAKH_hash)][1]
 
     print('Fetching MIDI score...')
 
@@ -86,13 +81,14 @@ def find_midi(input_search_string):
     
     fnames = [f[0] for f in scores_data]
     
-    fnameidx = fnames.index(song_artist)
-    
-    MIDI_score_data = scores_data[fnameidx][1]
+    fnameidx = fnames.index(LAKH_hash)
+
+    MIDI_score_metadata = scores_data[fnameidx][1]
+    MIDI_score_data = scores_data[fnameidx][2]
     
     print('Rendering results...')
     print('=' * 70)
-    print('MIDi Title:', song_artist_title)
+    print('MIDi Title:', LAKH_hash)
     print('Sample INTs', MIDI_score_data[:12])
     print('=' * 70)
     
@@ -163,12 +159,12 @@ def find_midi(input_search_string):
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(output_score,
                                                           output_signature = 'LAKH MIDI Dataset Search',
-                                                          output_file_name = song_artist_title,
+                                                          output_file_name = LAKH_hash,
                                                           track_name='Project Los Angeles',
                                                           list_of_MIDI_patches=patches
                                                           )
     
-    new_fn = song_artist_title + '.mid'
+    new_fn = LAKH_hash + '.mid'
         
     audio = midi_to_colab_audio(new_fn, 
                     soundfont_path=soundfont,
@@ -182,8 +178,8 @@ def find_midi(input_search_string):
     
     #========================================================
     
-    output_midi_title = str(song_artist_title)
-    output_midi_summary = str(top_ten_matches)
+    output_midi_title = str(LAKH_hash)
+    output_midi_summary = str(MIDI_score_metadata)
     output_midi = str(new_fn)
     output_audio = (16000, audio)