Update components/pexels.py

components/pexels.py

@@ -30,13 +30,13 @@ def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoi
 
 
 # Video download function
-def download_video(data, parent_path, height, width, i):
+def download_video(data, parent_path, i):#height, width, i):
     for x in data['videos'][0]['video_files'] :
         
-        if width != None and x['width'] < width:
-            continue
-        if height != None and x['height'] < height :
-            continue
+        #if width != None and x['width'] < width:
+           # continue
+        #if height != None and x['height'] < height :
+            #continue
         
         vid = x
         print(vid['link'])
@@ -48,12 +48,12 @@ def download_video(data, parent_path, height, width, i):
 
 # Utilizing the LLMs to find the relevant videos 
 def generate_videos(product, api_key, orientation, llm_chain=None, sum_llm_chain=None):
-    prod = product.replace(" ", "_")
+    prod = product.strip().replace(" ", "_")
     
     try :
         # Split the paragraph by sentences
             
-        sentences = llm_chain.run(product)
+        sentences = llm_chain.run(product.strip())
         print('Sentence :', sentences)
         
 #         sentences = sentences.split(".")[:-1]
@@ -71,7 +71,7 @@ def generate_videos(product, api_key, orientation, llm_chain=None, sum_llm_chain
             keyword = sum_llm_chain.run(s)
             print(i+1, ":", keyword)
             data = search_pexels(keyword, api_key, orientation.lower())
-            download_video(data, prod, height, width, i)
+            download_video(data, prod, i)
             
         print("Success! videos has been generated")
     except Exception as e :