dirz

1 files changed, 13 insertions, 8 deletions

diff --git a/cli/app/commands/cortex/upload_video.py b/cli/app/commands/cortex/upload_video.py
index f8b9686..6c46d0c 100644
--- a/cli/app/commands/cortex/upload_video.py
+++ b/cli/app/commands/cortex/upload_video.py
@@ -19,15 +19,20 @@ def cli(ctx, opt_fp_in, opt_folder_id):
     opt_folder_id = folder['id']
 
   if opt_fp_in is None:
-    results_dir = os.path.join(app_cfg.RESULTS_DIR)
-    for fn in sorted(os.listdir(app_cfg.RESULTS_DIR), reverse=True):
-      fp_frames = os.path.join(app_cfg.RESULTS_DIR, fn)
-      if os.path.isdir(f):
+    for fn in sorted(os.listdir(app_cfg.DIR_RESULTS), reverse=True):
+      fp_frames = os.path.join(app_cfg.DIR_RESULTS, fn)
+      if os.path.isdir(fp_frames):
+        opt_fp_in = fp_frames
         break
-    fp_out = os.path.join(app_cfg.DIR_RENDERS, fn + '.mp4')
-    export_video_final(fp_frames, fp_out)
-  else:
-    fp_out = opt_fp_in
+
+  frames = os.listdir(opt_fp_in)
+  if len(frames) < 10:
+    print("Didn't run long enough, skipping")
+    return
+
+  fp_out = os.path.join(app_cfg.DIR_RENDERS, fn + '.mp4')
+  export_video_final(opt_fp_in, fp_out)
+
   if not os.path.exists(fp_out):
     print("No video found")
     return