from modules.processing.dime_agents.diplomacy_agent import DiplomacyAgent
from modules.processing.dime_agents.information_agent import InformationAgent
from modules.processing.dime_agents.military_agent import MilitaryAgent
from modules.processing.dime_agents.economy_agent import EconomyAgent

class LLMAnalysis:
    def __init__(self):
        self.diplomacy_agent = DiplomacyAgent()
        self.information_agent = InformationAgent()
        self.military_agent = MilitaryAgent()
        self.economy_agent = EconomyAgent()

    def analyze(self, text):
        diplomacy_analysis = self.diplomacy_agent.analyze(text)
        information_analysis = self.information_agent.analyze(text)
        military_analysis = self.military_agent.analyze(text)
        economy_analysis = self.economy_agent.analyze(text)

        return {
            "diplomacy": diplomacy_analysis,
            "information": information_analysis,
            "military": military_analysis,
            "economy": economy_analysis
        }
