Files
ai-char-bot/models/enums.py

44 lines
916 B
Python

from enum import Enum
class AspectRatios(str, Enum):
NINESIXTEEN = "NINESIXTEEN"
SIXTEENNINE = "SIXTEENNINE"
THREEFOUR = "THREEFOUR"
FOURTHREE = "FOURTHREE"
@property
def value_ratio(self) -> str:
return {
AspectRatios.NINESIXTEEN: "9:16",
AspectRatios.SIXTEENNINE: "16:9",
AspectRatios.THREEFOUR: "3:4",
AspectRatios.FOURTHREE: "4:3",
}[self]
class Quality(str, Enum):
ONEK = 'ONEK'
TWOK = 'TWOK'
FOURK = 'FOURK'
@property
def value_quality(self) -> str:
return {
Quality.ONEK: '1K',
Quality.TWOK: '2K',
Quality.FOURK: '4K'
}[self]
class GenType(str, Enum):
TEXT = 'Text'
IMAGE = 'Image'
@property
def value_type(self) -> str:
return {
GenType.TEXT: 'Text',
GenType.IMAGE: 'Image',
}[self]