Add list of vid links to prevent a same video being loaded

components/pexels.py

@@ -30,26 +30,29 @@ def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoi
 
 
 # Video download function
-def download_video(data, parent_path, i):#height, width, i):
+def download_video(data, parent_path, link, 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:
+            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')
-            break
+            return vid['link']
+    
 
 
 # Utilizing the LLMs to find the relevant videos 
 def generate_videos(product, api_key, orientation, llm_chain=None, sum_llm_chain=None):
     prod = product.strip().replace(" ", "_")
-    
+    links = []
     try :
         # Split the paragraph by sentences
             
@@ -71,7 +74,8 @@ 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, i)
+            link= download_video(data, prod, links,i)
+            links.append(link)
             
         print("Success! videos has been generated")
     except Exception as e :