import enum

from pydantic_settings import BaseSettings


class LogLevel(str, enum.Enum):
    """Possible log levels."""

    NOTSET = "NOTSET"
    DEBUG = "DEBUG"
    INFO = "INFO"
    WARNING = "WARNING"
    ERROR = "ERROR"
    FATAL = "FATAL"


class Settings(BaseSettings):
    temperature: float = 0.2
    model: str = "llama3-8b-8192"
    log_level: LogLevel = LogLevel.INFO
    segment_time: int = 1500
    max_tokens: int = 375
    system_prompt: str = ("You are a helpful, respectful and honest assistant."
                          "You task will be to summarize video transcripts. "
                          "You will receive them in parts, so don't indicate beggining."
                          "Answer only in russian.")


app_settings = Settings()