Update app.py

app.py

@@ -431,7 +431,8 @@ def make_figure(d: int,
                 edge_w: int,
                 path: List[int],
                 scale_base: float,
-                subsel: dict | None = None, 
+                subsel: dict | None = None,
+                switchsel: dict | None = None,
                 layout_mode: str = "default"):
     nodes, edges = build_hypercube(d)
     pts, width, height = layout_positions(d, base=scale_base, mode=layout_mode)
@@ -450,7 +451,16 @@ def make_figure(d: int,
     if sel_active and sel_s is not None and sel_e is not None and path:
         selected_vertices = set(path[sel_s:sel_e + 1])
 
-    
+    # ---------- selected vertices for switch-dims (3 consecutive vertices) ----------
+    switchsel = switchsel or {}
+    switch_active = bool(switchsel.get("active"))
+    sw_s = switchsel.get("start_idx")
+    sw_e = switchsel.get("end_idx")
+
+    switch_vertices = set()
+    if switch_active and sw_s is not None and sw_e is not None and path:
+        switch_vertices = set(path[sw_s:sw_e + 1])
+
     
     # ---------- neighbors of path vertices ----------
     #neighbor_set = set()
@@ -575,6 +585,24 @@ def make_figure(d: int,
                 )
             )
 
+    # ---------- highlight switch-dims selection edges ----------
+    if switch_active and sw_s is not None and sw_e is not None and sw_e > sw_s:
+        for i in range(sw_s, sw_e):
+            a, b = path[i], path[i + 1]
+            x1, y1 = pos[a]
+            x2, y2 = pos[b]
+            edge_traces.append(
+                go.Scatter(
+                    x=[x1, x2],
+                    y=[y1, y2],
+                    mode="lines",
+                    line=dict(width=max(2, edge_w * 3), color="#DC2626"),
+                    opacity=1.0,
+                    hoverinfo="skip",
+                    name="switch dims selection",
+                )
+            )
+
     
     # ---------- nodes ----------
     xs = [pos[v][0] for v in nodes]
@@ -601,6 +629,8 @@ def make_figure(d: int,
         # --- size ---
         if sel_active and v in selected_vertices:
             sizes.append(base_size * 2.6)
+        elif switch_active and v in switch_vertices:
+            sizes.append(base_size * 2.6)
         elif mark_distances and v in neighbor_set:
             sizes.append(base_size * 1.4)
         else:
@@ -609,6 +639,8 @@ def make_figure(d: int,
         # --- color ---
         if sel_active and v in selected_vertices:
             colors.append("#2563EB")           # strong blue for selection
+        elif switch_active and v in switch_vertices:
+            colors.append("#DC2626")           # red for switch selection
         elif path and (v == path[0] or v == path[-1]):
             colors.append("#111")
         elif mark_negations and v in neg_set:
@@ -1146,6 +1178,7 @@ def update_path(clickData,
     Input("mark_negations", "value"),
     Input("mark_distances", "value"),
     Input("subpath_select_store", "data"),
+    Input("switch_dims_store", "data"),
     Input("layout_mode_store", "data"),
 )
 def render(d, show_labels_vals, path, mark_vals, mark_dist_vals, subsel, layout_mode):
@@ -1170,7 +1203,8 @@ def render(d, show_labels_vals, path, mark_vals, mark_dist_vals, subsel, layout_
         path=path or [],
         scale_base=float(DEFAULTS["scale"]),
         subsel=subsel or {},
-        layout_mode=layout_mode or "default",    # NEW
+        switchsel=switchsel or {},
+        layout_mode=layout_mode or "default",
     )
     return fig