Update components/pexels.py

components/pexels.py

@@ -2,7 +2,9 @@ import requests
 import shutil,os,re
 import datetime
 import torch
-import soundfile as sf
+import soundfile as SF
+
+from custom_llm import custom_chain
 
 # Searching for the videos
 def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoint='videos', num_pages=50, pages=4):
@@ -73,8 +75,9 @@ def generate_videos(text, api_key, orientation, height, width, model, tokenizer,
     links = []
     try :
         # Split the paragraph by sentences
-        sentences = list(filter(None,[x.strip() for x in re.split(r'[^A-Za-z0-9 -]', text)]))
+        # sentences = list(filter(None,[x.strip() for x in re.split(r'[^A-Za-z0-9 -]', text)]))
         # print(len(sentences))
+        sentences = list(filter(None,[x.strip() for x in re.split(r'[^A-Za-z -]', custom_chain().invoke(p))]))
         
         # Create directory with the name
         di = str(datetime.datetime.now())