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

Add settings engine to GenerationContext & load resolution from them

Browse Source
This commit is contained in:
Paillat
2024-02-18 22:49:36 +01:00
parent 517d6db8a6
commit 32fa60c92a
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/chore/GenerationContext.py
View File

@@ -16,6 +16,7 @@ class GenerationContext:
ttsengine, ttsengine,
captioningengine, captioningengine,
assetsengine, assetsengine,
settingsengine,
) -> None: ) -> None:
self.powerfulllmengine: engines.LLMEngine.BaseLLMEngine = powerfulllmengine[0] self.powerfulllmengine: engines.LLMEngine.BaseLLMEngine = powerfulllmengine[0]
self.powerfulllmengine.ctx = self self.powerfulllmengine.ctx = self
@@ -40,6 +41,9 @@ class GenerationContext:
self.assetsengineselector = engines.AssetsEngine.AssetsEngineSelector() self.assetsengineselector = engines.AssetsEngine.AssetsEngineSelector()
self.assetsengineselector.ctx = self self.assetsengineselector.ctx = self
self.settingsengine: engines.SettingsEngine.SettingsEngine = settingsengine[0]
self.settingsengine.ctx = self
def setup_dir(self): def setup_dir(self):
self.dir = f"output/{time.time()}" self.dir = f"output/{time.time()}"
os.makedirs(self.dir) os.makedirs(self.dir)
@@ -50,11 +54,10 @@ class GenerationContext:
def process(self): def process(self):
# ⚠️ IMPORTANT NOTE: All methods called here are expected to be defined as abstract methods in the base classes, if not there is an issue with the engine implementation. # ⚠️ IMPORTANT NOTE: All methods called here are expected to be defined as abstract methods in the base classes, if not there is an issue with the engine implementation.
# we start by loading the settings
progress = gr.Progress() progress = gr.Progress()
self.width, self.height = ( self.settingsengine.load()
1080,
1920,
) # TODO: Add support for custom resolution, for now it's tiktok's resolution
self.setup_dir() self.setup_dir()
self.script = self.scriptengine.generate() self.script = self.scriptengine.generate()