import os
from pathlib import Path

# Base directory configuration
#BASE_DIR = Path(__file__).resolve().parent
BASE_DIR = Path("/root/ds_tjc")

# Data directories
UPLOAD_DIR = BASE_DIR / "marketing_data"
CHROMA_PATH = BASE_DIR / "chroma_index"

# Ensure directories exist
UPLOAD_DIR.mkdir(exist_ok=True)
for category in ["email", "books", "article", "social"]:
    (UPLOAD_DIR / category).mkdir(exist_ok=True)

# Model configurations
MODEL_NAME = "BAAI/bge-large-en-v1.5"
RERANKER_NAME = "BAAI/bge-reranker-large"
GROQ_MODEL = "llama-3.3-70b-versatile"
COLLECTION_NAME = "marketing_docs"

# API configurations
API_KEY = "4BkwTtVd5VwhTiFDdG3NfzgATrCq7aD8AjnvWNeivirTntHgRvL6Xe84ULHcVTLB"
SERVER_URL = "https://ma.rommelcorral.com"
GROQ_API_KEY = "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"

# Valid document categories
VALID_CATEGORIES = ["email", "books", "article", "social"]