Update app.py

app.py

@@ -16,21 +16,25 @@ import statistics
 
 import gradio as gr
 
+from huggingface_hub import InferenceClient
+
 import TMIDIX
 
 #==========================================================================================================
 
-def ID_MIDI(input_midi, 
-            render_type, 
-            soundfont_bank, 
-            render_sample_rate, 
-            custom_render_patch,
-            render_align,
-            render_transpose_value,
-            render_transpose_to_C4,
-            render_output_as_solo_piano,
-            render_remove_drums 
-            ):
+def format_table_data(data_string):
+    # Split the string into rows based on newlines
+    rows = data_string.strip().split("\n")
+
+    # Split each row into columns based on the separator '|' and strip extra spaces
+    data_list = [row.split("|")[1:-1] for row in rows]
+    formatted_data = [[cell.strip() for cell in row] for row in data_list]
+
+    return formatted_data
+
+#==========================================================================================================
+
+def ID_MIDI(input_midi):
     
     print('*' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -51,16 +55,6 @@ def ID_MIDI(input_midi,
     print('=' * 70)
     print('Input MIDI file name:', fn)
     print('Input MIDI md5 hash', input_midi_md5hash)
-    print('-' * 70)
-    print('Render type:', render_type)
-    print('Soudnfont bank', soundfont_bank)
-    print('Audio render sample rate', render_sample_rate)
-    print('Custom MIDI render patch', custom_render_patch)
-    print('Align to bars:', render_align)
-    print('Transpose value:', render_transpose_value)
-    print('Transpose to C4', render_transpose_to_C4)
-    print('Output as Solo Piano', render_output_as_solo_piano)
-    print('Remove drums:', render_remove_drums)
     print('=' * 70)
     print('Processing MIDI...Please wait...')
     
@@ -85,31 +79,41 @@ def ID_MIDI(input_midi,
     print('=' * 70)
     print('Processing...Please wait...')
 
+    if input_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. Please add best possible description and best possible summary fields. Please respond with the table only. Do not say anything else. Thank you."
+        
+        messages = [
+        	{
+        		"role": "user",
+        		"content": prompt + "\n\n" + data
+        	}
+        ]
+    
+        completion = client.chat.completions.create(
+            #model="Qwen/Qwen2.5-72B-Instruct",
+            model="mistralai/Mistral-Nemo-Instruct-2407",
+        	messages=messages, 
+        	max_tokens=500
+        )
+    
+        output_str = completion.choices[0].message['content']
+    
+        output_table_data = format_table_data(output_str)
 
-    print('-' * 70)
+    else:
 
-    new_md5_hash = hashlib.md5(open(new_fn,'rb').read()).hexdigest()
+        output_table_data = None
     
     print('Done!')
     print('=' * 70)
 
     #========================================================
 
-    output_midi_md5 = str(new_md5_hash)
-    output_midi_title = str(fn1)
-    output_midi_summary = str(meta_data)
-    output_midi = str(new_fn)
-    output_audio = (srate, audio)
-    
-    output_plot = None
-
-    print('Output MIDI file name:', output_midi)
-    print('Output MIDI title:', output_midi_title)
-    print('Output MIDI hash:', output_midi_md5)
-    print('Output MIDI summary:', output_midi_summary[:5])
-    print('=' * 70) 
-    
+    output_midi_md5 = str(input_midi_md5hash)
+   
     #========================================================
 
     print('Req end time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -119,7 +123,7 @@ def ID_MIDI(input_midi,
     
     #========================================================
     
-    return output_midi_md5, output_midi_title, output_midi_summary, output_midi, output_audio, output_plot
+    return output_midi_md5, output_table_data
     
 #==========================================================================================================
 
@@ -131,6 +135,11 @@ if __name__ == "__main__":
     print('App start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     print('=' * 70)
 
+    print('Loading MIDID database...')
+    MIDID_database = TMIDIX.Tegridy_Any_Pickle_File_Reader('MIDID_Basic_Database_CC_BY_NC_SA.pickle')
+    
+    print('=' * 70)
+
     app = gr.Blocks()
     
     with app:
@@ -150,15 +159,13 @@ if __name__ == "__main__":
 
         gr.Markdown("## MIDI identification results")
         
-        output_midi_md5 = gr.Textbox(label="Output MIDI md5 hash")
-        output_midi_title = gr.Textbox(label="Output MIDI title")
-        output_midi_summary = gr.Textbox(label="Output MIDI summary")
+        output_midi_md5 = gr.Textbox(label="Input MIDI md5 hash")
+        output_MIDID_results_table = gr.Table(label="MIDID results table")
         
         run_event = submit.click(ID_MIDI, [input_midi,                                         
                                               ],
                                                 [output_midi_md5, 
-                                                 output_midi_title, 
-                                                 output_midi_summary, 
+                                                    output_MIDID_results_table 
                                                 ])
         
     app.queue().launch()