Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ feat(VideoBackgroundEngine): Loop video if too short

Browse Source
This commit is contained in:
Paillat
2024-05-03 09:18:05 +02:00
parent 8ba28bd1df
commit a9ab813589
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/engines/BackgroundEngine/VideoBackgroundEngine.py
View File

@@ -42,11 +42,10 @@ class VideoBackgroundEngine(BaseBackgroundEngine):
def get_background(self): def get_background(self):
background = mp.VideoFileClip(f"{self.background_video.path}", audio=False) background = mp.VideoFileClip(f"{self.background_video.path}", audio=False)
if background.duration < self.ctx.duration:
num = int(self.ctx.duration / background.duration) + 1
background = mp.concatenate_videoclips([background for _ in range(num)])
background_max_start = background.duration - self.ctx.duration background_max_start = background.duration - self.ctx.duration
if background_max_start < 0:
raise ValueError(
"The background video is shorter than the video to be generated."
)
start = random.uniform(0, background_max_start) start = random.uniform(0, background_max_start)
clip = background.with_subclip(start, start + self.ctx.duration) clip = background.with_subclip(start, start + self.ctx.duration)
self.ctx.credits += f"\n{self.background_video.data['credits']}" self.ctx.credits += f"\n{self.background_video.data['credits']}"
@@ -58,6 +57,7 @@ class VideoBackgroundEngine(BaseBackgroundEngine):
else: else:
clip = clip.with_effects([vfx.Resize(width=self.ctx.width)]) clip = clip.with_effects([vfx.Resize(width=self.ctx.width)])
clip = clip.with_position(("center", "center")) clip = clip.with_position(("center", "center"))
clip: mp.VideoClip = clip.without_audio()
return clip return clip
@classmethod @classmethod