Update app.py

app.py

@@ -160,6 +160,8 @@ if __name__ == "__main__":
     print('App start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
     print('=' * 70)
 
+    #===============================================================================
+
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
     print('=' * 70)
@@ -171,6 +173,140 @@ if __name__ == "__main__":
     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)
+    
+    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])
+    
+    print('Done!')
+    print('=' * 70)
+    print('Total chords chunks count:', len(all_long_good_chords_chunks))
+    print('=' * 70)
+    
+    #===============================================================================
+    
     app = gr.Blocks()
     with app:
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Chords Progressions Generator</h1>")