import awesome_streamlit as ast
import streamlit as st
from home import write as write_home
from pos_tagger import write as write_pos_tagger

st.set_page_config(
    page_title="Hamshetsnag POS Tagger",
    page_icon="📚",
    layout='wide'
)

PAGES = {
    "Home": write_home,
    "POS Tagging": write_pos_tagger,
}

st.sidebar.title("Navigation")
selection = st.sidebar.radio("Pages", list(PAGES.keys()))

page = PAGES[selection]
# with st.spinner(f"Loading {selection} ..."):
ast.shared.components.write_page(page)

st.sidebar.header("Info")
st.sidebar.write(
    "Models available on [HF Hub](https://huggingface.co/onurkeles/hamshetsnag-pos-tagger)"
)
st.sidebar.write(
    "Model source code will be soon available on [GitHub](http://github.com/kelesonur)"
)