# config.py

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

load_dotenv()

# AI Provider Configuration
GROQ_API_KEY = os.getenv("GROQ_API_KEY", "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")

# Model Settings
MODEL_SETTINGS = {
    "model": "deepseek-r1-distill-llama-70b",
    "max_tokens": 4000,
    "temperature": 0.7,
    "top_p": 1.0,
    "api_base": "https://api.groq.com/openai/v1"
}

# 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"""
    config = {
        "api_key": GROQ_API_KEY,
        "api_base": MODEL_SETTINGS["api_base"],
        "language": "ar",  # Set language to Arabic
    }
    config.update(MODEL_SETTINGS)
    return config