Update app.py

app.py

@@ -410,7 +410,11 @@ def make_figure(d: int,
     colors = []
     for v in nodes:
         base_size = node_r * 2
-        sizes.append(base_size * (3 if v in in_path else 1))
+
+        if mark_distances and v in neighbor_set:
+            sizes.append(base_size * 1.5)
+        else:
+            sizes.append(base_size * (3 if v in in_path else 1))
 
         if path and (v == path[0] or v == path[-1]):
             colors.append("#111")          # endpoints