hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add pexels.video_concat_mode

Browse Source
This commit is contained in:
harry
2024-03-16 09:44:57 +08:00
parent 753ec011d8
commit 74e634d64c
2 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/services/material.py
View File

@@ -1,3 +1,4 @@
import random
import time
import requests
@@ -85,6 +86,9 @@ def download_videos(task_id: str,
minimum_duration: int = 5
) -> List[str]:
valid_video_urls = []
video_concat_mode = config.pexels.get("video_concat_mode", "")
for search_term in search_terms:
# logger.info(f"searching videos for '{search_term}'")
video_urls = search_videos(search_term=search_term,
@@ -92,13 +96,18 @@ def download_videos(task_id: str,
minimum_duration=minimum_duration,
video_aspect=video_aspect)
logger.info(f"found {len(video_urls)} videos for '{search_term}'")
i = 0
for url in video_urls:
if video_concat_mode == "random":
url = random.choice(video_urls)
if url not in valid_video_urls:
valid_video_urls.append(url)
i += 1
if i >= 3:
break
if i >= 3:
break
logger.info(f"downloading videos: {len(valid_video_urls)}")
video_paths = []