from dataclasses import dataclass
from dotenv import load_dotenv
import os

load_dotenv()

# 🔥 This becomes /data on Hugging Face, and stays local when developing
BASE_DATA_DIR = os.getenv("HF_HOME", "data")


@dataclass
class Settings:
    app_env: str = os.getenv("APP_ENV", "development")
    log_level: str = os.getenv("LOG_LEVEL", "INFO")

    # 📂 Data paths
    base_data_dir: str = BASE_DATA_DIR
    raw_data_dir: str = os.path.join(BASE_DATA_DIR, "raw")
    processed_data_dir: str = os.path.join(BASE_DATA_DIR, "processed")
    chroma_dir: str = os.path.join(BASE_DATA_DIR, "chroma")

    # 🤖 Models
    embedding_model: str = "sentence-transformers/all-MiniLM-L6-v2"
    llm_model: str = "llama-3.1-8b-instant"


settings = Settings()