Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -3,6 +3,34 @@ import streamlit as st
 import plotly.graph_objects as go
 from utils import go_to
 
+choices_list = [
+    {"label": "Data Collection", "icon": "📥", "questions": [
+        "Where will you source the data from?",
+        "How will you ensure data quality?",
+        "Will you balance classes?"
+    ]},
+    {"label": "Preprocessing", "icon": "🛠️", "questions": [
+        "What features will you select?",
+        "Will you impute missing values or remove them?",
+        "How will you handle outliers?"
+    ]},
+    {"label": "Model Selection", "icon": "🤖", "questions": [
+        "Which algorithms will you consider?",
+        "Will you use pre-trained models?",
+        "How will you handle hyperparameters?"
+    ]},
+    {"label": "Training", "icon": "🏋️", "questions": [
+        "What loss function will you use?",
+        "How will you split train/validation?",
+        "Will you use early stopping?"
+    ]},
+    {"label": "Evaluation", "icon": "📊", "questions": [
+        "What metrics will you use?",
+        "Will you test on unseen data?",
+        "Will you consider fairness metrics?"
+    ]}
+]
+
 def build_tree_and_trace_path(selected_path):
     """
     Build tree nodes and edges. Then trace selected_path (one choice per stage)
@@ -18,7 +46,7 @@ def build_tree_and_trace_path(selected_path):
     y_spacing_base = 1.0
 
     # Build nodes and edges stage by stage
-    for stage_idx, stage in enumerate(development_stages, start=1):
+    for stage_idx, stage in enumerate(choices_list, start=1):
         options = stage["questions"]
         next_nodes = []
 
@@ -67,11 +95,21 @@ def build_tree_and_trace_path(selected_path):
 
 
 def render():
-    st.title("Multiverse of Developer Decisions — Tree View")
+    st.markdown(
+        """
+        <div style='text-align: center; font-size:18px; color:gray;'>
+        Consider data about individuals who either paid their loans (green) or defaulted (red). <br>
+        Which model out of the two will you choose to give loan applications? <br>
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
+    st.markdown("---")
 
     # --- User picks one choice per stage via dropdowns ---
     selected_path = []
-    for stage in development_stages:
+    for stage in choices_list:
         # use a stable key per stage to avoid conflicts
         key = f"multiverse_choice_{stage['label']}"
         choice = st.selectbox(f"{stage['icon']} {stage['label']}", stage["questions"], key=key)
@@ -130,3 +168,9 @@ def render():
     )
 
     st.plotly_chart(fig, use_container_width=True)
+
+    st.markdown("---")
+    col1, col2, col3, col4 = st.columns([2, 1, 1, 1])
+    with col3:
+        if st.button("Go Home"):
+            go_to("home")