mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 01:06:19 +00:00
Add SettingsEngine class for handling base settings
This commit is contained in:
25
src/engines/SettingsEngine/SettingsEngine.py
Normal file
25
src/engines/SettingsEngine/SettingsEngine.py
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
import gradio as gr
|
||||||
|
from abc import ABC, abstractmethod
|
||||||
|
from ..BaseEngine import BaseEngine
|
||||||
|
|
||||||
|
|
||||||
|
class SettingsEngine(BaseEngine):
|
||||||
|
def __init__(self, options: list) -> None:
|
||||||
|
self.width = options[0]
|
||||||
|
self.height = options[1]
|
||||||
|
super().__init__()
|
||||||
|
|
||||||
|
name = "Settings Engine"
|
||||||
|
description = "Engine for handling all the base settings for the content generation"
|
||||||
|
num_options = 2
|
||||||
|
|
||||||
|
def load(self):
|
||||||
|
self.ctx.width = self.width
|
||||||
|
self.ctx.height = self.height
|
||||||
|
@classmethod
|
||||||
|
def get_options(cls):
|
||||||
|
#minimum is 720p, maximum is 4k, default is portrait hd
|
||||||
|
width = gr.Number(value=1080, minimum=720, maximum=3840, label="Width", step=1)
|
||||||
|
height = gr.Number(value=1920, minimum=720, maximum=3840, label="Height", step=1)
|
||||||
|
|
||||||
|
return [width, height]
|
||||||
Reference in New Issue
Block a user