Update app.py

app.py

@@ -35,12 +35,14 @@ class HiveMind:
         self.pheromone_importance = {'food': 0.5, 'danger': 0.3, 'exploration': 0.2}
 
     def update_collective_memory(self, ant_memories):
-        for memory in ant_memories:
-            for position, info in memory:
-                if position not in self.collective_memory:
-                    self.collective_memory[position] = info
-                else:
-                    self.collective_memory[position] = (self.collective_memory[position] + info) / 2
+    for memory in ant_memories:
+        for position, info in memory:
+            if position not in self.collective_memory:
+                self.collective_memory[position] = info
+            else:
+                old_info = self.collective_memory[position]
+                new_info = tuple((old + new) / 2 for old, new in zip(old_info, info))
+                self.collective_memory[position] = new_info
 
     def update_global_strategy(self, ant_performances):
         best_ants = sorted(ant_performances, key=lambda x: x[1], reverse=True)[:5]