Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -5,41 +5,43 @@ from utils import development_stages
 def render():
     st.title("Multiverse of Developer Decisions")
 
-    # Generate placeholder options for each stage
-    stage_options = {}
-    for stage in development_stages:
-        stage_options[stage] = [f"{stage} Option {i}" for i in range(1, 4)]
+    # Extract stage labels
+    stage_labels = [stage["label"] for stage in development_stages]
 
-    # User selects one option for each stage
+    # Generate options from each stage's questions
+    stage_options = {stage["label"]: stage["questions"] for stage in development_stages}
+
+    # Let user pick one option from each stage
     selected_path = []
     st.subheader("Choose your path")
-    for stage, options in stage_options.items():
-        choice = st.selectbox(f"{stage}:", options, key=stage)
+    for stage_label, options in stage_options.items():
+        choice = st.selectbox(f"{stage_label}:", options, key=stage_label)
         selected_path.append(choice)
 
-    # Create Sankey diagram data
+    # Prepare Sankey diagram labels
     labels = []
-    for stage in development_stages:
-        labels.extend(stage_options[stage])
+    for stage_label in stage_labels:
+        labels.extend(stage_options[stage_label])
 
-    # Create links between every option in one stage to all options in the next
+    # Create links between options in consecutive stages
     source = []
     target = []
     value = []
     colors = []
 
-    for i in range(len(development_stages) - 1):
-        start_idx = i * 3
-        end_idx = (i + 1) * 3
-        next_start_idx = (i + 1) * 3
+    options_per_stage = [stage_options[stage] for stage in stage_labels]
+
+    for i in range(len(stage_labels) - 1):
+        start_idx = sum(len(opts) for opts in options_per_stage[:i])
+        next_start_idx = sum(len(opts) for opts in options_per_stage[:i+1])
 
-        for s in range(start_idx, start_idx + 3):
-            for t in range(next_start_idx, next_start_idx + 3):
+        for s in range(start_idx, start_idx + len(options_per_stage[i])):
+            for t in range(next_start_idx, next_start_idx + len(options_per_stage[i+1])):
                 source.append(s)
                 target.append(t)
-                value.append(1)  # constant flow thickness
+                value.append(1)  # constant thickness for now
 
-                # Highlight if both source and target are in selected path
+                # Highlight chosen path
                 if labels[s] == selected_path[i] and labels[t] == selected_path[i+1]:
                     colors.append("rgba(0, 150, 0, 0.8)")
                 else: