Update app.py

app.py

@@ -130,6 +130,47 @@ def render_midi_output(final_composition):
     )
     return (16000, midi_audio), midi_plot, fname + '.mid', time_val
 
+def remove_duplicate_pitches(tokens):
+
+    chords = []
+    cho = []
+
+    for t in tokens:
+        if t < 256:
+            if cho:
+                chords.append(cho)
+
+            cho = [t]
+
+        else:
+            cho.append(t)
+            
+    if cho:
+        chords.append(cho)
+
+    checked_tokens = []
+
+    for c in chords:
+
+        tchord = [t for t in c if 256 < t < 18816]
+
+        nchord = []
+        seen = set()
+
+        for i in range(0, len(tchord), 2):
+            tup = tchord[i: i+2]
+            if tup[0] not in seen:
+                nchord.extend(tup)
+                seen.add(tup[0])
+
+        fchord = chords[0]
+        fchord.extend(nchord)
+        fchord.extend([t for t in c if c > 18815])
+        
+        checked_tokens.extend(fchord) 
+
+    return checked_tokens
+    
 # -----------------------------
 # MIDI PROCESSING FUNCTIONS
 # -----------------------------