Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -23,11 +23,22 @@ choices_list = [
     ]}
 ]
 
-def build_tree_and_trace_path(selected_path):
+def build_tree_and_trace_path(selected_path, spread_factor=5.0):
     """
     Build tree nodes and edges. Then trace selected_path (one choice per stage)
     by walking children of the current node to find the matching label at each stage.
-    Returns: node_labels, node_positions, edges, highlight_edges, highlight_nodes
+
+    Parameters
+    ----------
+    selected_path : list of str
+        The path to highlight.
+    spread_factor : float (default=5.0)
+        Controls vertical spread. Larger values => more separation early,
+        less separation as depth increases.
+
+    Returns
+    -------
+    node_labels, node_positions, edges, highlight_edges, highlight_nodes
     """
     node_labels = ["Start"]
     node_positions = [(0.0, 0.0)]
@@ -36,14 +47,13 @@ def build_tree_and_trace_path(selected_path):
 
     prev_nodes = [0]  # nodes at previous stage (start)
     y_spacing_base = 1.0
-    spread_factor = 1.5  # tune this to taste
 
     # Build nodes and edges stage by stage
     for stage_idx, stage in enumerate(choices_list, start=1):
         options = stage["options"]
         next_nodes = []
 
-        # scaling: large spacing early, smaller spacing deeper
+        # scaling: huge spread at stage 1, tapering off deeper
         scale = spread_factor ** (1.0 / stage_idx)
 
         for parent_order, parent_idx in enumerate(prev_nodes):
@@ -89,6 +99,7 @@ def build_tree_and_trace_path(selected_path):
     return node_labels, node_positions, edges, highlight_edges, highlight_nodes
 
 
+
 def render():
     st.markdown(
         """