# config.py

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

load_dotenv()

# AI Provider Configuration
AI_API_KEY = os.getenv("GROQ_API_KEY")  # Groq API key

# Model Settings
MODEL_SETTINGS = {
    "model": "deepseek-ai/deepseek-r1-distill-llama-70b",
    "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"""
    config = {
        "api_key": os.getenv("GROQ_API_KEY")
    }
    config.update(MODEL_SETTINGS)
    return config