Update app.py

app.py

@@ -130,57 +130,11 @@ def render_midi_output(final_composition):
     )
     return (16000, midi_audio), midi_plot, fname + '.mid', time_val
 
-def remove_duplicate_pitches(tokens):
-    """Remove duplicate notes from tokens seq"""
-    chords = []
-    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)
-
-    result = []
-    
-    for chord in chords:
-
-        header = chord[0]
-
-        deduped = []
-
-        if len(chord) > 1:
-
-            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)
-
-        high_tokens = [t for t in chord if 18816 < t < 18819]
-
-        result.extend([header] + deduped + high_tokens)
-
-    return result
-
 # -----------------------------
 # MIDI PROCESSING FUNCTIONS
 # -----------------------------
 def load_midi(input_midi):
-    """Process the input MIDI file and create a token sequence using without velocity logic."""
+    """Process the input MIDI file and create a token sequence."""
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi.name)
     
     escore_notes = TMIDIX.advanced_score_processor(raw_score, return_enhanced_score_notes=True, apply_sustain=True)
@@ -232,8 +186,6 @@ def load_midi(input_midi):
                 dur_vel = (8 * dur) + velocity
         
                 melody_chords.extend([pat_ptc+256, dur_vel+16768])
-    
-        melody_chords = remove_duplicate_pitches(melody_chords)
             
         return melody_chords
 
@@ -241,8 +193,7 @@ def load_midi(input_midi):
         return [18816]
 
 def save_midi(tokens):
-    """Convert token sequence back to a MIDI score and write it using TMIDIX (without velocity).
-       The output MIDI file name incorporates a date-time stamp.
+    """Convert token sequence back to a MIDI score and write it using TMIDIX.
     """
 
     time = 0
@@ -259,8 +210,6 @@ def save_midi(tokens):
 
     song_f = []
 
-    tokens = remove_duplicate_pitches(tokens)
-
     for ss in tokens:
 
         if 0 <= ss < 256:
@@ -338,13 +287,10 @@ def generate_music(prime, num_gen_tokens, num_mem_tokens, num_gen_batches, model
         )
 
     outputs = []
-
-    for o in out.tolist():
-        outputs.append(remove_duplicate_pitches(o))
         
     print("Done!")
     print_sep()
-    return outputs
+    return out.tolist(
 
 def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens,
                              model_temperature, final_composition, generated_batches, block_lines):