import streamlit as st
from streamlit_option_menu import option_menu
from pages import main, about

if __name__ == '__main__':

    st.set_page_config(page_title="Generalized Additive Models", page_icon="🚀")

    page = option_menu(
        menu_title=None,
        options=["Home", "About"],
        icons=["house", "book"],
        menu_icon="cast",
        default_index=0,
        orientation="horizontal",
    )

    if page == "Home":
        st.switch_page("pages/main.py")
    elif page == "About":
        st.switch_page("pages/about.py")
        
    st.info("This app is for explaining the problem domain using Generalized Additive Models")