# config.py

import os
from dotenv import load_dotenv
from typing import Dict, Any

load_dotenv()
GROQ_API_KEY="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"

# AI Provider Configuration
AI_PROVIDER = "groq"  # We're now using Groq exclusively
AI_MODEL = os.getenv("AI_MODEL", "deepseek-ai/deepseek-r1-distill-llama-70b")  # Default model
AI_API_KEY = GROQ_API_KEY

# Model Settings
MODEL_SETTINGS = {
    "max_tokens": 4000,
    "temperature": 0.7,
    "top_p": 1.0
}

# Language Settings
DEFAULT_LANGUAGE = 'ar'  # Arabic by default
SUPPORTED_LANGUAGES = ['ar', 'en']

# UAE Legal Resources
UAE_LEGAL_DOMAINS = [
    'https://elaws.moj.gov.ae',
    'https://www.mohre.gov.ae',
    'https://www.dm.gov.ae',
    'https://www.adjd.gov.ae',
    'https://www.dc.gov.ae'
]

# Legal Categories
LEGAL_CATEGORIES = {
    'civil': 'القانون المدني',
    'criminal': 'القانون الجنائي',
    'commercial': 'القانون التجاري',
    'labor': 'قانون العمل',
    'family': 'قانون الأسرة',
    'property': 'قانون العقارات'
}

def get_ai_config() -> Dict[str, Any]:
    """Get AI configuration based on environment variables"""
    return {
        "provider": AI_PROVIDER,
        "model": AI_MODEL,
        "api_key": AI_API_KEY,
        "temperature": MODEL_SETTINGS["temperature"],
        "max_tokens": MODEL_SETTINGS["max_tokens"],
        "top_p": MODEL_SETTINGS["top_p"]
    }