from orchestrator import Orchestrator
from agents.agent_json_analyser import AgentJsonAnalyser
from agents.agent_image_sorter import AgentImageSorter
from agents.agent_image_analyser import AgentImageAnalyser
from agents.agent_report_generator import AgentReportGenerator

from llm_classes.mistral_large import MistralLarge
from llm_classes.pixtral_12b import Pixtral12b
from llm_classes.ollama import Ollama

def test_orchestrator():
    # Initialisation des LLM
    json_llm = MistralLarge()
    image_sorter_llm = Pixtral12b()
    image_analyser_llm = Ollama()
    report_generator_llm = MistralLarge()
    
    # Création des agents
    json_agent = AgentJsonAnalyser(json_llm)
    image_sorter = AgentImageSorter(image_sorter_llm)
    image_analyser = AgentImageAnalyser(image_analyser_llm)
    report_generator = AgentReportGenerator(report_generator_llm)
    
    # Initialisation de l'orchestrateur avec les agents
    orchestrator = Orchestrator(
        output_dir="output/",
        json_agent=json_agent,
        image_sorter=image_sorter,
        image_analyser=image_analyser,
        report_generator=report_generator
    )
    
    # Exécution complète de l'orchestrateur
    orchestrator.executer()

if __name__ == "__main__":
    test_orchestrator()