Update app.py

app.py

@@ -113,7 +113,7 @@ def load_model():
 
 #==================================================================================
 
-def load_midi(input_midi, model_selector=''):
+def load_midi(input_midi, melody_patch=-1):
 
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi.name)
 
@@ -121,27 +121,38 @@ def load_midi(input_midi, model_selector=''):
     escore_notes = TMIDIX.augment_enhanced_score_notes(escore_notes, timings_divider=32)
     
     sp_escore_notes = TMIDIX.solo_piano_escore_notes(escore_notes, keep_drums=False)
-    zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
+
+    if melody_patch == -1:
+        zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
+
+    else:
+        zscore = TMIDIX.recalculate_score_timings([e for e in sp_escore_notes if e[6] == melody_patch])
     
     cscore = TMIDIX.chordify_score([1000, zscore])
     
     score = []
     
+    score_list = []
+    
     pc = cscore[0]
     
     for c in cscore:
         score.append(max(0, min(127, c[0][1]-pc[0][1])))
     
-        for n in c:
-            if model_selector == 'with velocity - 3 epochs':
-                score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256, max(1, min(127, n[5]))+384])
-
-            else:
-                score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256])
+        scl = [[max(0, min(127, c[0][1]-pc[0][1]))]]
+    
+        n = c[0]
+        
+        score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256])
+        scl.append([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256])
+    
+        score_list.append(scl)
     
         pc = c
+    
+    score_list.append(scl)
 
-    return score
+    return score, score_list
 
 #==================================================================================