Update agents/brain.py

agents/brain.py

@@ -11,18 +11,17 @@ class AutoWarehouseAgent:
         task = detect_intent(message)
 
         if task == "slotting":
-            explanation, slotting_plan, _ = run_slotting_analysis(message, slotting_df)
-            return explanation, None, None, slotting_plan
+            explanation, slotting_plan = run_slotting_analysis(message, slotting_df)
+            return explanation, None, slotting_plan
 
         if task == "picking":
             explanation, route_img = run_picking_optimization(message, picking_df)
-            return explanation, route_img, None, None
+            return explanation, route_img, None
 
         if task == "report":
-            # Combine both
-            exp1, slot_plan, _ = run_slotting_analysis(message, slotting_df)
+            exp1, slot_plan = run_slotting_analysis(message, slotting_df)
             exp2, route_img = run_picking_optimization(message, picking_df)
             full_exp = exp1 + "\n\n" + exp2
-            return full_exp, route_img, None, slot_plan
+            return full_exp, route_img, slot_plan
 
-        return "Sorry, I could not understand your request.", None, None, None
+        return "Sorry, I could not understand your request.", None, None