Update app.py

app.py

@@ -16,6 +16,10 @@ import TMIDIX
 
 import matplotlib.pyplot as plt
 
+import time
+import datetime
+from pytz import timezone
+
 in_space = os.getenv("SYSTEM") == "spaces"
 
 
@@ -24,9 +28,13 @@ in_space = os.getenv("SYSTEM") == "spaces"
 @torch.no_grad()
 def GenerateMIDI(num_tok, idrums, iinstr):
     print('=' * 70)
-    print('Req num tok', num_tok)
-    print('Req instr', iinstr)
-    print('Drums', idrums)
+    print('Req time: {:%Y-%m-%d %H:%M:%S}'.format(datetime.datetime.now(PDT)))
+    start_time = time.time()
+
+    print('=' * 70)
+    print('Req num tok:', num_tok)
+    print('Req instr:', iinstr)
+    print('Drums:', idrums)
     print('=' * 70)
 
     if idrums:
@@ -77,7 +85,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     yield output, None, None, [create_msg("visualizer_clear", None)]
 
     outy = start_tokens
-    time = 0
+    ctime = 0
     dur = 0
     vel = 0
     pitch = 0
@@ -97,7 +105,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
         ss1 = int(out0[0])
 
         if 0 < ss1 < 256:
-            time += ss1 * 8
+            ctime += ss1 * 8
 
         if 256 <= ss1 < 1280:
             dur = ((ss1 - 256) // 8) * 32
@@ -106,7 +114,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
         if 1280 <= ss1 < 2816:
             channel = (ss1 - 1280) // 128
             pitch = (ss1 - 1280) % 128
-            event = ['note', int(time), int(dur), int(channel), int(pitch), int(vel)]
+            event = ['note', ctime, dur, channel, pitch, vel]
             output[-1].append(event)
 
             yield output, None, None, [create_msg("visualizer_append", event), create_msg("progress", [i + 1, num_tok])]
@@ -117,7 +125,13 @@ def GenerateMIDI(num_tok, idrums, iinstr):
         f.write(midi_data)
 
     audio = synthesis(TMIDIX.score2opus(output), 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2')
-
+    
+    print('=' * 70)
+    print('Sample output INTs', outy[:16])
+    print('=' * 70)
+    print('Req execution time:', (time.time() - start_time), 'sec')
+    print('=' * 70)
+    
     yield output, "Allegro-Music-Transformer-Music-Composition.mid", (44100, audio), [
         create_msg("visualizer_end", None)]
 
@@ -176,10 +190,13 @@ def create_msg(name, data):
 
 
 if __name__ == "__main__":
+    print('=' * 70)
     parser = argparse.ArgumentParser()
     parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
     parser.add_argument("--port", type=int, default=7860, help="gradio server port")
     opt = parser.parse_args()
+    
+    PDT = timezone('US/Pacific')
 
     print('Loading model...')