Update app.py

app.py

@@ -96,13 +96,21 @@ def ID_MIDI(input_midi):
     print('=' * 70)
     print('Processing...Please wait...')
 
+    output_str = 'None'
+    output_midi_src_dataset= 'Unknown'
+    output_midi_path_str = 'None'
+    
     if new_midi_md5hash in MIDID_database:
 
         client = InferenceClient(api_key=HF_TOKEN)
         
         prompt = "Please create a summary table for a MIDI file based on the following keywords strings, best possible description and best possible summary fields. Please respond with the table only. Do not say anything else. Thank you."
 
-        data = 'Source MIDI dataset: ' + MIDID_database[new_midi_md5hash][0]['midi_dataset'] + '\n' + MIDID_database[new_midi_md5hash][0]['midi_path']
+        output_midi_src_dataset = MIDID_database[new_midi_md5hash][0]['midi_dataset']
+
+        output_midi_path_str = MIDID_database[new_midi_md5hash][0]['midi_path']
+        
+        data = 'Source MIDI dataset: ' + output_midi_src_dataset + '\n' + output_midi_path_str
         
         messages = [
         	{
@@ -128,7 +136,9 @@ def ID_MIDI(input_midi):
     
     print('Done!')
     print('=' * 70)
-    
+    print('Source MIDI dataset', output_midi_src_dataset)
+    print('Original MIDI path string', output_midi_path_str)
+    print('=' * 70)
     print(output_str)
     print('=')
 
@@ -182,12 +192,16 @@ if __name__ == "__main__":
         gr.Markdown("## MIDI identification results")
         
         output_midi_md5 = gr.Textbox(label="Monster MIDI dataset md5 hash")
+        output_midi_src_dataset = gr.Textbox(label="Source MIDI dataset pretty name")
+        output_midi_path_str = gr.Textbox(label="Original MIDI raw path string")
         output_MIDID_results_table = gr.Dataframe(label="MIDID results table", wrap=True, col_count=(3, 'dynamic'))
         
         run_event = submit.click(ID_MIDI, [input_midi,                                         
                                               ],
-                                                [output_midi_md5, 
-                                                    output_MIDID_results_table 
+                                                [output_midi_md5,
+                                                 output_midi_src_dataset,
+                                                 output_midi_path_str,                                                 
+                                                 output_MIDID_results_table
                                                 ])
         
     app.queue().launch()