Updated lines 388-392 with: def Dispersion(self): # """Calculate the dispersion of the swarm""" dispersion = np.std(self.gpos[:, 0]) + np.std(self.gpos[:, 1]) return dispersion

app.py

@@ -378,12 +378,17 @@ class GWO:
         plt.close()  # Close the figure to free up memory
         return Image.open(buf)
     
+    #def Dispersion(self):
+        #"""Calculate the dispersion of the swarm"""
+        #x, y = self.gpos[:, 0], self.gpos[:, 1]
+        #dx = x.max() - x.min()
+        #dy = y.max() - y.min()
+        #return (dx + dy) / 2.0
+
     def Dispersion(self):
-        """Calculate the dispersion of the swarm"""
-        x, y = self.gpos[:, 0], self.gpos[:, 1]
-        dx = x.max() - x.min()
-        dy = y.max() - y.min()
-        return (dx + dy) / 2.0
+        # """Calculate the dispersion of the swarm"""
+        dispersion = np.std(self.gpos[:, 0]) + np.std(self.gpos[:, 1])
+        return dispersion
 
     def plot_dispersion(self):
         """Plot the dispersion over time"""