Update app.py

app.py

@@ -52,13 +52,20 @@ def find_midi(title, artist):
     
     # Compute cosine similarity between query and each sentence in the corpus
     similarities = util.cos_sim(query_embedding, corpus_embeddings)
-    
+
+    top_ten_matches_idxs = np.argsort(-similarities)[0][:10].tolist()
+   
     # Find the index of the most similar sentence
     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 += all_MIDI_files_names[t][0] + '\n'
+
     print('Done!')
     print('=' * 70)
     
@@ -89,6 +96,8 @@ def find_midi(title, artist):
     
     MIDI_score_data = scores_data[fnameidx][1]
     
+    print('Rendering results...')
+    print('=' * 70)
     print('Sample INTs', MIDI_score_data[:12])
     print('=' * 70)
     
@@ -154,11 +163,6 @@ def find_midi(title, artist):
     print('=' * 70)
     
     #===============================================================================
-    print('Rendering results...')
-    
-    print('=' * 70)
-    print('Sample INTs', song_f[:3])
-    print('=' * 70)
     
     output_score, patches, overflow_patches = TMIDIX.patch_enhanced_score_notes(song_f)
     
@@ -166,8 +170,7 @@ def find_midi(title, artist):
                                                           output_signature = 'Los Angeles MIDI Dataset Search',
                                                           output_file_name = song_artist,
                                                           track_name='Project Los Angeles',
-                                                          list_of_MIDI_patches=patches,
-                                                          timings_multiplier=16
+                                                          list_of_MIDI_patches=patches
                                                           )
     
     new_fn = song_artist + '.mid'
@@ -185,7 +188,7 @@ def find_midi(title, artist):
     #========================================================
     
     output_midi_title = str(song_artist)
-    output_midi_summary = str(song_f[:3])
+    output_midi_summary = str(top_ten_matches)
     output_midi = str(new_fn)
     output_audio = (16000, audio)
     
@@ -253,7 +256,7 @@ if __name__ == "__main__":
         gr.Markdown("# Search results")
 
         output_midi_title = gr.Textbox(label="Output MIDI title")
-        output_midi_summary = gr.Textbox(label="Output MIDI summary")
+        output_midi_summary = gr.Textbox(label="Top ten MIDI matches")
         output_audio = gr.Audio(label="Output MIDI audio", format="wav", elem_id="midi_audio")
         output_plot = gr.Plot(label="Output MIDI score plot")
         output_midi = gr.File(label="Output MIDI file", file_types=[".mid"])