Update components/pexels.py

components/pexels.py

@@ -32,10 +32,12 @@ def download_video(data, parent_path, height, width, links, i):
         vid = x['video_files']
         for v in vid: 
             if v['height'] == height and v['width'] == width:
-                with open(f"{os.path.join(parent_path, str(i) + '_' + str(v['id']))}.mp4", 'wb') as f:
+                video_id = str(x['id'])
+                video_path = os.path.join(parent_path, f"{i}_{video_id}.mp4")
+                with open(video_path, 'wb') as f:
                     f.write(requests.get(v['link']).content)
-                    print("Successfully saved video in", os.path.join(parent_path, str(i) + '_' + str(v['id'])) + '.mp4')
-                    return x['id']
+                    print("Successfully saved video in", video_path)
+                    return video_id
 
 # Utilizing the LLMs to find the relevant videos 
 def generate_videos(product, api_key, orientation, height, width, llm_chain=None, sum_llm_chain=None):
@@ -46,13 +48,14 @@ def generate_videos(product, api_key, orientation, height, width, llm_chain=None
         sentences = llm_chain.run(product.strip())
         print('Sentences:', sentences)
         
-        # sentences = sentences.split(".")[:-1]
-        sentences = [x.strip() for x in re.split(r'\d+\.', sentences) if len(x) > 6]
+        # Tách các câu từ văn bản
+        sentences = [x.strip() for x in re.split(r'[.!?]', sentences) if len(x.strip()) > 6]
     
-        # Create directory with the product's name
-        if os.path.exists(prod):
-            shutil.rmtree(prod)
-        os.mkdir(prod)
+        # Sử dụng UUID để tạo tên thư mục ngắn gọn
+        folder_name = f"video_{uuid.uuid4().hex}"
+        os.makedirs(folder_name, exist_ok=True)
+        folder_path = os.path.join(folder_name, "images")
+        os.makedirs(folder_path, exist_ok=True)
         
         # Generate video for every sentence
         print("Keywords:")
@@ -60,12 +63,13 @@ def generate_videos(product, api_key, orientation, height, width, llm_chain=None
             keyword = sum_llm_chain.run(s)
             print(i + 1, ":", keyword)
             data = search_pexels(keyword, api_key, orientation.lower())
-            link = download_video(data, prod, height, width, links, i)
-            links.append(link)
+            link = download_video(data, folder_path, height, width, links, i)
+            if link:
+                links.append(link)
             
         print("Success! Videos have been generated")
     except Exception as e:
         print("Error! Failed generating videos")
         print(e)
     
-    return prod, sentences
+    return folder_name, sentences