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

9
app/services/material.py
View File

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

1
config.example.toml
View File

@@ -53,6 +53,7 @@
compute_type="int8" compute_type="int8"
[pexels] [pexels]
video_concat_mode="sequential" # "random" or "sequential"
[pexels.proxies] [pexels.proxies]
### Use a proxy to access the Pexels API ### Use a proxy to access the Pexels API
### Format: "http://<username>:<password>@<proxy>:<port>" ### Format: "http://<username>:<password>@<proxy>:<port>"