Update my_pages/developer_decisions.py

my_pages/developer_decisions.py

@@ -1,35 +1,7 @@
 import streamlit as st
+from utils import development_stages
 
 def render():
-    # Define pipeline stages
-    stages = [
-        {"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?"
-        ]}
-    ]
-
     # Initialize session state
     if "active_stage" not in st.session_state:
         st.session_state.active_stage = None
@@ -37,8 +9,8 @@ def render():
     st.markdown("<h2 style='text-align: center;'>Developer Decisions Pipeline</h2>", unsafe_allow_html=True)
 
     # Display pipeline horizontally
-    cols = st.columns(len(stages))
-    for idx, stage in enumerate(stages):
+    cols = st.columns(len(development_stages))
+    for idx, stage in enumerate(development_stages):
         with cols[idx]:
             if st.button(f"{stage['icon']}\n{stage['label']}", key=f"btn_{idx}"):
                 st.session_state.active_stage = idx
@@ -47,7 +19,7 @@ def render():
 
     # Display sub-questions for the active stage
     if st.session_state.active_stage is not None:
-        stage = stages[st.session_state.active_stage]
+        stage = development_stages[st.session_state.active_stage]
         st.markdown(f"### {stage['icon']} {stage['label']}")
         for q in stage["questions"]:
             st.write(f"- {q}")