| """ | |
| 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", | |
| "colorize-662df-firebase-adminsdk-fbsvc-e080668793.json" | |
| ) | |
| # API settings | |
| BASE_URL: str = os.getenv("BASE_URL", "http://localhost:8000") | |
| # Model settings | |
| MODEL_ID: str = os.getenv("MODEL_ID", "rsortino/ColorizeNet") | |
| NUM_INFERENCE_STEPS: int = int(os.getenv("NUM_INFERENCE_STEPS", "20")) | |
| # 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() | |