Update app.py

app.py

@@ -109,9 +109,9 @@ print('=' * 70)
 def load_midi(input_midi):
 
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi)
-
+    
     escore_notes = TMIDIX.advanced_score_processor(raw_score, return_enhanced_score_notes=True, apply_sustain=True)
-
+    
     if escore_notes:
     
         escore_notes = TMIDIX.augment_enhanced_score_notes(escore_notes[0], sort_drums_last=True)
@@ -121,6 +121,8 @@ def load_midi(input_midi):
         dcscore = TMIDIX.chordify_score([d[1:] for d in dscore])
         
         melody_chords = [18816]
+    
+        chords = []
         
         #=======================================================
         # MAIN PROCESSING CYCLE
@@ -131,6 +133,10 @@ def load_midi(input_midi):
             delta_time = c[0][0]
         
             melody_chords.append(delta_time)
+    
+            cho = []
+            
+            cho.append(delta_time)
         
             for e in c:
             
@@ -160,14 +166,17 @@ def load_midi(input_midi):
                 dur_vel = (8 * dur) + velocity
         
                 melody_chords.extend([pat_ptc+256, dur_vel+16768]) # 18816
+                cho.extend([pat_ptc+256, dur_vel+16768])
     
+            chords.append(cho)
         
         print('Done!')
         print('=' * 70)
-        print('Score hss', len(melody_chords), 'tokens')
+        print('Score has', len(melody_chords), 'tokens')
+        print('Score has', len(chords), 'chords')
         print('=' * 70)
 
-        return melody_chords
+        return melody_chords, chords
 
     else:
         return None
@@ -176,9 +185,9 @@ def load_midi(input_midi):
 
 @spaces.GPU
 def Generate_Drums(input_midi,
-                          model_temperature,
-                          model_sampling_top_p
-                         ):
+                   model_temperature,
+                   model_sampling_top_p
+                  ):
 
     #===============================================================================