Update components/pexels.py

components/pexels.py

@@ -47,7 +47,7 @@ def download_video(data, parent_path, height, width, i):
 
 
 # Utilizing the LLMs to find the relevant videos 
-def generate_videos(product, api_key, height=None, width=None, llm_chain=None, sum_llm_chain=None):
+def generate_videos(product, api_key, orientation, llm_chain=None, sum_llm_chain=None):
     prod = product.replace(" ", "_")
     
     try :
@@ -70,7 +70,7 @@ def generate_videos(product, api_key, height=None, width=None, llm_chain=None, s
         for i,s in enumerate(sentences):
             keyword = sum_llm_chain.run(s)
             print(i+1, ":", keyword)
-            data = search_pexels(keyword, api_key, )
+            data = search_pexels(keyword, api_key, orientation.lower())
             download_video(data, prod, height, width, i)
             
         print("Success! videos has been generated")