Update app.py

app.py

@@ -126,10 +126,22 @@ orchestrator = AssistantAgent(
     function_map={}
 )
 
+# Custom speaker selection function (no LLM needed)
+def custom_select_speaker(last_speaker, selector, messages, agents):
+    """Select the next speaker in a fixed order: FoodRecognizer → SizeEstimator → NutritionFetcher → Orchestrator."""
+    if last_speaker is None:
+        return "FoodRecognizer"  # Start with FoodRecognizer
+    order = ["FoodRecognizer", "SizeEstimator", "NutritionFetcher", "Orchestrator"]
+    current_index = order.index(last_speaker)
+    next_index = (current_index + 1) % len(order)
+    return order[next_index]
+
+# Group Chat for Agent Coordination (no LLM, custom speaker selection)
 group_chat = GroupChat(
     agents=[food_recognizer, size_estimator, nutrition_fetcher, orchestrator],
     messages=[],
-    max_round=10
+    max_round=4,
+    select_speaker=custom_select_speaker  # Use custom speaker selection instead of LLM
 )
 manager = GroupChatManager(groupchat=group_chat)
 
@@ -150,7 +162,6 @@ def orchestrate_workflow(image, nutritionix_key):
     result = response[-1].get("content", "No output from agents.")
     print(f"Total time: {time.time() - start:.2f}s")
     return result
-
     
 # Gradio Interface
 interface = gr.Interface(