Update app.py

app.py

@@ -182,6 +182,87 @@ def Harmonize_Melody(input_src_midi,
     print('=' * 70)
     
     #===============================================================================
+
+    def find_best_match(matches_indexes, previous_match_index):
+    
+      msigs = []
+    
+      for midx in matches_indexes:
+    
+        mat = all_chords_ptcs_chunks[midx]
+    
+        msig = []
+    
+        for m in mat:
+          msig.extend([sum(m) / len(m), len(m)])
+    
+        msigs.append(msig)
+    
+      pmat = all_chords_ptcs_chunks[previous_match_index]
+    
+      psig = []
+    
+      for p in pmat:
+        psig.extend([sum(p) / len(p), len(p)])
+    
+      dists = []
+    
+      for m in msigs:
+        dists.append(TMIDIX.minkowski_distance(psig, m))
+    
+      min_dist = min(dists)
+      min_dist_idx = dists.index(min_dist)
+    
+      return matches_indexes[min_dist_idx]
+
+    #===============================================================================
+
+    if texture_harmonized_chords:
+
+        print('=' * 70)
+        print('Texturing harmonized chords...')
+        print('=' * 70)
+    
+        chunk_length = 2
+        
+        harm_chords = [TMIDIX.ALL_CHORDS_FILTERED[s-12] for s in song if 11 < s < 141]
+        
+        harm_toks = [TMIDIX.ALL_CHORDS_FILTERED.index(c) for c in harm_chords] + [TMIDIX.ALL_CHORDS_FILTERED.index(harm_chords[-1])] * (chunk_length - (len(harm_chords) % chunk_length))
+        
+        final_song = []
+        
+        trg_chunk = np.array(harm_toks[:chunk_length])
+        sidxs = np.where((src_chunks == trg_chunk).all(axis=1))[0].tolist()
+        
+        sidx = random.choice(sidxs)
+        pidx = sidx
+        
+        final_song.extend(all_chords_ptcs_chunks[sidx])
+        
+        for i in tqdm.tqdm(range(chunk_length, len(harm_toks), chunk_length)):
+        
+          trg_chunk = np.array(harm_toks[i:i+chunk_length])
+        
+          sidxs = np.where((src_chunks == trg_chunk).all(axis=1))[0].tolist()
+        
+          if len(sidxs) > 0:
+        
+            sidx = find_best_match(sidxs, pidx)
+            pidx = sidx
+        
+            final_song.extend(all_chords_ptcs_chunks[sidx])
+        
+          else:
+            print('Dead end!')
+            break
+        
+        final_song = final_song[:len(harm_chords)]
+        
+        print('=' * 70)
+        print(len(final_song))
+        print('=' * 70)
+        print('Done!')
+        print('=' * 70)
     
     print('Rendering results...')
     print('=' * 70)
@@ -197,8 +278,10 @@ def Harmonize_Melody(input_src_midi,
       patches[2] = base_MIDI_patch_number
     
     patches[3] = melody_MIDI_patch_number
+
+    i = 0
     
-    for i, s in enumerate(song):
+    for s in song:
 
         if 11 < s < 141:
     
@@ -217,6 +300,8 @@ def Harmonize_Melody(input_src_midi,
                 if base_MIDI_patch_number > -1:
                     output_score.append(['note', time, dur, 2, chord[-1]+24, 120-chord[-1], base_MIDI_patch_number])
 
+            i += 1
+
     fn1 = "Melody-Harmonizer-Transformer-Composition"
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(output_score,