Update app.py

app.py

@@ -2,6 +2,9 @@ import argparse
 import glob
 import os.path
 
+import time
+import datetime
+
 import gradio as gr
 
 import pickle
@@ -25,16 +28,19 @@ in_space = os.getenv("SYSTEM") == "spaces"
 #==========================================================================================================
 
 def find_midi(search_string, search_options):
-
     print('=' * 70)
-    print('Search string', search_string)
-    print('Search options', search_options)
+    print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
+    start_time = time.time()
+    
+    print('=' * 70)
+    print('Search string:', search_string)
+    print('Search options:', search_options)
+    
     print('=' * 70)
     print('Preparing to search...')
     
     #==================================================
     
-    
     random.shuffle(AUX_DATA)
     
     search_data = []
@@ -59,6 +65,7 @@ def find_midi(search_string, search_options):
     
     print('Done!')
     print('=' * 70)
+    print('Search match ratio:', search_match_ratio)
     print('Selected file/title:', AUX_DATA[search_match_index][:2])
     print('=' * 70)
 
@@ -93,19 +100,25 @@ def find_midi(search_string, search_options):
     ax=plt.axes(title=title)
     ax.set_facecolor('black')
     
-    plt.scatter(x,y, s=5, c=c)
+    plt.scatter(x,y, s=10, c=c)
     plt.xlabel("Time in MIDI ticks")
     plt.ylabel("MIDI Pitch")
 
     with open(fn+'.mid', 'wb') as f:
         f.write(TMIDIX.score2midi(raw_score))
+    
     audio = midi_to_colab_audio(fn+'.mid', 
                         soundfont_path=soundfont_path, 
                         sample_rate=16000, # 44100
                         volume_scale=10,
                         output_for_gradio=True
                         )
-    yield AUX_DATA[search_match_index][0], AUX_DATA[search_match_index][1], AUX_DATA[search_match_index][2], AUX_DATA[search_match_index][3],  fn+'.mid', (16000, audio), plt
+
+    print('Req end time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
+    print('-' * 70)
+    print('Req execution time:', (time.time() - start_time), 'sec')
+    
+    yield AUX_DATA[search_match_index][0], str(search_match_ratio), AUX_DATA[search_match_index][1], AUX_DATA[search_match_index][2], AUX_DATA[search_match_index][3],  fn+'.mid', (16000, audio), plt
 
 #==========================================================================================================
 
@@ -138,12 +151,13 @@ if __name__ == "__main__":
                     " for all features"
                     )
         
-        search_string = gr.Textbox(label="Enter search prompt here", value="And Nothing Else Matters")
-        search_options = gr.CheckboxGroup(["Titles", "Lyrics", "Summaries"], value="Lyrics", label="Search within", info="Where to search?")
+        search_string = gr.Textbox(label="Enter search prompt here", value="Nothing Else Matters")
+        search_options = gr.CheckboxGroup(["Titles", "Lyrics", "Summaries"], value="Lyrics", label="Search within")
         submit = gr.Button()
 
         gr.Markdown("# Search results")
-
+        
+        output_midi_search_match_ratio = gr.Textbox(label="Output Karaoke MIDI search match ratio")
         output_midi_md5 = gr.Textbox(label="Output Karaoke MIDI md5 hash")
         output_midi_title = gr.Textbox(label="Output Karaoke MIDI title")
         output_audio = gr.Audio(label="Output Karaoke MIDI audio", format="wav", elem_id="midi_audio")
@@ -153,6 +167,6 @@ if __name__ == "__main__":
         output_midi = gr.File(label="Output Karaoke MIDI original file", file_types=[".mid"])
         
         run_event = submit.click(find_midi, [search_string, search_options],
-                                                  [output_midi_md5, output_midi_title, output_midi_lyric, output_midi_summary, output_midi, output_audio, output_plot])
+                                                  [output_midi_md5, output_midi_search_match_ratio, output_midi_title, output_midi_lyric, output_midi_summary, output_midi, output_audio, output_plot])
         
     app.queue(1).launch(server_port=opt.port, share=opt.share, inbrowser=True)