agents/orchestrator.py

from agents.fashion_agent import FashionAgent
from agents.barber_agent import BarberAgent
from agents.vision_agent import VisionAgent
from agents.voice_agent import VoiceAgent

class Orchestrator:

    def __init__(self):
        self.fashion = FashionAgent()
        self.barber = BarberAgent()
        self.vision = VisionAgent()
        self.voice = VoiceAgent()

    async def process(self, image, prompt):

        vision_result = await self.vision.analyze(image)

        fashion_result = await self.fashion.analyze(
            image,
            vision_result
        )

        barber_result = await self.barber.generate_style(
            image,
            prompt
        )

        return {
            "vision": vision_result,
            "fashion": fashion_result,
            "barber": barber_result
        }