Update app.py

app.py

@@ -66,6 +66,7 @@ def GenerateSong(input_melody_seed_number):
 
     print('Done!')
     print('=' * 70)
+    seed_melody = seed_melodies_data[input_melody_seed_number]
     print('Input melody seed number:', input_melody_seed_number)
     print('-' * 70)
 
@@ -73,61 +74,21 @@ def GenerateSong(input_melody_seed_number):
 
     print('=' * 70)
     
-    print('Sample output events', melody_chords[:5])
+    print('Sample output events', seed_melody[:16])
     print('=' * 70)
     print('Generating...')
 
-    output = []
-    
-    max_chords_limit = 8
-    temperature=0.9
-    num_memory_tokens=4096
+    x = (torch.tensor(melody, dtype=torch.long, device='cuda')[None, ...])
 
-    output = []
-    
-    idx = 0
-    
-    for c in chords[:input_num_tokens]:
-     
-        output.append(c)
+    with ctx:
+      out = model.generate(x,
+                          1536,
+                          temperature=0.9,
+                          return_prime=False,
+                          verbose=False)
     
-        if input_conditioning_type == 'Chords-Times' or input_conditioning_type == 'Chords-Times-Durations':
-          output.append(times[idx])
-    
-        if input_conditioning_type == 'Chords-Times-Durations':
-          output.append(durs[idx])
-
-        x = torch.tensor([output] * 1, dtype=torch.long, device='cuda')
-        
-        o = 0
-        
-        ncount = 0
-        
-        while o < 384 and ncount < max_chords_limit:
-          with ctx:
-            out = model.generate(x[-num_memory_tokens:],
-                                1,
-                                temperature=temperature,
-                                return_prime=False,
-                                verbose=False)
-        
-          o = out.tolist()[0][0]
+    output = out.tolist()
         
-          if 256 <= o < 384:
-            ncount += 1
-        
-          if o < 384:
-            x = torch.cat((x, out), 1)
-        
-        outy =  x.tolist()[0][len(output):]
-          
-        output.extend(outy)
-    
-        idx += 1
-
-        if idx == len(chords[:input_num_tokens])-1:
-            break
-
     print('=' * 70)
     print('Done!')
     print('=' * 70)
@@ -136,7 +97,7 @@ def GenerateSong(input_melody_seed_number):
     print('Rendering results...')
     
     print('=' * 70)
-    print('Sample INTs', output[:12])
+    print('Sample INTs', output[:15])
     print('=' * 70)
     
     out1 = output
@@ -151,33 +112,40 @@ def GenerateSong(input_melody_seed_number):
         vel = 90
         pitch = 0
         channel = 0
-        
+    
         patches = [0] * 16
-        
-        channel = 0
-        
+        patches[3] = 40
+    
         for ss in song:
-        
-          if 0 <= ss < 128:
-        
-              time += ss * 32
-        
-          if 128 <= ss < 256:
-        
-              dur = (ss-128) * 32
-        
-          if 256 <= ss < 384:
-        
-              pitch = (ss-256)
-        
-              vel = max(40, pitch)
-        
-              song_f.append(['note', time, dur, channel, pitch, vel, 0])
+    
+            if 0 < ss < 128:
+    
+                time += (ss * 32)
+    
+            if 128 < ss < 256:
+    
+                dur = (ss-128) * 32
+    
+            if 256 < ss < 512:
+    
+                pitch = (ss-256) % 128
+    
+                channel = (ss-256) // 128
+    
+                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])
 
-    fn1 = "Chords-Progressions-Transformer-Composition"
+    fn1 = "Melody2Song-Seq2Seq-Music-Transformer-Composition"
     
     detailed_stats = TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(song_f,
-                                                              output_signature = 'Chords Progressions Transformer',
+                                                              output_signature = 'Melody2Song Seq2Seq Music Transformer',
                                                               output_file_name = fn1,
                                                               track_name='Project Los Angeles',
                                                               list_of_MIDI_patches=patches