Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -36,26 +36,27 @@ 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 = []
 
-        # For layout niceness: compute a span for children under each parent
+        # scaling: large spacing early, smaller spacing deeper
+        scale = spread_factor ** (1.0 / stage_idx)
+
         for parent_order, parent_idx in enumerate(prev_nodes):
             px, py = node_positions[parent_idx]
 
-            # (2) give each parent its own block of vertical space
-            parent_block_size = len(options) * y_spacing_base * (1.5 ** stage_idx)
+            # each parent gets its own block of vertical space
+            parent_block_size = len(options) * y_spacing_base * scale
             base_y = py + (parent_order - (len(prev_nodes) - 1) / 2.0) * parent_block_size
 
             for opt_idx, opt in enumerate(options):
                 child_x = float(stage_idx)
 
-                # (1) scale spacing UP with depth
-                # (3) nonlinear spreading with exponential factor
-                offset = (opt_idx - (len(options) - 1) / 2.0) * (y_spacing_base * (1.5 ** stage_idx))
+                offset = (opt_idx - (len(options) - 1) / 2.0) * (y_spacing_base * scale)
 
                 child_y = base_y + offset
                 node_index = len(node_labels)
@@ -65,7 +66,6 @@ def build_tree_and_trace_path(selected_path):
                 edges.append((parent_idx, node_index))
                 next_nodes.append(node_index)
 
-        # After all parents, the next stage's parents are all next_nodes
         prev_nodes = next_nodes
 
     # Trace the single chosen path by walking children
@@ -74,16 +74,13 @@ def build_tree_and_trace_path(selected_path):
     current_node = 0
 
     for stage_idx, chosen_label in enumerate(selected_path, start=1):
-        # children of current_node
         children = [dst for (src, dst) in edges if src == current_node]
-        # find a child among these whose label equals chosen_label
         found_child = None
         for c in children:
             if node_labels[c] == chosen_label:
                 found_child = c
                 break
         if found_child is None:
-            # chosen label not found under this parent — stop tracing
             break
         highlight_edges.add((current_node, found_child))
         highlight_nodes.add(found_child)
@@ -131,9 +128,9 @@ def render():
 
     node_trace = go.Scatter(
         x=x_vals, y=y_vals,
-        mode='markers+text',
+        mode='markers',
         text=labels,
-        textposition="top center",
+        # textposition="top center",
         marker=dict(size=18, color=node_colors, line=dict(width=1, color='black')),
         hoverinfo="text"
     )