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

Fix VideoBackgroundEngine.py not working

Browse Source
This commit is contained in:
Paillat
2024-03-01 12:50:23 +01:00
parent a76572beb0
commit d94d827220
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -6,6 +6,7 @@ import time
import gradio as gr import gradio as gr
import moviepy.editor as mp import moviepy.editor as mp
from moviepy.video.fx.crop import crop from moviepy.video.fx.crop import crop
from moviepy.video.fx.resize import resize
from . import BaseBackgroundEngine from . import BaseBackgroundEngine
@@ -47,15 +48,12 @@ class VideoBackgroundEngine(BaseBackgroundEngine):
) )
start = random.uniform(0, background_max_start) start = random.uniform(0, background_max_start)
clip = background.subclip(start, start + self.ctx.duration) clip = background.subclip(start, start + self.ctx.duration)
w, h = clip.size
self.ctx.credits += f"\n{self.background_video.data['credits']}" self.ctx.credits += f"\n{self.background_video.data['credits']}"
if w == h: clip = resize(clip, width=self.ctx.width, height=self.ctx.height)
clip = clip.resize(width=self.ctx.width) if w > h else clip.resize(height=self.ctx.height) w, h = clip.size
elif w > h: if w > h:
clip = clip.resize(width=self.ctx.width)
clip = crop(clip, width=self.ctx.width, height=self.ctx.height, x_center=w / 2, y_center=h / 2) clip = crop(clip, width=self.ctx.width, height=self.ctx.height, x_center=w / 2, y_center=h / 2)
else: else:
clip = clip.resize(height=self.ctx.height)
clip = crop(clip, width=self.ctx.width, height=self.ctx.height, x_center=w / 2, y_center=h / 2) clip = crop(clip, width=self.ctx.width, height=self.ctx.height, x_center=w / 2, y_center=h / 2)
self.ctx.index_0.append(clip) self.ctx.index_0.append(clip)