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

Support for upgraded moviepy

Browse Source
This commit is contained in:
Paillat
2024-03-02 15:19:30 +01:00
parent 6b6b18507a
commit 455514e475
13 changed files with 87 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,11 +2,11 @@ import os
from typing import Literal, TypedDict, List
import gradio as gr
import moviepy.editor as mp
import moviepy as mp
import moviepy.video.fx as vfx
import openai
from openai import OpenAI
import requests
from moviepy.video.fx.resize import resize
from . import BaseAssetsEngine
@@ -79,15 +79,11 @@ class DallEAssetsEngine(BaseAssetsEngine):
img = mp.ImageClip("temp.png")
os.remove("temp.png")
img: mp.ImageClip = img.set_duration(end - start)
img: mp.ImageClip = img.set_start(start)
img: mp.ImageClip = resize(img, width=max_width)
if self.aspect_ratio == "portrait":
img: mp.ImageClip = img.set_position(("center", "top"))
elif self.aspect_ratio == "landscape":
img: mp.ImageClip = img.set_position(("center", "top"))
elif self.aspect_ratio == "square":
img: mp.ImageClip = img.set_position(("center", "top"))
img: mp.ImageClip = img.with_duration(end - start)
img: mp.ImageClip = img.with_start(start)
img: mp.ImageClip = img.with_effects([vfx.Resize(width=max_width)])
position = img.with_position(("center", "top"))
img: mp.ImageClip = img.with_position(position)
clips.append(img)
return clips