Akoda35 commited on
Commit
3322fd7
·
verified ·
1 Parent(s): 52daf81

Create agents/orchestrator.py

Browse files
Files changed (1) hide show
  1. agents/orchestrator.py +32 -0
agents/orchestrator.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from agents.fashion_agent import FashionAgent
2
+ from agents.barber_agent import BarberAgent
3
+ from agents.vision_agent import VisionAgent
4
+ from agents.voice_agent import VoiceAgent
5
+
6
+ class Orchestrator:
7
+
8
+ def __init__(self):
9
+ self.fashion = FashionAgent()
10
+ self.barber = BarberAgent()
11
+ self.vision = VisionAgent()
12
+ self.voice = VoiceAgent()
13
+
14
+ async def process(self, image, prompt):
15
+
16
+ vision_result = await self.vision.analyze(image)
17
+
18
+ fashion_result = await self.fashion.analyze(
19
+ image,
20
+ vision_result
21
+ )
22
+
23
+ barber_result = await self.barber.generate_style(
24
+ image,
25
+ prompt
26
+ )
27
+
28
+ return {
29
+ "vision": vision_result,
30
+ "fashion": fashion_result,
31
+ "barber": barber_result
32
+ }