Update components/pexels.py

components/pexels.py

@@ -2,7 +2,7 @@ import requests
 import shutil,os,re
 
 # Searching for the videos
-def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoint='videos', num_pages=1):
+def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoint='videos', num_pages=50):
     
   if orientation not in ['potrait', 'landscape', 'square']:
     raise Exception("Error! orientation must be one of {'square', 'landscape', 'potrait'}")
@@ -30,27 +30,22 @@ def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoi
 
 
 # Video download function
-def download_video(data, parent_path, links, 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 x['link'] in links:
+def download_video(data, parent_path, height, width, links, i):
+    for x in data['videos'] :
+        if x['id'] in links:
             continue
-            
-        vid = x
-        print(vid['link'])
-        with open(f"{os.path.join(parent_path,str(i) + '_' + str(vid['id']))}.mp4", 'bw') as f:
-            f.write(requests.get(vid['link']).content)
-            print("Sucessfully saved video in", os.path.join(parent_path,str(i) + '_' + str(vid['id'])) + '.mp4')
-            return vid['link']
-    
+        
+        vid = x['video_files']
+        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:
+                    f.write(requests.get(v['link']).content)
+                    print("Sucessfully saved video in", os.path.join(parent_path,str(i) + '_' + str(v['id'])) + '.mp4')
+                    return x['id']
 
 
 # Utilizing the LLMs to find the relevant videos 
-def generate_videos(product, api_key, orientation, llm_chain=None, sum_llm_chain=None):
+def generate_videos(product, api_key, orientation, height, width, llm_chain=None, sum_llm_chain=None):
     prod = product.strip().replace(" ", "_")
     links = []
     try :
@@ -74,7 +69,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())
-            link = download_video(data, prod, links,i)
+            link = download_video(data, prod, height, width, links,i)
             links.append(link)
             
         print("Success! videos has been generated")