File size: 956 Bytes
60c56d7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2ae242d
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
"""
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", "lllyasviel/control_v11f1e_sd15_color")
    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()