Update app.py

app.py

@@ -171,44 +171,54 @@ def GenerateBridge(input_midi, input_start_note):
     print('=' * 70)
     print('Generating...')
 
-    x = (torch.tensor(seed_melody, dtype=torch.long, device='cuda')[None, ...])
-
+    x = (torch.tensor(td, dtype=torch.long, device=DEVICE)[None, ...])
+    
     with ctx:
       out = model.generate(x,
-                          1536,
+                          1032,
                           temperature=0.9,
                           return_prime=False,
                           verbose=False)
     
-    output = out[0].tolist()
+    y = out.tolist()
+  
+    output = []
+    
+    for i in range(0, len(y[0]), 5):
+        if len(y[0][i:i+5]) == 5:
+            output.append(y[0][i:i+5])
         
     print('=' * 70)
     print('Done!')
     print('=' * 70)
+
+    start_note_idx = output.index(start_note)
+    end_note_idx = len(output)-output[::-1].index(end_note)-1
+    
+    print('Start note check:', start_note in output, '---', start_note_idx)
+    print('End note check:',end_note in output, '---', end_note_idx)
     
     #===============================================================================
     print('Rendering results...')
+
+    data = score_chunk[:STEP] + TMIDIX.flatten(output[:end_note_idx]) + score_chunk[-STEP:]
     
     print('=' * 70)
-    print('Sample INTs', output[:15])
+    print('Sample INTs', data[:15])
     print('=' * 70)
-    
-    out1 = output
 
-    if len(out1) != 0:
+    if len(data) != 0:
     
-        song = out1
+        song = data
         song_f = []
     
         time = 0
         dur = 0
         vel = 90
         pitch = 0
+        pat = 0
         channel = 0
     
-        patches = [0] * 16
-        patches[3] = 40
-    
         for ss in song:
     
             if 0 < ss < 128:
@@ -219,26 +229,33 @@ def GenerateBridge(input_midi, input_start_note):
     
                 dur = (ss-128) * 32
     
-            if 256 < ss < 512:
+            if 256 <= ss <= 384:
+    
+                pat = (ss-256)
+                channel = pat // 8
+    
+                if channel == 9:
+                    channel = 15
+                if channel == 16:
+                    channel = 9
+    
+            if 384 < ss < 640:
     
-                pitch = (ss-256) % 128
+                pitch = (ss-384) % 128
     
-                channel = (ss-256) // 128
+            if 640 <= ss < 648:
     
-                if channel == 1:
-                    channel = 3
-                    vel = 110 + (pitch % 12)
-                    song_f.append(['note', time, dur, channel, pitch, vel, 40])
-                    
-                else:
-                    vel = 80 + (pitch % 12)
-                    channel = 0
-                    song_f.append(['note', time, dur, channel, pitch, vel, 0])
+                vel = ((ss-640)+1) * 15
+    
+                song_f.append(['note', time, dur, channel, pitch, vel, pat])
+    
+    song_f, patches, overflow_patches = TMIDIX.patch_enhanced_score_notes(song_f)
+
 
-    fn1 = "Melody2Song-Seq2Seq-Music-Transformer-Composition"
+    fn1 = "Bridge-Music-Transformer-Composition"
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
-                                                              output_signature = 'Melody2Song Seq2Seq Music Transformer',
+                                                              output_signature = 'Bridge Music Transformer',
                                                               output_file_name = fn1,
                                                               track_name='Project Los Angeles',
                                                               list_of_MIDI_patches=patches