File size: 1,508 Bytes
60c56d7 2999873 60c56d7 8f6f449 60c56d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""
Configuration settings for the application
"""
import os
from pydantic_settings import BaseSettings
class Settings(BaseSettings):
"""Application settings"""
# Firebase settings
ENABLE_APP_CHECK: bool = os.getenv("ENABLE_APP_CHECK", "true").lower() == "true"
FIREBASE_CREDENTIALS_PATH: str = os.getenv(
"FIREBASE_CREDENTIALS_PATH",
"/data/firebase-adminsdk.json"
)
# API settings
BASE_URL: str = os.getenv("BASE_URL", "http://localhost:8000")
# Model settings
MODEL_ID: str = os.getenv("MODEL_ID", "nickpai/sdxl_light_caption_output")
NUM_INFERENCE_STEPS: int = int(os.getenv("NUM_INFERENCE_STEPS", "8"))
POSITIVE_PROMPT: str = os.getenv(
"POSITIVE_PROMPT",
"high quality color photo, vibrant natural colors, detailed lighting"
)
NEGATIVE_PROMPT: str = os.getenv(
"NEGATIVE_PROMPT",
"low quality, monochrome, black and white, desaturated, blurry, grainy"
)
GUIDANCE_SCALE: float = float(os.getenv("GUIDANCE_SCALE", "1.0"))
CONTROLNET_SCALE: float = float(os.getenv("CONTROLNET_SCALE", "1.0"))
CAPTION_PREFIX: str = os.getenv("CAPTION_PREFIX", "a photography of")
COLORIZE_SEED: int = int(os.getenv("COLORIZE_SEED", "123"))
# Storage settings
UPLOAD_DIR: str = os.getenv("UPLOAD_DIR", "uploads")
RESULT_DIR: str = os.getenv("RESULT_DIR", "results")
class Config:
env_file = ".env"
case_sensitive = False
settings = Settings()
|