Update app.py

app.py

@@ -183,6 +183,44 @@ def int_to_bin(n: int, d: int) -> str:
 # Even bits → X offsets, odd bits → Y offsets, decreasing magnitudes.
 def layout_positions(d: int, base: float = 900.0):
     n = 1 << d # number of nodes = 2^d
+
+
+    if mode == "bipartite":
+        # Left: odd parity, Right: even parity (as requested)
+        odds  = [v for v in range(n) if (bin(v).count("1") % 2) == 1]
+        evens = [v for v in range(n) if (bin(v).count("1") % 2) == 0]
+
+        # Deterministic order (by integer id)
+        odds.sort()
+        evens.sort()
+
+        col_gap = max(400.0, base * 0.9)   # horizontal distance between columns
+        y_gap = max(12.0, base / max(1, (n // 2)))  # vertical spacing
+
+        pts = []
+
+        # Put odds on the left (x=0)
+        for idx, v in enumerate(odds):
+            x = 0.0
+            y = idx * y_gap
+            pts.append((v, x, y))
+
+        # Put evens on the right (x=col_gap)
+        for idx, v in enumerate(evens):
+            x = col_gap
+            y = idx * y_gap
+            pts.append((v, x, y))
+
+        # normalize to start at (0,0) like your current function
+        minx = min(x for _, x, _ in pts)
+        miny = min(y for _, _, y in pts)
+        pts2 = [(vid, x - minx, y - miny) for vid, x, y in pts]
+
+        width = col_gap
+        height = (len(odds) - 1) * y_gap if odds else 0.0
+        return pts2, width, height
+
+    
     dx, dy = [0.0] * d, [0.0] * d # per-dimension offsets
     for k in range(d): # dimension k
         tier = k // 2 # tier 0,1,2,...
@@ -311,7 +349,8 @@ def make_figure(d: int,
                 edge_w: int,
                 path: List[int],
                 scale_base: float,
-                subsel: dict | None = None):
+                subsel: dict | None = None, 
+                layout_mode: str = "default"):
     nodes, edges = build_hypercube(d)
     pts, width, height = layout_positions(d, base=scale_base)
     pos = {vid: (x, y) for vid, x, y in pts}
@@ -681,6 +720,16 @@ app.layout = html.Div(
                     style={"width": "60px"},
                 ),
                 html.Button("Flip", id="btn_flip", n_clicks=0),
+
+                # Bipartite Layout Button
+                html.Button(
+                    "Bipartite layout",
+                    id="btn_bipartite_layout",
+                    n_clicks=0,
+                    style={"background": "#6B7280", "color": "white"},
+                ),
+                dcc.Store(id="layout_mode_store", data="default"),
+
             ],
         ),
 
@@ -918,7 +967,8 @@ def update_path(clickData,
     Input("path_store", "data"),
     Input("mark_negations", "value"),
     Input("mark_distances", "value"),
-    Input("subpath_select_store", "data"),   # NEW
+    Input("subpath_select_store", "data"),
+    Input("layout_mode_store", "data"),
 )
 def render(d, show_labels_vals, path, mark_vals, mark_dist_vals, subsel):
     labels_vals = show_labels_vals or []
@@ -941,7 +991,8 @@ def render(d, show_labels_vals, path, mark_vals, mark_dist_vals, subsel):
         edge_w=DEFAULTS["edge_width"],
         path=path or [],
         scale_base=float(DEFAULTS["scale"]),
-        subsel=subsel or {},                  # NEW
+        subsel=subsel or {},
+        layout_mode=layout_mode or "default",    # NEW
     )
     return fig
 
@@ -1041,12 +1092,35 @@ def style_flip_subpath_button(subsel):
 
     if active:
         # selection mode ON
-        return {**base, "background": "#DC2626"}  # red
+        return {**base, "background": "#2563EB"}  # strong blue for selection
     else:
         # selection mode OFF
         return {**base, "background": "#2563EB"}  # blue
 
 
+@app.callback(
+    Output("layout_mode_store", "data"),
+    Input("btn_bipartite_layout", "n_clicks"),
+    State("layout_mode_store", "data"),
+    prevent_initial_call=True
+)
+def toggle_layout(n, mode):
+    mode = mode or "default"
+    return "bipartite" if mode == "default" else "default"
+
+
+@app.callback(
+    Output("btn_bipartite_layout", "style"),
+    Input("layout_mode_store", "data"),
+)
+def style_layout_button(mode):
+    base = {"color": "white", "border": "none", "padding": "6px 12px", "borderRadius": "8px", "cursor": "pointer"}
+    if mode == "bipartite":
+        return {**base, "background": "#059669"}  # green
+    return {**base, "background": "#6B7280"}      # gray
+
+
+
 
 if __name__ == "__main__":
     import os