From 74e634d64ca9cdc310b97ac0507d061be7ac71a3 Mon Sep 17 00:00:00 2001
From: harry <harry@wangxutech.com>
Date: Sat, 16 Mar 2024 09:44:57 +0800
Subject: [PATCH] add pexels.video_concat_mode

---
 app/services/material.py | 13 +++++++++++--
 config.example.toml      |  1 +
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/services/material.py b/app/services/material.py
index 2c0bbf7..e2d9298 100644
--- a/app/services/material.py
+++ b/app/services/material.py
@@ -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 = []
diff --git a/config.example.toml b/config.example.toml
index 7a669ec..b394786 100644
--- a/config.example.toml
+++ b/config.example.toml
@@ -53,6 +53,7 @@
     compute_type="int8"
 
 [pexels]
+    video_concat_mode="sequential" # "random" or "sequential"
     [pexels.proxies]
         ### Use a proxy to access the Pexels API
         ### Format: "http://<username>:<password>@<proxy>:<port>"