Update app.py

app.py

@@ -25,7 +25,10 @@ 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('=' * 70)
     print('Preparing to search...')
     
@@ -90,7 +93,7 @@ def find_midi(search_string, search_options):
     ax=plt.axes(title=title)
     ax.set_facecolor('black')
     
-    plt.scatter(x,y, s=2, c=c)
+    plt.scatter(x,y, s=5, c=c)
     plt.xlabel("Time in MIDI ticks")
     plt.ylabel("MIDI Pitch")
 
@@ -102,7 +105,7 @@ def find_midi(search_string, search_options):
                         volume_scale=10,
                         output_for_gradio=True
                         )
-    yield AUX_DATA[search_match_index][0], fn+'.mid', (16000, audio), plt
+    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
 
 #==========================================================================================================
 
@@ -141,12 +144,15 @@ if __name__ == "__main__":
 
         gr.Markdown("# Search results")
 
-        output_midi_seq = gr.Textbox(label="Found MIDI search title")
-        output_audio = gr.Audio(label="Output MIDI search sample audio", format="wav", elem_id="midi_audio")
-        output_plot = gr.Plot(label="Output MIDI search sample plot")
-        output_midi = gr.File(label="Output MIDI search sample MIDI", file_types=[".mid"])
+        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")
+        output_plot = gr.Plot(label="Output Karaoke MIDI score plot")
+        output_midi_lyric = gr.Textbox(label="Output Karaoke MIDI lyric")
+        output_midi_summary = gr.Textbox(label="Output Karaoke MIDI summary")
+        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_seq, output_midi, output_audio, output_plot])
+                                                  [output_midi_md5, 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)