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

Add image resizing in DallEAssetsEngine

Browse Source
This commit is contained in:
Paillat
2024-02-18 22:49:56 +01:00
parent 3a4e032799
commit 56ae10301a
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/engines/AssetsEngine/DallEAssetsEngine.py
View File

@@ -1,6 +1,7 @@
import gradio as gr
import openai
import moviepy.editor as mp
from moviepy.video.fx.resize import resize
import io
import base64
import time
@@ -38,6 +39,7 @@ class DallEAssetsEngine(BaseAssetsEngine):
super().__init__()
def get_assets(self, options: list[Spec]) -> list[mp.ImageClip]:
max_width = self.ctx.width / 3 * 2
clips = []
for option in options:
prompt = option["prompt"]
@@ -67,9 +69,10 @@ class DallEAssetsEngine(BaseAssetsEngine):
os.remove("temp.png")
img: mp.ImageClip = img.set_duration(end - start)
img = img.set_start(start)
img: mp.ImageClip = img.set_start(start)
img: mp.ImageClip = resize(img, width=max_width)
if self.aspect_ratio == "portrait":
img = img.set_position(("center", "top"))
img: mp.ImageClip = img.set_position(("center", "top"))
elif self.aspect_ratio == "landscape":
img = img.set_position(("center", "center"))
elif self.aspect_ratio == "square":