Update app.py

app.py

@@ -97,7 +97,9 @@ def find_max_exact_match(src, trg):
     trg (numpy.ndarray): 2D target array.
 
     Returns:
-    int: Index of the row in trg that has the maximum number of exact matches with src.
+    tuple: A tuple containing:
+        - int: Index of the row in trg with the maximum number of exact matches.
+        - int: Number of matched elements in that row.
     """
     # Compare src with each row in trg and count exact matches
     match_counts = np.sum(trg == src, axis=1)
@@ -105,7 +107,10 @@ def find_max_exact_match(src, trg):
     # Find the index of the row with the maximum number of matches
     max_match_idx = np.argmax(match_counts)
     
-    return max_match_idx
+    # Get the number of matched elements in the best-matching row
+    num_matched_elements = match_counts[max_match_idx]
+    
+    return max_match_idx, num_matched_elements
 
 #==================================================================================
 
@@ -139,16 +144,13 @@ def Match_Cores(input_midi,
     
     #===============================================================================
     
-    #==================================================================
-    
     print('Matching MIDI cores...')
     print('=' * 70)
 
-    match_idx = find_max_exact_match(src_core_chords, all_core_chords)
+    match_idx, num_matches = find_max_exact_match(src_core_chords, all_core_chords)
 
     print('MAX MATCH IDX', match_idx)
-
-    #==================================================================
+    print('NUM MATCHES', num_matches)
 
     print('=' * 70)
     print('Done!')
@@ -160,10 +162,6 @@ def Match_Cores(input_midi,
 
     final_core_score = midi_cores_dataset['train'][int(match_idx)]['core_score']
 
-    
-        
-    #===============================================================================
-
     print('Done!')
     print('=' * 70)
 
@@ -280,15 +278,11 @@ with gr.Blocks() as demo:
     #==================================================================================
 
     gr.Markdown("## Upload source MIDI")
+    gr.Markdown("### Source MIDI must have at least 128 chords")
     
     input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
     
-    gr.Markdown("## Mixing options")
-    
-    chords_chunks_len = gr.Slider(3, 7, value=4, step=1, label="Number of chords to match for each repeating chunk")
-    num_mix_chunks = gr.Slider(4, 20, value=10, step=1, label="Number of repeating chunks to mix")
-    
-    mix_btn = gr.Button("Remix", variant="primary")
+    mix_btn = gr.Button("Match", variant="primary")
 
     gr.Markdown("## Mixing results")