MAX_FILE_SIZE   = 20  * 1024 * 1024  # 20 MB
MAX_VIDEO_SIZE  = 200 * 1024 * 1024  # 200 MB

ALLOWED_IMAGE_MIMETYPES = [
    "image/jpeg", "image/png", "image/webp",
    "image/heic", "image/heif",
    "image/jfif", "image/pjpeg", "image/bmp",
    "image/gif",  "image/tiff", "image/avif",
    "image/x-jfif",
]

ALLOWED_AUDIO_MIMETYPES = [
    "audio/wav", "audio/mpeg", "audio/mp3",
    "audio/ogg", "audio/flac", "audio/x-m4a", "audio/x-wav",
]

ALLOWED_VIDEO_MIMETYPES = [
    "video/mp4", "video/quicktime", "video/x-msvideo",
    "video/webm", "video/mpeg", "video/x-matroska",
]

# ── Image models ───────────────────────────────────────────────────────────────

IMAGE_ENSEMBLE = [
    {
        "key":    "ai_vs_human",
        "name":   "Ateeqq/ai-vs-human-image-detector",
        "weight": 0.45,
        "desc":   "AI vs Human 120k (ViT)",
    },
    {
        "key":    "ai_vs_deepfake_vs_real",
        "name":   "prithivMLmods/AI-vs-Deepfake-vs-Real",
        "weight": 0.35,
        "desc":   "AI/Deepfake/Real 3-class (ViT)",
    },
    {
        "key":    "deepfake_detector",
        "name":   "prithivMLmods/Deep-Fake-Detector-Model",
        "weight": 0.20,
        "desc":   "Deepfake faces (ViT)",
    },
]

IMAGE_FAST_ENSEMBLE = [
    {
        "key":    "ai_vs_human",
        "name":   "Ateeqq/ai-vs-human-image-detector",
        "weight": 0.45,
        "desc":   "AI vs Human 120k (ViT)",
    },
    {
        "key":    "deepfake_detector",
        "name":   "prithivMLmods/Deep-Fake-Detector-Model",
        "weight": 0.55,
        "desc":   "Deepfake faces (ViT)",
    },
]

# ── Audio model ────────────────────────────────────────────────────────────────

AUDIO_MODEL = {
    "key":  "deepfake_audio_v2",
    "name": "MelodyMachine/Deepfake-audio-detection-V2",
    "desc": "Deepfake Audio Detection V2",
}

# ── Video ensemble (reuses ViT models) ────────────────────────────────────────

VIDEO_ENSEMBLE = [
    {
        "key":    "deepfake_detector",
        "name":   "prithivMLmods/Deep-Fake-Detector-Model",
        "weight": 0.40,
        "desc":   "Deepfake faces (ViT)",
    },
    {
        "key":    "ai_vs_deepfake_vs_real",
        "name":   "prithivMLmods/AI-vs-Deepfake-vs-Real",
        "weight": 0.35,
        "desc":   "AI/Deepfake/Real 3-class",
    },
    {
        "key":    "ai_vs_human",
        "name":   "Ateeqq/ai-vs-human-image-detector",
        "weight": 0.25,
        "desc":   "AI vs Human 120k",
    },
]

# ── Text models ────────────────────────────────────────────────────────────────

TEXT_MODELS = {
    "ai1": {
        "name": "fakespot-ai/roberta-base-ai-text-detection-v1",
        "desc": "RoBERTa AI text detector (Fakespot)",
    },
    "ai2": {
        "name": "Hello-SimpleAI/chatgpt-detector-roberta",
        "desc": "RoBERTa ChatGPT detector",
    },
    "fn1": {
        "name": "vikram71198/distilroberta-base-finetuned-fake-news-detection",
        "desc": "DistilRoBERTa fake news detector",
    },
    "fn2": {
        "name": "jy46604790/Fake-News-Bert-Detect",
        "desc": "BERT fake news detector",
    },
}