import streamlit as st

intro_page = st.Page(
    "application_pages/intro_page.py", title="Introduction", icon=":material/guardian:"
)
chat_page = st.Page(
    "application_pages/chat_app.py",
    title="Playground",
    icon=":material/sports_esports:",
)
evaluation_page = st.Page(
    "application_pages/evaluation_app.py",
    title="Evaluation",
    icon=":material/monitoring:",
)
train_classifier_page = st.Page(
    "application_pages/train_classifier.py",
    title="Train Classifier",
    icon=":material/fitness_center:",
)
llama_guard_fine_tuning_page = st.Page(
    "application_pages/llama_guard_fine_tuning.py",
    title="Fine-Tune LLama Guard",
    icon=":material/star:",
)
page_navigation = st.navigation(
    [
        intro_page,
        chat_page,
        evaluation_page,
        train_classifier_page,
        llama_guard_fine_tuning_page,
    ]
)
st.set_page_config(page_title="Guardrails Genie", page_icon=":material/guardian:")
page_navigation.run()