From 56ae10301a4ac6cbb8207e238089368933cb336d Mon Sep 17 00:00:00 2001 From: Paillat Date: Sun, 18 Feb 2024 22:49:56 +0100 Subject: [PATCH] Add image resizing in DallEAssetsEngine --- src/engines/AssetsEngine/DallEAssetsEngine.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/engines/AssetsEngine/DallEAssetsEngine.py b/src/engines/AssetsEngine/DallEAssetsEngine.py index 4826e07..b980403 100644 --- a/src/engines/AssetsEngine/DallEAssetsEngine.py +++ b/src/engines/AssetsEngine/DallEAssetsEngine.py @@ -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":