Update app.py

app.py

@@ -279,17 +279,51 @@ def Generate_Accompaniment(input_midi,
     if generation_type == 'Guided':
 
         input_seq = []
-        
+
         input_seq.extend(start_score_seq)
         input_seq.extend(score_list[0][0])
         
-        for i in tqdm.tqdm(range(len(score_list)-1)):
+        block_seq_lens = []
+        
+        idx = 0
+        
+        max_retries = 3
+        mrt = 0
+        
+        while idx < len(score_list)-1:
+        
+            if idx % 10 == 0:
+                print('Generating', idx, 'block')
+        
+            input_seq.extend(score_list[idx][1])
+        
+            block_seq = []
+        
+            for _ in range(max_retries):
+        
+                block_seq = generate_block_seq(input_seq, score_list[idx+1][0][0])
+        
+                if block_seq:
+                    break
+        
+            if block_seq:
+                input_seq.extend(block_seq)
+                block_seq_lens.append(len(block_seq))
+                idx += 1
+                mrt = 0
+        
+            else:
         
-            input_seq.extend(score_list[i][1])
+                if block_seq_lens:
+                    input_seq = input_seq[:-(block_seq_lens[-1]+2)]
+                    idx -= 1
+                    mrt += 1
         
-            block_seq = generate_block_seq(input_seq, score_list[i+1][0][0], temperature=model_temperature)
+                else:
+                    break
         
-            input_seq.extend(block_seq)
+                if mrt == max_retries:
+                    break
 
     else:
         input_seq = generate_full_seq(start_score_seq, temperature=model_temperature)