Update app.py

app.py

@@ -156,58 +156,6 @@ def load_midi(input_midi, melody_patch=-1):
 
 #==================================================================================
 
-def save_midi(tokens, batch_number=None, model_selector=''):
-
-    song = tokens
-    song_f = []
-    
-    time = 0
-    dur = 0
-    vel = 90
-    pitch = 0
-    channel = 0
-    patch = 0
-
-    patches = [0] * 16
-
-    for m in song:
-
-        if 0 <= m < 128:
-            time += m * 32
-
-        elif 128 < m < 256:
-            dur = (m-128) * 32
-
-        elif 256 < m < 384:
-            pitch = (m-256)
-
-            if model_selector == 'without velocity - 3 epochs' or model_selector == 'without velocity - 7 epochs':
-                song_f.append(['note', time, dur, 0, pitch, max(40, pitch), 0])
-
-        elif 384 < m < 512:
-            vel = (m-384)
-
-            if model_selector == 'with velocity - 3 epochs':
-                song_f.append(['note', time, dur, 0, pitch, vel, 0])
-
-    if batch_number == None:
-        fname = 'Guided-Accompaniment-Transformer-Music-Composition'
-        
-    else:
-        fname = 'Guided-Accompaniment-Transformer-Music-Composition_'+str(batch_number)
-    
-    data = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
-                                                  output_signature = 'Guided Accompaniment Transformer',
-                                                  output_file_name = fname,
-                                                  track_name='Project Los Angeles',
-                                                  list_of_MIDI_patches=patches,
-                                                  verbose=False
-                                                  )
-
-    return song_f
-
-#==================================================================================
-
 @spaces.GPU
 def Generate_Accompaniment(input_midi, 
                            generation_type,
@@ -269,39 +217,40 @@ def Generate_Accompaniment(input_midi,
         time = 0
         dur = 0
         vel = 90
-        pitch = 60
+        pitch = 0
         channel = 0
         patch = 0
-        
-        patches = [0] * 16
-        
-        for ss in final_song:
-        
-            if 0 <= ss < 256:
-        
-                time += ss * 16
-        
-            if 256 <= ss < 384:
-        
-                pitch = ss-256
-        
-            if 384 <= ss < 640:
-        
-                dur = (ss-384) * 16
-        
-            if 640 <= ss < 768:
-                
-                vel = (ss-640)
-            
+    
+        channels_map = [0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 9, 12, 13, 14, 15]
+        patches_map = [40, 0, 10, 19, 24, 35, 40, 52, 56, 9, 65, 73, 0, 0, 0, 0]
+        velocities_map = [125, 80, 100, 80, 90, 100, 100, 80, 110, 110, 110, 110, 80, 80, 80, 80]
+    
+        for m in final_song:
+    
+            if 0 <= m < 128:
+                time += m * 32
+    
+            elif 128 < m < 256:
+                dur = (m-128) * 32
+    
+            elif 256 < m < 1792:
+                cha = (m-256) // 128
+                pitch = (m-256) % 128
+    
+                channel = channels_map[cha]
+                patch = patches_map[channel]
+                vel = velocities_map[channel]
+    
+                    
                 song_f.append(['note', time, dur, channel, pitch, vel, patch])
 
-    fn1 = "Score-2-Performance-Transformer-Composition"
+    fn1 = "Guided-Accompaniment-Transformer-Composition"
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
-                                                              output_signature = 'Score 2 Performance Transformer',
+                                                              output_signature = 'Guided Accompaniment Transformer',
                                                               output_file_name = fn1,
                                                               track_name='Project Los Angeles',
-                                                              list_of_MIDI_patches=patches
+                                                              list_of_MIDI_patches=patches_map
                                                               )
     
     new_fn = fn1+'.mid'
@@ -319,19 +268,14 @@ def Generate_Accompaniment(input_midi,
 
     #========================================================
 
-    output_midi_title = str(fn1)
-    output_midi_summary = str(song_f[:3])
     output_midi = str(new_fn)
     output_audio = (16000, audio)
     
     output_plot = TMIDIX.plot_ms_SONG(song_f, plot_title=output_midi, return_plt=True)
 
     print('Output MIDI file name:', output_midi)
-    print('Output MIDI title:', output_midi_title)
-    print('Output MIDI summary:', output_midi_summary)
     print('=' * 70) 
     
-
     #========================================================
     
     print('-' * 70)