Update app.py

app.py

@@ -9,6 +9,7 @@ print('=' * 70)
 print('Loading core Guided Accompaniment Transformer modules...')
 
 import os
+import copy
 
 import time as reqtime
 import datetime
@@ -155,6 +156,81 @@ def Generate_Accompaniment(input_midi,
                           ):
 
     #===============================================================================
+
+    def generate_full_seq(input_seq, temperature=0.9, verbose=True):
+    
+        seq_abs_run_time = sum([t for t in input_seq if t < 128])
+    
+        cur_time = 0
+    
+        full_seq = input_seq
+    
+        toks_counter = 0
+    
+        while cur_time < seq_abs_run_time:
+    
+            if verbose:
+                if toks_counter % 128 == 0:
+                    print('Generated', toks_counter, 'tokens')
+    
+            x = torch.LongTensor(full_seq).cuda()
+    
+            with ctx:
+                out = model.generate(x,
+                                     1,
+                                     temperature=temperature,
+                                     return_prime=False,
+                                     verbose=False)
+            
+            y = out.tolist()[0][0]
+    
+            if y < 128:
+                cur_time += y
+    
+            full_seq.append(y)
+    
+            toks_counter += 1
+    
+        return full_seq
+
+    #===============================================================================
+
+    def generate_block_seq(input_seq, trg_dtime, temperature=0.9):
+    
+        cur_time = 0
+    
+        block_seq = [128]
+    
+        while cur_time != trg_dtime and len(block_seq) < 2 and block_seq[-1] > 127:
+    
+            inp_seq = copy.deepcopy(input_seq)
+        
+            block_seq = []
+        
+            cur_time = 0
+        
+            while cur_time < trg_dtime:
+        
+                x = torch.LongTensor(inp_seq).cuda()
+                
+                with ctx:
+                    out = model.generate(x,
+                                         1,
+                                         temperature=temperature,
+                                         return_prime=False,
+                                         verbose=False)
+                
+                y = out.tolist()[0][0]
+        
+                if y < 128:
+                    cur_time += y
+        
+                inp_seq.append(y)
+                block_seq.append(y)
+    
+        return block_seq
+
+    #===============================================================================
     
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -178,9 +254,10 @@ def Generate_Accompaniment(input_midi,
 
     score, score_list = load_midi(input_midi.name)
 
+    print('Sample score events', score[:12])
+
     #==================================================================
     
-    print('Sample score events', score[:12])
     print('=' * 70)
     print('Generating...')