import os

import dotenv

dotenv.load_dotenv()

ACCESS_RATE = "20/minute"


class Config:
    Nepali_model_folder = os.getenv("Nepali_model")
    English_model_folder = os.getenv("English_model")
    REPO_ID_LANG = os.getenv("English_model") or "Pujan-Dev/Ai_vs_HUMAN"
    LANG_MODEL = os.getenv("LANG_MODEL")
    HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN")
    SECRET_TOKEN = os.getenv("MY_SECRET_TOKEN")

    IMAGE_CLASSIFIER_REPO_ID = os.getenv("IMAGE_CLASSIFIER_REPO_ID", "can-org/AI-VS-HUMAN-IMAGE-classifier")
    IMAGE_CLASSIFIER_MODEL_DIR = os.getenv("IMAGE_CLASSIFIER_MODEL_DIR", "./IMG_Models")
    IMAGE_CLASSIFIER_WEIGHTS_FILE = os.getenv("IMAGE_CLASSIFIER_WEIGHTS_FILE", "latest-my_cnn_model.h5")

    AI_HUMAN_CLIP_MODEL_NAME = os.getenv("AI_HUMAN_CLIP_MODEL_NAME", "ViT-L/14")
    AI_HUMAN_SVM_REPO_ID = os.getenv("AI_HUMAN_SVM_REPO_ID", "rhnsa/ai_human_image_detector")
    AI_HUMAN_SVM_FILENAME = os.getenv("AI_HUMAN_SVM_FILENAME", "svm_model_real.joblib")

    REAL_FORGED_MODEL_REPO_ID = os.getenv("REAL_FORGED_MODEL_REPO_ID", "rhnsa/real_forged_classifier")
    REAL_FORGED_MODEL_FILENAME = os.getenv("REAL_FORGED_MODEL_FILENAME", "fft_cnn_model_78.pth")
    REAL_FORGED_MODEL_LOCAL_PATH = os.getenv("REAL_FORGED_MODEL_LOCAL_PATH", "Model/real_forged/fft_cnn_model_78.pth")
    DOCUMENT_FORGERY_MODEL_REPO_ID = os.getenv(
        "DOCUMENT_FORGERY_MODEL_REPO_ID",
        REPO_ID_LANG
    )
    DOCUMENT_FORGERY_MODEL_FILENAME = os.getenv(
        "DOCUMENT_FORGERY_MODEL_FILENAME",
        "document_forgery/pixel_forgery_v3_best.pth",
    )
    DOCUMENT_FORGERY_MODEL_PATH = os.getenv(
        "DOCUMENT_FORGERY_MODEL_PATH",
        "features/Modelsdfa/document_forgery/pixel_forgery_v3_best.pth",
    )
    # Decision thresholds for document forgery detector (probabilities in 0..1)
    DOCUMENT_FORGERY_POSSIBLE_LOW = float(os.getenv("DOCUMENT_FORGERY_POSSIBLE_LOW", "0.40"))
    DOCUMENT_FORGERY_FORGED_LOW = float(os.getenv("DOCUMENT_FORGERY_FORGED_LOW", "0.55"))

    RAG_CHROMA_HOST = os.getenv("CHROMA_HOST", "localhost")
    RAG_CHROMA_PORT = int(os.getenv("CHROMA_PORT", "8000"))
    RAG_COLLECTION_NAME = os.getenv("RAG_COLLECTION_NAME", "company_docs_collection")

    RAG_LLM_PROVIDER = os.getenv("LLM_PROVIDER", "openai").lower()
    RAG_LLM_API_KEY = os.getenv("LLM_API_KEY")
    RAG_LLM_MODEL = os.getenv("LLM_MODEL", "gpt-3.5-turbo")
    RAG_LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0"))
    RAG_LLM_MAX_TOKENS = int(os.getenv("LLM_MAX_TOKENS", "2048"))

    RAG_MAX_FILE_SIZE = int(os.getenv("RAG_MAX_FILE_SIZE", str(100 * 1024 * 1024)))
    RAG_MAX_QUERY_LENGTH = int(os.getenv("RAG_MAX_QUERY_LENGTH", "1000"))
    RAG_SUPPORTED_CONTENT_TYPES = {
        "application/pdf",
        "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
        "text/plain",
    }