Update app.py

app.py

@@ -7,6 +7,8 @@ import time as reqtime
 import datetime
 from pytz import timezone
 
+import tqdm
+
 import spaces
 import gradio as gr
 
@@ -27,6 +29,8 @@ def Generate_Rock_Song(input_midi,
                        input_use_original_durations,
                        input_match_original_pitches_counts
                       ):
+
+    #===============================================================================
     
     print('=' * 70)
     print('Req start time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
@@ -46,6 +50,8 @@ def Generate_Rock_Song(input_midi,
     print('Match original pitches counts:', input_match_original_pitches_counts)
     print('=' * 70)
 
+    #===============================================================================
+
     print('Loading model...')
 
     SEQ_LEN = 4096
@@ -83,6 +89,10 @@ def Generate_Rock_Song(input_midi,
 
     print('Done!')
     print('=' * 70)
+
+    #===============================================================================
+
+    print('Loading MIDI...')
     
     #===============================================================================
     # Raw single-track ms score
@@ -169,12 +179,10 @@ def Generate_Rock_Song(input_midi,
         control_toks.append(ctoks)
         prime_toks.append(ptoks)
         
-    print('Input melody seed number:', input_melody_seed_number)
-    print('-' * 70)
+    print('Done!')
+    print('=' * 70)
 
     #==================================================================
-
-    print('=' * 70)
     
     print('Sample output events', prime_toks[:16])
     print('=' * 70)
@@ -235,28 +243,34 @@ def Generate_Rock_Song(input_midi,
 
     #==================================================================
 
-    num_prime_chords = 128
-    pass_chan_dur_tok = False
-    match_ptcs_counts = False
-    
     song = []
     
-    for i in range(num_prime_chords):
-        song.extend(prime_toks[i])
-    
-    for i in tqdm.tqdm(range(num_prime_chords, len(score_toks))):
+    if input_freestyle_continuation:
+        output = generate_continuation(512, 1024)
+        song.extend(output)
+
+    else:
+
+        for i in range(input_number_prime_chords):
+            song.extend(prime_toks[i])
+        
+        for i in tqdm.tqdm(range(input_number_prime_chords, len(score_toks))):
+            
+            song.extend(score_toks[i])
         
-        song.extend(score_toks[i])
-    
-        if control_toks[i]:
-            for ct in control_toks[i]:
-                if pass_chan_dur_tok:
-                    song.append(ct[0])
-                if match_ptcs_counts:
-                    out_seq = generate_tokens(song, ct[1])
-                else:
-                    out_seq = generate_tokens(song)
-                song.extend(out_seq)
+            if control_toks[i]:
+                for ct in control_toks[i]:
+                    
+                    if input_use_original_durations:
+                        song.append(ct[0])
+                        
+                    if input_match_original_pitches_counts:
+                        out_seq = generate_tokens(song, ct[1])
+                        
+                    else:
+                        out_seq = generate_tokens(song)
+                        
+                    song.extend(out_seq)
 
     #==================================================================