#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Configuration des agents dans l'application
"""

# Configuration des agents activés/désactivés
ACTIVE_AGENTS = {
    "ocr": True,         # Agent de reconnaissance optique de caractères
    "vision": True,      # Agent d'analyse d'images
    "translation": True, # Agent de traduction
    "summary": False,    # Agent de résumé (désactivé)
    "rewriter": True     # Agent de reformulation
}

# Configuration des modèles par défaut pour chaque agent
DEFAULT_MODELS = {
    "ocr": None,  # OCR utilise Tesseract, pas de modèle LLM
    "vision": "llava:34b-v1.6-fp16",  # Modèle par défaut pour l'analyse d'images
    "translation": "mistral:latest", # Modèle par défaut pour la traduction
    "summary": "mistral:latest",     # Modèle par défaut pour le résumé
    "rewriter": "mistral:latest"     # Modèle par défaut pour la reformulation
}

# Configuration de l'endpoint Ollama par défaut
DEFAULT_ENDPOINT = "http://217.182.105.173:11434"

# Configuration de journalisation détaillée
VERBOSE_LOGGING = True  # Enregistre tous les détails des entrées/sorties

def is_agent_enabled(agent_name: str) -> bool:
    """
    Vérifie si un agent est activé dans la configuration
    
    Args:
        agent_name (str): Nom de l'agent à vérifier
        
    Returns:
        bool: True si l'agent est activé, False sinon
    """
    return ACTIVE_AGENTS.get(agent_name, False)

def get_default_model(agent_name: str) -> str:
    """
    Renvoie le modèle par défaut pour un agent donné
    
    Args:
        agent_name (str): Nom de l'agent
        
    Returns:
        str: Nom du modèle par défaut
    """
    return DEFAULT_MODELS.get(agent_name, "mistral")