Update app.py

app.py

@@ -180,6 +180,7 @@ def find_midi(input_search_string):
     
     output_midi_title = str(LAKH_hash)
     output_midi_summary = str(MIDI_score_metadata)
+    output_midi_caps = str(best_corpus_match)
     output_midi = str(new_fn)
     output_audio = (16000, audio)
     
@@ -188,6 +189,7 @@ def find_midi(input_search_string):
     print('Output MIDI file name:', output_midi)
     print('Output MIDI title:', output_midi_title)
     print('Output MIDI summary:', output_midi_summary)
+    print('Output MidiCaps information:', output_midi_caps)
     print('=' * 70) 
     
     #========================================================
@@ -197,7 +199,7 @@ def find_midi(input_search_string):
     print('-' * 70)
     print('Req execution time:', (reqtime.time() - start_time), 'sec')
     
-    return output_midi_title, output_midi_summary, output_midi, output_audio, output_plot        
+    return output_midi_title, output_midi_summary, output_midi_caps, output_midi, output_audio, output_plot        
     
 #==========================================================================================================
 
@@ -238,7 +240,7 @@ if __name__ == "__main__":
     
     with app:
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>LAKH MIDI Dataset Search</h1>")
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Search and explore LAKH MIDI dataset with sentence transformer</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Search and explore LAKH MIDI dataset with MidiCaps dataset and sentence transformer</h1>")
         
         gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.LAKH-MIDI-Dataset-Search&style=flat)\n\n"
                     "This is a demo for MidiCaps dataset\n\n"
@@ -254,12 +256,13 @@ if __name__ == "__main__":
         gr.Markdown("# Search results")
 
         output_midi_title = gr.Textbox(label="Output MIDI title")
-        output_midi_summary = gr.Textbox(label="Top ten MIDI matches")
+        output_midi_summary = gr.Textbox(label="Aggregated MIDI file text metadata")
+        output_midi_caps = gr.Textbox(label="MidiCaps dataset information")
         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"])
         
         run_event = submit.click(find_midi, [input_search_string],
-                                                  [output_midi_title, output_midi_summary, output_midi, output_audio, output_plot ])
+                                                  [output_midi_title, output_midi_summary, output_midi_caps, output_midi, output_audio, output_plot ])
         
     app.launch()