From 3a54f34f6cbf8f8bcab003c9f9d91ad4f61d030f Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@gmail.com>
Date: Sat, 6 Jul 2024 23:09:43 +0300
Subject: [PATCH] [~] Refactor

---
 python/tasks/tiktok/utils.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/python/tasks/tiktok/utils.py b/python/tasks/tiktok/utils.py
index aecbee0..e5f6e89 100644
--- a/python/tasks/tiktok/utils.py
+++ b/python/tasks/tiktok/utils.py
@@ -1,4 +1,5 @@
 import celery.app.task
+import os
 import asyncio
 import inspect
 import importlib
@@ -141,11 +142,13 @@ def grid_of_videos(paths: Iterable[str]) -> Any:
 
     grid = GridspecLayout(1, len(paths))
 
-    for i, filepath in enumerate(paths):
+    for i, path in enumerate(paths):
+        assert os.path.exists(path)
+
         out = Output()
         with out:
             display.display(display.Video(
-                url='/files/%s' % filepath,
+                url='/files/%s' % path,
                 #embed=True
             ))