from pydantic_settings import BaseSettings

class Settings(BaseSettings):
    DATABASE_URL: str = "postgresql+psycopg2://user:pass@db:5432/news"
    REDIS_URL: str = "redis://redis:6379/0"
    STORAGE_DIR: str = "/data/storage"
    DPI: int = 300
    LAYOUT_MODEL: str = "PubLayNet"        # or DocLayNet / custom YOLOv11
    OCR_LANGS: list = ["en", "hi", "te", "ta"]
    LLM_PROVIDER: str = "gemini"           # gemini | openai | local
    LLM_API_KEY: str = ""
    EMBED_MODEL: str = "sentence-transformers/all-MiniLM-L6-v2"
    USE_GPU: bool = True

    class Config:
        env_file = ".env"

settings = Settings()
