Update components/pexels.py

components/pexels.py

@@ -2,7 +2,7 @@ import requests
 import shutil,os,re
 import datetime
 import torch
-import soundfile as SF
+import soundfile as sf
 
 from components.custom_llm import custom_chain
 
@@ -41,6 +41,7 @@ def download_video(data, parent_path, height, width, links, i):
             continue
         
         vid = x['video_files']
+        print(vid)
         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", 'bw') as f:
@@ -77,7 +78,8 @@ def generate_videos(text, api_key, orientation, height, width, model, tokenizer,
         # Split the paragraph by sentences
         # 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(text))]))
+        # sentences = list(filter(None,[x.strip() for x in re.split(r'[^A-Za-z -]', custom_chain().invoke(text))]))
+        sentences = list([x.split('-')[0].strip() for x in filter(lambda x:'-' in x,re.split(r'[^A-Za-z -]', cap_chain.invoke(text)))])
         
         # Create directory with the name
         di = str(datetime.datetime.now())