Update app.py

app.py

@@ -97,6 +97,7 @@ def ID_MIDI(input_midi):
     print('Processing...Please wait...')
 
     output_str = 'None'
+    output_midi_records_count = 0
     output_midi_src_dataset= 'Unknown'
     output_midi_path_str = 'None'
     
@@ -106,9 +107,13 @@ def ID_MIDI(input_midi):
         
         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."
 
-        output_midi_src_dataset = MIDID_database[new_midi_md5hash][0]['midi_dataset']
+        output_midi_records_count = len(MIDID_database[new_midi_md5hash])
 
-        output_midi_path_str = MIDID_database[new_midi_md5hash][0]['midi_path']
+        output_entry = random.choice(MIDID_database[new_midi_md5hash])
+
+        output_midi_src_dataset = output_entry['midi_dataset']
+
+        output_midi_path_str = output_entry['midi_path']
         
         data = 'Source MIDI dataset: ' + output_midi_src_dataset + '\n' + output_midi_path_str
         
@@ -136,7 +141,8 @@ def ID_MIDI(input_midi):
     
     print('Done!')
     print('=' * 70)
-    print('Source MIDI dataset', output_midi_src_dataset)
+    print('Original MIDI unique records count', output_midi_records_count)
+    print('Original MIDI dataset', output_midi_src_dataset)
     print('Original MIDI path string', output_midi_path_str)
     print('=' * 70)
     print(output_str)
@@ -155,7 +161,7 @@ def ID_MIDI(input_midi):
     
     #========================================================
     
-    return output_midi_md5, output_midi_src_dataset, output_midi_path_str, output_table_data
+    return output_midi_md5, output_midi_records_count, output_midi_src_dataset, output_midi_path_str, output_table_data
     
 #==========================================================================================================
 
@@ -192,13 +198,15 @@ 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_records_count = gr.Textbox(label="Original MIDI unique records count")
+        output_midi_src_dataset = gr.Textbox(label="Original MIDI dataset pretty name")
         output_midi_path_str = gr.Textbox(label="Original MIDI raw path string")
         output_MIDID_results_table = gr.Dataframe(label="MIDID database results table", wrap=True, col_count=(3, 'dynamic'))
         
         run_event = submit.click(ID_MIDI, [input_midi,                                         
                                               ],
                                                 [output_midi_md5,
+                                                 output_midi_records_count,
                                                  output_midi_src_dataset,
                                                  output_midi_path_str,                                                 
                                                  output_MIDID_results_table