"""
config.py
---------
Global configuration, shared state, and constants for the modular AI agent system.
"""

# Global configuration, shared state, and constants for the modular AI agent system.
import os

agents_config = {}
rag_retriever = None

skills_library = {
    "🛡️ Antimicrobial Stewardship": [
        "recommend_deescalation",
        "recommend_empiric_therapy",
        "alert_prolonged_antibiotic_use"
    ],
    "🦠 Infection Prevention and Control": [
        "IPC_reporting",
        "NHSN_criteria_evaluator",
        "recommend_isolation_precautions"
    ],
    "🔬 Research Assistant": [
        "search_pubmed",
        "suggest_journals_for_submission",
        "format_references"
    ],
    "🏥 Clinical Assistant": [
        "retrieve_guidelines",
        "explain_in_layman_language",
        "history_taking"
    ],
    "📚 Education Assistant": [
        "generate_board_exam_question",
        "generate_flash_cards",
        "create_educational_presentation"
    ],
    "🎼 Orchestrator": [
        # Note: Orchestrator capabilities are built-in and don't require skill selection
    ]
}

prefilled_agents = {
    "Example Stewardship Agent": {
        "agent_name": "SmartSteward",
        "agent_type": "🛡️ Antimicrobial Stewardship",
        "agent_mission": "Assist stewardship team in optimizing antibiotic use.",
        "skills": [
            "recommend_deescalation",
            "alert_prolonged_antibiotic_use",
            "recommend_empiric_therapy"
        ]
    },
    "Example Infection Control Agent": {
        "agent_name": "InfectoGuard",
        "agent_type": "🦠 Infection Prevention and Control",
        "agent_mission": "Support IPC team in monitoring outbreaks and adherence to precautions. When users ask about reportable diseases or reporting requirements, offer to help with the specific reporting process using available tools.",
        "skills": [
            "IPC_reporting",
            "NHSN_criteria_evaluator",
            "recommend_isolation_precautions"
        ]
    },
    "Example Research Assistant Agent": {
        "agent_name": "ResearchRanger",
        "agent_type": "🔬 Research Assistant",
        "agent_mission": "Assist in literature search, article summarization, and citation formatting.",
        "skills": [
            "search_pubmed",
            "suggest_journals_for_submission",
            "format_references"
        ]
    },
    "Example Clinical Assistant Agent": {
        "agent_name": "ClinicoPilot",
        "agent_type": "🏥 Clinical Assistant",
        "agent_mission": "Assist clinicians in diagnostic workflows and patient education by translating complex medical information into layman terms.",
        "skills": [
            "retrieve_guidelines",
            "explain_in_layman_language"
        ]
    },
    "Example Education Assistant Agent": {
        "agent_name": "EduMedCoach",
        "agent_type": "📚 Education Assistant",
        "agent_mission": "Generate educational materials like board exam questions, flash cards, and comprehensive presentations.",
        "skills": [
            "generate_board_exam_question",
            "generate_flash_cards",
            "create_educational_presentation"
        ]
    },
    "Example Orchestrator Agent": {
        "agent_name": "ID Maestro",
        "agent_type": "🎼 Orchestrator",
        "agent_mission": "Intelligently coordinate multiple ID agents to provide comprehensive analysis and recommendations.",
        "skills": [
            # Orchestrator uses built-in coordination capabilities
        ]
    }
}