from functools import lru_cache
from pathlib import Path

from pydantic_settings import BaseSettings, SettingsConfigDict


class Settings(BaseSettings):
    model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8", extra="ignore")

    app_name: str = "DocsQA Assignment"
    secret_key: str = "change-me"
    algorithm: str = "HS256"
    access_token_expire_minutes: int = 720
    database_url: str = "postgresql+psycopg://postgres:postgres@localhost:5432/postgres"
    upload_directory: str = "./uploads"
    storage_backend: str = "local"
    supabase_url: str | None = None
    supabase_service_role_key: str | None = None
    supabase_storage_bucket: str = "documents"
    supabase_storage_prefix: str = "docsqa"
    model_name: str = "llama-3.1-8b-instant"
    embedding_model: str = "mixedbread-ai/mxbai-embed-large-v1"
    embedding_dimensions: int = 1024
    jina_api_key: str | None = None
    jina_api_base: str = "https://api.jina.ai/v1/embeddings"
    jina_embedding_model: str = "jina-embeddings-v3"
    jina_reranker_api_base: str = "https://api.jina.ai/v1/rerank"
    jina_reranker_model: str = "jina-reranker-v3"
    retrieval_k: int = 4
    rerank_candidate_k: int = 12
    groq_api_key: str | None = None
    web_search_provider: str = "tavily"
    tavily_api_key: str | None = None
    cookie_secure: bool = False

    @property
    def upload_path(self) -> Path:
        return Path(self.upload_directory)


@lru_cache
def get_settings() -> Settings:
    return Settings()