Update app.py

app.py

@@ -102,9 +102,12 @@ def auto_extract_melody(mid, debug=False):
         print(f"Melody Track: {melody_track} | Program: {melody_program}")
 
     if melody_program is not None:
+        all_events_copy = events.copy()
         events, melody = extract_instruments(events, [melody_program])
-        if debug:
-            print(f"Extracted {len(melody)} melody events from program {melody_program}")
+
+        for e in all_events_copy:
+            if hasattr(e, "program") and e.program == melody_program:
+                events.append(e)
     else:
         if debug:
             print("No program number found; using all events as melody.")