Create multiverse.py

my_pages/multiverse.py

@@ -0,0 +1,66 @@
+import streamlit as st
+import plotly.graph_objects as go
+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)]
+
+    # User selects one option for each stage
+    selected_path = []
+    st.subheader("Choose your path")
+    for stage, options in stage_options.items():
+        choice = st.selectbox(f"{stage}:", options, key=stage)
+        selected_path.append(choice)
+
+    # Create Sankey diagram data
+    labels = []
+    for stage in development_stages:
+        labels.extend(stage_options[stage])
+
+    # Create links between every option in one stage to all options in the next
+    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
+
+        for s in range(start_idx, start_idx + 3):
+            for t in range(next_start_idx, next_start_idx + 3):
+                source.append(s)
+                target.append(t)
+                value.append(1)  # constant flow thickness
+
+                # Highlight if both source and target are in selected path
+                if labels[s] == selected_path[i] and labels[t] == selected_path[i+1]:
+                    colors.append("rgba(0, 150, 0, 0.8)")
+                else:
+                    colors.append("rgba(200, 200, 200, 0.3)")
+
+    # Create Sankey diagram
+    fig = go.Figure(data=[go.Sankey(
+        arrangement="snap",
+        node=dict(
+            pad=15,
+            thickness=20,
+            line=dict(color="black", width=0.5),
+            label=labels,
+            color="rgba(100, 100, 200, 0.8)"
+        ),
+        link=dict(
+            source=source,
+            target=target,
+            value=value,
+            color=colors
+        )
+    )])
+
+    st.plotly_chart(fig, use_container_width=True)