import os
import streamlit as st

pages = {
    "home": [
        st.Page("home.py",title="about",icon=":material/home:")
    ],
    "generate embeddings": [
        st.Page("generate_embeddings.py", title="generate",icon=":material/dataset:")
    ],
    "annotation": [
        st.Page("train_model.py", title="train model",icon=":material/model_training:"),
        st.Page("apply_model.py", title="apply model",icon=":material/grade:"),
    ],
    "behavior discovery": [
        st.Page("explore.py", title="explore",icon=":material/search:")
    ],
}

pg = st.navigation(pages)
pg.run()