Video Search

def search_for_stock_videos(query: str, api_key: str, it: int, min_dur: int) -> List[str]:
    """
    Searches for stock videos based on a query.
    
    Args:
        query (str): The query to search for.
        api_key (str): The Pexels API key to use.
        it (int): Number of videos to fetch per query.
        min_dur (int): Minimum duration in seconds for each video.
    
    Returns:
        List[str]: A list of stock video URLs.
    """

headers = {
    "Authorization": api_key
}

qurl = f"https://api.pexels.com/videos/search?query={query}&per_page={it}"
r = requests.get(qurl, headers=headers)
response = r.json()

def get_search_terms(
    video_subject: str, amount: int, script: str, ai_model: str
) -> List[str]:
    """
    Generate a JSON-Array of search terms for stock videos,
    depending on the subject of a video.
    """

prompt = f"""
Generate {amount} search terms for stock videos,
depending on the subject of a video.
Subject: {video_subject}

The search terms are to be returned as
a JSON-Array of strings.

Each search term should consist of 1-3 words,
always add the main subject of the video.

YOU MUST ONLY RETURN THE JSON-ARRAY OF STRINGS.
YOU MUST NOT RETURN ANYTHING ELSE.

For context, here is the full text:
{script}
"""

[
  "space exploration rocket",
  "astronaut spacewalk",
  "mars rover",
  "space station orbit",
  "galaxy nebula"
]

for video in raw_urls:
    # Check if video has a valid download link
    if ".com/video-files" in video["link"]:
        # Only save the URL with the largest resolution
        if (video["width"] * video["height"]) > video_res:
            temp_video_url = video["link"]
            video_res = video["width"] * video["height"]

for i in range(it):
    # Check if video has desired minimum duration
    if response["videos"][i]["duration"] < min_dur:
        continue
    # Process video...

from Backend.search import search_for_stock_videos
from Backend.gpt import get_search_terms

# Generate search terms from script
script = "Space exploration has revolutionized our understanding..."
search_terms = get_search_terms(
    video_subject="Space Exploration",
    amount=5,
    script=script,
    ai_model="llama3.1:8b"
)

print(f"Search terms: {search_terms}")
# Output: ['space exploration', 'astronaut mission', 'rocket launch', ...]

# Search for videos for each term
api_key = "YOUR_PEXELS_API_KEY"
video_urls = []

for term in search_terms:
    urls = search_for_stock_videos(
        query=term,
        api_key=api_key,
        it=3,  # Fetch 3 videos per term
        min_dur=5  # Minimum 5 seconds
    )
    video_urls.extend(urls)

print(f"Found {len(video_urls)} videos")

try:
    # Loop through each video in the result
    for i in range(it):
        # Process videos...
except Exception as e:
    log("[-] No Videos found.", "error")
    log(str(e), "error")

{
  "videos": [
    {
      "id": 123456,
      "duration": 15,
      "video_files": [
        {
          "id": 789,
          "link": "https://player.vimeo.com/video-files/...",
          "width": 1920,
          "height": 1080,
          "quality": "hd"
        }
      ]
    }
  ]
}

log(f"\t=> \"{query}\" found {len(video_url)} Videos", "info")

=> "space exploration" found 3 Videos
=> "astronaut mission" found 2 Videos
=> "rocket launch" found 3 Videos