Update app.py

app.py

@@ -278,137 +278,13 @@ if __name__ == "__main__":
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
     print('=' * 70)
-    print('Loading Pitches Chords Progressions dataset...')
+    print('Loading processed Pitches Chords Progressions dataset data...')
     print('=' * 70)
-    good_chords_chunks = TMIDIX.Tegridy_Any_Pickle_File_Reader('pitches_chords_progressions_5_3_15')
-
-    print('=' * 70)
-    print('Done!')
-    print('=' * 70)
-
-    #===============================================================================
-
-    minimum_chords_chunk_length = 4 # @param {"type":"slider","min":4,"max":8,"step":1}
-    chords_chunks_overlap_value = 4 # @param {"type":"slider","min":2,"max":8,"step":1}
-    
-    print('=' * 70)
-    print('Selecting chords chunks...')
-    print('=' * 70)
-    
-    chunk_size = minimum_chords_chunk_length
-    
-    long_chords_chunks = []
-    
-    for c in tqdm(good_chords_chunks):
-      if chunk_size + chords_chunks_overlap_value <= len(c):
-        long_chords_chunks.append(c)
-    
-    print('Done!')
-    print('=' * 70)
-    print('Selected chords chunks of minimum length:', minimum_chords_chunk_length+chords_chunks_overlap_value)
-    print('=' * 70)
-    print('Total number of selected chord chunks:', len(long_chords_chunks))
-    print('=' * 70)
-
-    chords_chunks_multiplicatrion_factor = 6 # @param {"type":"slider","min":1,"max":6,"step":1}
-    
-    #===============================================================================
-    # Helper chord function
-    #===============================================================================
-    
-    def check_chord(chord):
-    
-      tones_chord = sorted(set([p % 12 for p in chord]))
-    
-      new_tones_chord = []
-    
-      if 0 in tones_chord and 11 in tones_chord:
-        tones_chord.remove(11)
-    
-      for t in tones_chord:
-        if t+1 in tones_chord:
-          tones_chord.remove(t+1)
-        if t-1 in tones_chord:
-          tones_chord.remove(t-1)
-    
-      new_chord = tuple()
-    
-      for p in chord:
-        if p % 12 in tones_chord:
-          new_chord += tuple([p])
-    
-      if len(new_chord) > 2:
-        return new_chord
-    
-      else:
-        return None
-    
-    #===============================================================================
-    
-    print('=' * 70)
-    print('Multiplying chords chunks...')
-    print('=' * 70)
-    print('Chords chunks will be multiplied', chords_chunks_multiplicatrion_factor * 2, 'times' )
-    print('=' * 70)
-    
-    long_chords_chunks_mult = set()
-    
-    for c in tqdm(long_chords_chunks):
-    
-      for tv in range(-chords_chunks_multiplicatrion_factor, chords_chunks_multiplicatrion_factor):
-        gc = []
-        for cc in c:
-          chord = [max(1, min(127, p+tv)) for p in cc]
-          checked_chord = check_chord(chord)
-          if checked_chord is not None:
-            gc.append(checked_chord)
-        if len(gc) == len(c) or (len(gc) >= chunk_size + chords_chunks_overlap_value and gc == c[:len(gc)]) or (len(gc) >= chunk_size + chords_chunks_overlap_value and gc == c[len(gc):]):
-          long_chords_chunks_mult.add(tuple(gc))
-    
-    print('Done!')
-    print('=' * 70)
-    print('Total number of multiplied chords chunks:', len(long_chords_chunks_mult))
-    print('=' * 70)
-
-    #===============================================================================
-    
-    print('=' * 70)
-    print('Creating chords dictionary...')
-    print('=' * 70)
-    
-    long_tones_chords_dict = set()
-    
-    for a in tqdm(long_chords_chunks_mult):
-      for aa in a:
-        tones_chord = tuple(sorted(set([p % 12 for p in aa])))
-        long_tones_chords_dict.add(tones_chord)
-    
-    long_tones_chords_dict = list(long_tones_chords_dict)
+    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')
     
     print('=' * 70)
     print('Resulting chords dictionary size:', len(long_tones_chords_dict))
     print('=' * 70)
-    print('Preparing chords chunks...')
-    print('=' * 70)
-    
-    all_long_chords_tokens_chunks = []
-    all_long_good_chords_chunks = []
-    
-    for a in tqdm(long_chords_chunks_mult):
-    
-      chunk = []
-    
-      for aa in a:
-    
-        tones_chord = tuple(sorted(set([p % 12 for p in aa])))
-        chunk.append(long_tones_chords_dict.index(tones_chord))
-    
-      if chunk:
-        all_long_chords_tokens_chunks.append(chunk)
-        all_long_good_chords_chunks.append(a)
-    
-    print('Done!')
-    print('=' * 70)
     print('Loading chords chunks...')
     
     src_long_chunks = np.array([a[:chunk_size] for a in all_long_chords_tokens_chunks])