import awesome_streamlit as ast
import streamlit as st
from transformers import pipeline

import apps.summarization
import apps.home

st.set_page_config(
        page_title="Turna",
        page_icon="📖",
        layout='wide'
    )

PAGES = {
    "Turna": apps.home,
    "Text Summarization": apps.summarization
}


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 are available on [HF Hub](https://huggingface.co/collections/boun-tabi-LMG)"
)
st.sidebar.write(
    "Model source code available on [GitHub](https://github.com/boun-tabi-LMG/turkish-lm-tuner)"
)