Update app.py

app.py

@@ -131,46 +131,51 @@ 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)
+    current_chord = []
+    
+    for token in tokens:
+        if token < 256:
+            if current_chord:
+                chords.append(current_chord)
+                
+            current_chord = [token]
+        
+        else:
+            current_chord.append(token)
+    
+    if current_chord:
+        chords.append(current_chord)
 
-            cho = [t]
+    result = []
+    
+    for chord in chords:
 
-        else:
-            cho.append(t)
-            
-    if cho:
-        chords.append(cho)
+        header = chord[0]
 
-    checked_tokens = []
+        deduped = []
 
-    for c in chords:
+        if len(chord) > 1:
 
-        tchord = [t for t in c if 256 < t < 18816]
+            mid_tokens = [t for t in chord if 256 < t < 18816]
+            
+            seen = set()
+    
+            for i in range(0, len(mid_tokens), 2):
+                
+                pair = mid_tokens[i:i + 2]
+                
+                if pair and pair[0] not in seen:
+                    seen.add(pair[0])
+                    deduped.extend(pair)
 
-        nchord = []
-        seen = set()
+        high_tokens = [t for t in chord if t > 18815]
 
-        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])
+        result.extend([header] + deduped + high_tokens)
 
-        fchord = chords[0]
-        fchord.extend(nchord)
-        fchord.extend([t for t in c if c > 18815])
-        
-        checked_tokens.extend(fchord) 
+    return result
 
-    return checked_tokens
-    
 # -----------------------------
 # MIDI PROCESSING FUNCTIONS
 # -----------------------------