write and upload video

1 files changed, 27 insertions, 16 deletions

diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index 8038e72..6730c96 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -413,31 +413,45 @@ class Listener:
     tag = "biggan_" + timestamp()
     # path_out = os.path.join(app_cfg.DIR_RESULTS, tag)
     # os.makedirs(path_out, exist_ok=True)
-    gen_time_total = 0
-    to_pil_time_total = 0
-    save_time_total = 0
-    resize_time_total = 0
-    send_time_total = 0
-    p = Popen([
+    fp_out = os.path.join(app_cfg.DIR_RENDERS, '{}.mp4'.format(tag))
+    pipe = Popen([
       'ffmpeg',
       '-y',
       '-f', 'image2pipe',
       '-vcodec', 'png',
       '-r', '25',
       '-i', '-',
-  
       '-c:v', 'libx264',
       '-preset', 'slow',
       '-crf', '19',
       '-vf', 'fps=25',
       '-pix_fmt', 'yuv420p',
       '-s', '512x512',
-
-      # '-vcodec', 'mpeg4',
-      # '-qscale', '5',
-      '-r', '25', '{}.mp4'.format(tag)
+      '-r', '25',
+      fp_out
     ], stdin=PIPE, stdout=PIPE)
 
+    # catch interrupts
+    try:
+      self.run(interpolator, pipe)
+    except e:
+      pass
+    self.rpc_client.send_status('processing', False)
+    self.sess.close()
+    print("Writing video...")
+    pipe.stdin.close()
+    pipe.wait()
+    print("Uploading video...")
+    data = upload_file_to_cortex(opt_folder_id, fp_out, datatype='video', activity='live')
+    print(json.dumps(data, indent=2))
+    print("Done")
+
+  def run(self, interpolator, pipe):
+    gen_time_total = 0
+    to_pil_time_total = 0
+    save_time_total = 0
+    resize_time_total = 0
+    send_time_total = 0
     for i in range(99999):
       if i == 0:
         print("Loading network...")
@@ -463,7 +477,7 @@ class Listener:
       if out_img is not None:
         save_time = time.time()
         # out_img.save(os.path.join(path_out, "frame_{:05d}.png".format(i)), format='png', compression_level=3)
-        out_img.save(p.stdin, format='png', compression_level=3)
+        out_img.save(pipe.stdin, format='png', compression_level=3)
         save_time_total += time.time() - save_time
 
         resize_time = time.time()
@@ -481,7 +495,4 @@ class Listener:
         send_time_total += time.time() - send_time
       if (i % 100) == 0 or i == 1:
         print("gen: {:2f}, pil: {:2f}, save: {:2f}, resize: {:2f}, send: {:2f}".format(gen_time_total / i, to_pil_time_total / i, save_time_total / i, resize_time_total / i, send_time_total / i))
-    p.stdin.close()
-    p.wait()
-    self.rpc_client.send_status('processing', False)
-    self.sess.close()
+