Update app.py

app.py

@@ -116,13 +116,22 @@ def generate_midi(temperature, top_k):
             if sequence:
                 midi_events.append(sequence)
 
-    if midi_events:
-        midi_events = max(midi_events, key=len)
-        midi_events = [midi_events]
+    round_bars = []
+    
+    for sequence in midi_events:
+        filtered_sequence = []
+        for event in sequence:
+            if event['start'] < 768 and event['end'] <= 768:
+                filtered_sequence.append(event)
+        if filtered_sequence:
+            round_bars.append(filtered_sequence)
+
+    midi_events = round_bars
 
     return midi_events
 
 
+
 def write_midi(midi_events, bpm):
     midi_data = pretty_midi.PrettyMIDI(initial_tempo=bpm, resolution=96)
     midi_data.time_signature_changes.append(pretty_midi.containers.TimeSignature(4, 4, 0))