Update app.py

app.py

@@ -281,6 +281,7 @@ def make_figure(d: int,
                 show_bits: bool,
                 show_ints: bool,
                 mark_negations: bool,
+                mark_distances: bool,
                 node_r: int,
                 edge_w: int,
                 path: List[int],
@@ -292,6 +293,15 @@ def make_figure(d: int,
     path = path or []
     in_path = set(path)
 
+    # ---------- neighbors of path vertices ----------
+    neighbor_set = set()
+    if mark_distances and path:
+        for v in in_path:
+            for bit in range(d):
+                u = v ^ (1 << bit)
+                if u not in in_path:
+                    neighbor_set.add(u)
+    
     # ---------- negations (vertices + edges) ----------
     neg_set = set()
     neg_edges = set()
@@ -406,6 +416,8 @@ def make_figure(d: int,
             colors.append("#111")          # endpoints
         elif mark_negations and v in neg_set:
             colors.append("red")           # negation vertices
+        elif mark_distances and v in neighbor_set:
+            colors.append("yellow")           # negation vertices
         else:
             colors.append("#222")
 
@@ -498,8 +510,16 @@ app.layout = html.Div(
                         options=[{"label": " Mark negations", "value": "neg"}],
                         value=[],  # unchecked by default
                         style={"marginTop": "0px"},
-                    )
+                    ),
+                    html.Div(style={"height": "4px"}),  # small gap
+                    dcc.Checklist(
+                        id="mark_distances",
+                        options=[{"label": " Mark distances", "value": "dist"}],
+                        value=[],  # unchecked by default
+                        style={"marginTop": "0px"},
+                    ),
                 ]),
+
                 html.Div(),  # empty cell just to keep the grid tidy
             ],
         ),
@@ -725,8 +745,9 @@ def update_path(clickData,
     Input("show_labels", "value"),
     Input("path_store", "data"),
     Input("mark_negations", "value"),
+    Input("mark_distances", "value"),
 )
-def render(d, show_labels_vals, path, mark_vals):
+def render(d, show_labels_vals, path, mark_vals, mark_dist_vals):
     labels_vals = show_labels_vals or []
     show_bits = "bits" in labels_vals
     show_ints = "ints" in labels_vals
@@ -734,11 +755,15 @@ def render(d, show_labels_vals, path, mark_vals):
     mark_vals = mark_vals or []
     mark_negations = "neg" in mark_vals
 
+    mark_dist_vals = mark_dist_vals or []
+    mark_distances = "dist" in mark_dist_vals
+
     fig = make_figure(
         d=int(d),
         show_bits=show_bits,
         show_ints=show_ints,
         mark_negations=mark_negations,
+        mark_distances=mark_distances,
         node_r=DEFAULTS["node_radius"],
         edge_w=DEFAULTS["edge_width"],
         path=path or [],