Update app.py

app.py

@@ -18,6 +18,7 @@ import math
 from tqdm import tqdm
 
 import TMIDIX
+import HaystackSearch
 
 from midi_to_colab_audio import midi_to_colab_audio
 
@@ -316,22 +317,25 @@ if __name__ == "__main__":
 
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
-    print('Loading processed Pitches Chords Progressions dataset data...')
+    print('Loading Monster Harmonized Melodies MIDI Dataset...')
     print('=' * 70)
-    long_tones_chords_dict, all_long_chords_tokens_chunks, all_long_good_chords_chunks = TMIDIX.Tegridy_Any_Pickle_File_Reader('processed_chords_progressions_chunks_data')
+    all_chords_chunks = TMIDIX.Tegridy_Any_Pickle_File_Reader('Monster_Harmonized_Melodies_MIDI_Dataset')
     
     print('=' * 70)
-    print('Resulting chords dictionary size:', len(long_tones_chords_dict))
+    print('Total number of harmonized melodies:', len(all_chords_chunks))
     print('=' * 70)
-    print('Loading chords chunks...')
+    print('Loading melodies...')
 
-    chunk_size = 4
+    long_mels_chunks_mult = []
+    long_chords_chunks_mult = []
     
-    src_long_chunks = np.array([a[:chunk_size] for a in all_long_chords_tokens_chunks])
+    for c in tqdm(all_chords_chunks):
+      long_mels_chunks_mult.append([p % 12 for p in c[0]])
+      long_chords_chunks_mult.append(c[1])
     
     print('Done!')
     print('=' * 70)
-    print('Total chords chunks count:', len(all_long_good_chords_chunks))
+    print('Total loaded melodies count:', len(long_mels_chunks_mult))
     print('=' * 70)
     
     #===============================================================================