import uuid
import streamlit as st

import training
import validation
import prediction


# Session initialization
if "key" not in st.session_state:
    st.session_state["key"] = str(uuid.uuid4()).split("-")[-1]


def training_page():
    training.main()


def validation_page():
    validation.main()


def prediction_page():
    prediction.main()


page_names_to_funcs = {
    "Training": training_page,
    "Validation": validation_page,
    "Prediction": prediction_page,
}

selected_page = st.sidebar.selectbox("Select a page", page_names_to_funcs.keys())
page_names_to_funcs[selected_page]()