Update components/pexels.py

components/pexels.py

@@ -1,81 +1,89 @@
 import requests
-import shutil,os,re
+import shutil
+import os
+import re
 
 # Searching for the videos
-def search_pexels(keyword, api_key, orientation='potrait', size='medium', endpoint='videos', num_pages=50):
+def search_pexels(keyword, api_key, orientation='portrait', size='medium', endpoint='videos', num_pages=50):
+    if orientation not in ['portrait', 'landscape', 'square']:
+        raise ValueError("Error! orientation must be one of {'portrait', 'landscape', 'square'}")
+    if size not in ['medium', 'small', 'large']:
+        raise ValueError("Error! size must be one of ['medium', 'small', 'large']")
     
-  if orientation not in ['potrait', 'landscape', 'square']:
-    raise Exception("Error! orientation must be one of {'square', 'landscape', 'potrait'}")
-
-  if size not in ['medium', 'small', 'large']:
-    raise Exception("Error! size must be one of ['medium', 'small', 'large']")
-
-  base_url = 'https://api.pexels.com/'
-
-  headers = {
-      'Authorization': f'{api_key}'
-  }
-
-  url = f'{base_url}{endpoint}/search?query={keyword}&per_page={num_pages}&orientation={orientation}&size={size}'
-
-
-  response = requests.get(url, headers=headers)
-
-  # Check if request was successful (status code 200)
-  if response.status_code == 200:
-      data = response.json()
-      return data
-  else:
-      print(f'Error: {response.status_code}')
+    base_url = 'https://api.pexels.com/'
+    headers = {'Authorization': api_key}
+    url = f'{base_url}{endpoint}/search?query={keyword}&per_page={num_pages}&orientation={orientation}&size={size}'
 
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        data = response.json()
+        return data
+    else:
+        print(f'Error: {response.status_code}')
+        return None
 
 # Video download function
-def download_video(data, parent_path, height, width, links, i):
-    for x in data['videos'] :
-        if x['id'] in links:
+def download_video(video_data, parent_path, height, width, links, index):
+    for video in video_data['videos']:
+        if video['id'] in links:
             continue
         
-        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']
-
+        for video_file in video['video_files']:
+            if video_file['height'] == height and video_file['width'] == width:
+                # Stream the video to avoid memory issues
+                video_url = video_file['link']
+                file_path = os.path.join(parent_path, f"{index}_{video['id']}.mp4")
+                
+                with requests.get(video_url, stream=True) as r:
+                    r.raise_for_status()
+                    with open(file_path, 'wb') as f:
+                        for chunk in r.iter_content(chunk_size=8192):
+                            f.write(chunk)
+                print(f"Successfully saved video in {file_path}")
+                return video['id']
+    return None
 
-# Utilizing the LLMs to find the relevant videos 
-def generate_videos(product, api_key, orientation, height, width, llm_chain=None, sum_llm_chain=None):
+# Generate videos without LLM dependency
+def generate_videos(product, api_key, orientation, height, width):
     prod = product.strip().replace(" ", "_")
     links = []
-    try :
-        # Split the paragraph by sentences
-            
-        sentences = llm_chain.run(product.strip())
-        print('Sentence :', sentences)
-        
-#         sentences = sentences.split(".")[:-1]
-        sentences = [x.strip() for x in re.split(r'\d+\.', sentences) if len(x) > 6]
     
-        
+    try:
         # Create directory with the product's name
         if os.path.exists(prod):
             shutil.rmtree(prod)
         os.mkdir(prod)
         
+        # Define static sentences based on the product
+        sentences = [
+            f"Introducing {product}, the ultimate solution for your needs.",
+            f"Experience the power of {product} today.",
+            f"Why choose {product}? Discover its amazing features.",
+            f"Transform your life with {product}.",
+            f"Get ready to enjoy {product} like never before."
+        ]
+        
         # Generate video for every sentence
-        print("Keyword :")
-        for i,s in enumerate(sentences):
-            keyword = sum_llm_chain.run(s)
-            print(i+1, ":", keyword)
-            data = search_pexels(keyword, api_key, orientation.lower())
-            link = download_video(data, prod, height, width, links,i)
-            links.append(link)
+        print("Keywords:")
+        for i, sentence in enumerate(sentences):
+            keyword = extract_keywords(sentence)  # Replace with a simple keyword extraction function
+            print(f"{i+1}: {keyword}")
             
-        print("Success! videos has been generated")
-    except Exception as e :
-        print("Error! Failed generating videos")
+            data = search_pexels(keyword, api_key, orientation.lower())
+            if data and 'videos' in data:
+                link = download_video(data, prod, height, width, links, i)
+                if link:
+                    links.append(link)
+        
+        print("Success! Videos have been generated.")
+    except Exception as e:
+        print("Error! Failed generating videos.")
         print(e)
     
     return prod, sentences
-        
+
+# Simple keyword extraction function (replace this with a more advanced method if needed)
+def extract_keywords(sentence):
+    # Extract nouns and key phrases using regex or simple heuristics
+    keywords = re.findall(r'\b\w+\b', sentence)
+    return " ".join(keywords[:3])  # Return the first 3 words as keywords