Update app.py

app.py

@@ -25,6 +25,8 @@ print('Loading main MIDI Remixer modules...')
 
 import TMIDIX
 
+import numpy as np
+
 from midi_to_colab_audio import midi_to_colab_audio
 
 from huggingface_hub import hf_hub_download
@@ -48,8 +50,25 @@ SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
 #==================================================================================
 
+print('=' * 70)
+
 midi_cores_dataset = load_dataset("asigalov61/MIDI-Cores")
 
+print('=' * 70)
+print('Prepping MIDI cores data...')
+print('=' * 70)
+
+all_core_chords = []
+
+for entry in tqdm.tqdm(midi_cores_dataset['train']):
+    all_core_chords.append(entry['core_chords'])
+
+all_core_chords = np.array(all_core_chords)
+
+print('=' * 70)
+print('Done!')
+print('=' * 70)
+
 #==================================================================================
 
 def load_midi(midi_file):
@@ -92,10 +111,31 @@ def load_midi(midi_file):
 
 #==================================================================================
 
-def Remix_MIDI(input_midi,
-               chords_chunks_len,
-               num_mix_chunks
-              ):
+def find_max_exact_match(src, trg):
+    """
+    Find the row in the 2D trg array that has the maximum number of exact element matches with the 1D src array.
+
+    Parameters:
+    src (numpy.ndarray): 1D source array.
+    trg (numpy.ndarray): 2D target array.
+
+    Returns:
+    int: Index of the row in trg that has the maximum number of exact matches with src.
+    """
+    # Compare src with each row in trg and count exact matches
+    match_counts = np.sum(trg == src, axis=1)
+    
+    # Find the index of the row with the maximum number of matches
+    max_match_idx = np.argmax(match_counts)
+    
+    return max_match_idx
+
+#==================================================================================
+
+def Match_Cores(input_midi,
+                chords_chunks_len,
+                num_mix_chunks
+               ):
 
     #===============================================================================
     
@@ -334,7 +374,7 @@ with gr.Blocks() as demo:
     output_plot = gr.Plot(label="MIDI score plot")
     output_midi = gr.File(label="MIDI file", file_types=[".mid"])
 
-    mix_btn.click(Remix_MIDI, 
+    mix_btn.click(Match_Cores, 
                    [input_midi,
                     chords_chunks_len,
                     num_mix_chunks