import streamlit as st

# Define pipeline stages
development_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?"
    ]}
]

def go_to(page_name, from_callback=False):
    """
    Updates the session_state page and optionally triggers a rerun.
    - from_callback=True if called inside an on_click callback
    """
    st.session_state.page = page_name
    if not from_callback:
        st.rerun()