import os

from dotenv import find_dotenv, load_dotenv
from pydantic_settings import BaseSettings

load_dotenv(find_dotenv())


class Settings(BaseSettings):
    cohere_api_key: str = ""
    groq_api_key: str = ""
    openai_api_key: str = ""
    openrouter_api_key: str = ""
    # HuggingFace Hub configuration
    hf_token: str = os.getenv("HF_TOKEN", "")
    
    chroma_db: str = os.getenv("CHROMA_DB", "bge-small-finetuned-chroma")
    chroma_collection: str = os.getenv("CHROMA_COLLECTION", "bge_small_finetuned_astra_collection")


settings = Settings()

SRC_DIR: str = os.path.dirname(__file__)
DATA_DIR: str = os.path.join(SRC_DIR, "../../data")
MODELS_DIR: str = os.path.join(SRC_DIR, "../../models")
CHROMA_DIR: str = os.path.join(SRC_DIR, "../../.chroma")
CHROMA_DB: str = os.path.join(CHROMA_DIR, settings.chroma_db)
CHROMA_COLLECTION: str =  settings.chroma_collection
HF_MODELS_REPO: str = "NenJa/astra-models"
HF_CHROMADB_REPO: str = "NenJa/astra-chromadb"