from Packages import *
st.set_page_config(page_title="NIRS Utils", page_icon=":goat:", layout="wide",)
if 'interface' not in st.session_state:
    st.session_state['interface'] = 'simple'
from Modules import *
from Class_Mod.DATA_HANDLING import *


add_header()

pages_folder = Path("pages/")
# # TOC menu on the left
show_pages(
    [Page("app.py", "Home"),
     Page(str(pages_folder / "4-inputs.py"), "Inputs"),
     Page(str(pages_folder / "1-samples_selection.py"), "Samples Selection"),
     Page(str(pages_folder / "2-model_creation.py"), "Models Creation & Predictions"),

     ]
) 

with st.sidebar:
    interface = st.selectbox(label="Interface", options=['simple', 'advanced'], key='interface')
    # st.page_link(str(pages_folder / '1-samples_selection.py'))
    if st.session_state['interface'] == 'simple':
    #     st.page_link(str(pages_folder / '2-model_creation.py'))
        pass
    # if advanced interface, split Models Creation and Predictions
    elif st.session_state['interface'] == 'advanced':
        show_pages(
            [Page("app.py", "Home"),
             Page(str(pages_folder / "4-inputs.py"), "Inputs"),
             Page(str(pages_folder / "1-samples_selection.py"), "Samples Selection"),
             Page(str(pages_folder / "2-model_creation.py"), "Models Creation"),
             Page(str(pages_folder / "3-prediction.py"), "Predictions"),
            ]
        )
        # st.page_link(str(pages_folder / '2-model_creation.py'))
        # st.page_link(str(pages_folder / '3-prediction.py'))


# Page header
with st.container():
    st.subheader("Plateforme d'Analyses Chimiques pour l'Ecologie-PACE :goat:")
    st.title("NIRS Utils")
    st.markdown("#### This app allows users to perform visualization, pre-treatments, samples selection & predictions on their NIRS spectra.")
    header1, header2, header3,header4 = st.columns(4)
    if header1.button("Inputs"):
        st.switch_page(pages_folder / '4-inputs.py')
    if header2.button("Samples Selection"):
        st.switch_page(pages_folder / '1-samples_selection.py')
    if header3.button("Models Creation"):
        st.switch_page(pages_folder / '2-model_creation.py')
    if header4.button("Predictions"):
        st.switch_page(pages_folder / '3-prediction.py')
    st.markdown("We could add documentation here")
    st.write("Samples selection (PCA, [UMAP](https://umap-learn.readthedocs.io/en/latest/how_umap_works.html), ...), Predictive Modelling ([Pinard](https://github.com/GBeurier/pinard), [LWPLSR](https://doi.org/10.1002/cem.3209), ...), and Predictions using your data (CSV or DX files) and/or PACE NIRS Database.")