# config.py
# This is the central "control panel" for MudabbirAI.

# --- 1. MODEL CONFIGURATION ---
MODELS = {
    "Gemini": {
        "default": "gemini-2.0-flash", # 1M Context
        "judge": "gemini-2.0-flash",
        "classifier": "gemini-2.0-flash",
        "manager": "gemini-2.0-flash"
    },
    "Anthropic": {
        "default": "claude-3-5-haiku-20241022" # 200K Context
    },
    "SambaNova": {
        "default": "Llama-4-Maverick-17B-128E-Instruct" # 512K Context
    },
    "OpenAI": {
        "default": "gpt-4o-mini" # 128K Context
    },
    # Updated: Using Qwen via Nebius
    "Nebius": {
        "default": "Qwen/Qwen3-235B-A22B-Thinking-2507" # 262K Context
    }
}

# --- 2. PRICING (USD per 1 Million Tokens) ---
PRICING = {
    "Gemini":     {"input": 0.10, "output": 0.40},
    "Anthropic":  {"input": 0.25, "output": 1.25},
    "SambaNova":  {"input": 0.00, "output": 0.00},
    "OpenAI":     {"input": 0.15, "output": 0.60},
    # Updated Pricing for Qwen (based on your screenshot)
    "Nebius":     {"input": 0.20, "output": 0.60}
}

# --- 3. CALIBRATION CONFIGURATION ---
CALIBRATION_CONFIG = {
    "roles_to_test": {
        "Plant": "Culture_5",
        "Implementer": "Culture_Expert",
        "Monitor": "Culture_11"
    },
    "role_metrics": {
        "Plant": "Novelty",
        "Implementer": "Usefulness_Feasibility",
        "Monitor": "Cultural_Appropriateness"
    }
}

# --- 4. PROMPT FILE PATHS ---
PROMPT_FILES = {
    "evaluator": "prompts/evaluator_judge.txt",
    "classifier": "prompts/classifier.txt",
    "manager_homogeneous": "prompts/manager_homogeneous.txt",
    "manager_heterogeneous": "prompts/manager_heterogeneous.txt"
}

# --- 5. STATIC CONFIGS ---
DEFAULT_PERSONA_KEY = "Culture_Expert"