Add SettingsEngine class for handling base settings

This commit is contained in:
2024-02-18 22:50:11 +01:00
parent 59ba7b8bf6
commit 329d542133

View 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]