# config.py

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

load_dotenv()

# AI Provider Configuration
AI_PROVIDER = os.getenv("AI_PROVIDER", "groq")  # Default to Groq
AI_MODEL = os.getenv("AI_MODEL", "deepseek-ai/deepseek-r1-distill-llama-70b")  # Default model
AI_API_KEY = os.getenv("GROQ_API_KEY")  # Groq API key

# Model Settings for Groq
MODEL_SETTINGS = {
    "groq": {
        "default_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 based on environment variables"""
    provider = os.getenv("AI_PROVIDER", "groq").lower()
    
    config = {
        "provider": provider,
        "model": os.getenv("AI_MODEL"),
        "api_key": os.getenv("GROQ_API_KEY")
    }
    
    # Set default model and settings for Groq
    if provider == "groq":
        config["model"] = config["model"] or MODEL_SETTINGS["groq"]["default_model"]
        config.update(MODEL_SETTINGS["groq"])
        
    return config