Spaces:
Runtime error
Runtime error
| 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): | |
| 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}') | |
| # 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: | |
| 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'] | |
| # 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 | |
| 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] | |
| # Create directory with the product's name | |
| if os.path.exists(prod): | |
| shutil.rmtree(prod) | |
| os.mkdir(prod) | |
| # 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, links,i) | |
| links.append(link) | |
| print("Success! videos has been generated") | |
| except Exception as e : | |
| print("Error! Failed generating videos") | |
| print(e) | |
| return prod, sentences | |