Spaces:
Running
Running
File size: 3,080 Bytes
9b70717 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os
import requests
from utility.utils import log_response,LOG_TYPE_PEXEL
PEXELS_API_KEY = os.environ.get('PEXELS_KEY')
def search_videos(query_string, orientation_landscape=True):
url = "https://api.pexels.com/videos/search"
headers = {
"Authorization": PEXELS_API_KEY,
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
}
params = {
"query": query_string,
"orientation": "landscape" if orientation_landscape else "portrait",
"per_page": 15
}
response = requests.get(url, headers=headers, params=params)
json_data = response.json()
log_response(LOG_TYPE_PEXEL,query_string,response.json())
return json_data
def getBestVideo(query_string, orientation_landscape=True, used_vids=[]):
vids = search_videos(query_string, orientation_landscape)
videos = vids['videos'] # Extract the videos list from JSON
# Filter and extract videos with width and height as 1920x1080 for landscape or 1080x1920 for portrait
if orientation_landscape:
filtered_videos = [video for video in videos if video['width'] >= 1920 and video['height'] >= 1080 and video['width']/video['height'] == 16/9]
else:
filtered_videos = [video for video in videos if video['width'] >= 1080 and video['height'] >= 1920 and video['height']/video['width'] == 16/9]
# Sort the filtered videos by duration in ascending order
sorted_videos = sorted(filtered_videos, key=lambda x: abs(15-int(x['duration'])))
# Extract the top 3 videos' URLs
for video in sorted_videos:
for video_file in video['video_files']:
if orientation_landscape:
if video_file['width'] == 1920 and video_file['height'] == 1080:
if not (video_file['link'].split('.hd')[0] in used_vids):
return video_file['link']
else:
if video_file['width'] == 1080 and video_file['height'] == 1920:
if not (video_file['link'].split('.hd')[0] in used_vids):
return video_file['link']
print("NO LINKS found for this round of search with query :", query_string)
return None
def generate_video_url(timed_video_searches,video_server):
timed_video_urls = []
if video_server == "pexel":
used_links = []
for (t1, t2), search_terms in timed_video_searches:
url = ""
for query in search_terms:
url = getBestVideo(query, orientation_landscape=True, used_vids=used_links)
if url:
used_links.append(url.split('.hd')[0])
break
timed_video_urls.append([[t1, t2], url])
elif video_server == "stable_diffusion":
timed_video_urls = get_images_for_video(timed_video_searches)
return timed_video_urls
|