# config.py
import os
import sys
import logging
from pathlib import Path


BASE_DIR = Path(__file__).resolve().parent

if str(BASE_DIR) not in sys.path:
    sys.path.insert(0, str(BASE_DIR))


try:
    import google.colab
    IN_COLAB = True
except ImportError:
    IN_COLAB = False


if IN_COLAB:
    VECTORSTORE_DIR = Path("/content/drive/MyDrive/bioinformatics_tutor_ai/vectorstore")
else:
    VECTORSTORE_DIR = BASE_DIR / "vectorstore"

# Models
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
LLM_MODEL = "BioMistral/BioMistral-7B-DARE"

# Confidence threshold for TutorAgent
CONFIDENCE_THRESHOLD = 0.65


OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
if not OPENAI_API_KEY:
    logging.warning("OPENAI_API_KEY not set! AutoGen Tutor Agent will not work without it.")


if "HF_HOME" not in os.environ:
    os.environ["HF_HOME"] = "/home/user/.cache/huggingface"
    os.environ["TRANSFORMERS_CACHE"] = os.path.join(os.environ["HF_HOME"], "transformers")
    os.environ["HF_HUB_CACHE"] = os.path.join(os.environ["HF_HOME"], "hub")