Update components/utils.py

components/utils.py

@@ -14,11 +14,23 @@ import os
 
 def combine_videos(folder_name, length_speech):
 
-    length = len(os.listdir(folder_name))
+    # length = len(os.listdir(folder_name))
 
-    vids = [VideoFileClip(os.path.join(folder_name,x)) for x in sorted(os.listdir(folder_name))]
+    def check_len(dir):
+        if len(dir) < len(length_speech):
+            return check_len(dir*2)
+        else:
+            return dir
 
-    print(vids, length, len(length_speech))
+    vid_dir = os.listdir(folder_name)
+    if len(vid_dir) == 0:
+        raise Exception("Canot find any video! Please try different words")
+        
+    vid_dir = check_len(vid_dir)
+
+    vids = [VideoFileClip(os.path.join(folder_name,x)) for x in sorted(vid_dir)]
+
+    print(vids, len(vid_dir), len(length_speech))
     
     combined = concatenate_videoclips(
         [vid.subclip(0,ls).fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) if vid.duration > ls else vid.fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) for vid, ls in zip(vids, length_speech)],