Update components/pexels.py

components/pexels.py

@@ -79,7 +79,7 @@ def generate_videos(text, api_key, orientation, height, width, model, tokenizer,
         # 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([x.split('-')[0].strip() for x in filter(lambda x:'-' in x,re.split(r'[^A-Za-z -]', custom_chain().invoke(text)))])
+        sentences = [x.split('-')[0].strip() for x in filter(lambda x:'-' in x,re.split(r'[^A-Za-z -]', custom_chain().invoke(text)))]
         
         # Create directory with the name
         di = str(datetime.datetime.now())
@@ -90,6 +90,8 @@ def generate_videos(text, api_key, orientation, height, width, model, tokenizer,
         # Generate video for every sentence
         print("Keyword :")
         for i,s in enumerate(sentences):
+            if s=='':
+                s='videos'
             # keyword = sum_llm_chain.run(s)
             print(i+1, ":", s)
             data = search_pexels(s, api_key, orientation.lower())