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

import apps.summarization
import apps.home
import apps.paraphrasing
import apps.title_generation
import apps.sentiment
import apps.categorization
import apps.ner
import apps.pos_tagging
import apps.sts
import apps.nli
import apps.turna

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

PAGES = {
    "TURNA": apps.home,
    "Text Summarization": apps.summarization,
    "Text Paraphrasing": apps.paraphrasing,
    "News Title Generation": apps.title_generation,
    "Sentiment Classification": apps.sentiment,
    "Text Categorization": apps.categorization,
    "Named Entity Recognition": apps.ner,
    "Part-of-Speech Tagging": apps.pos_tagging,
    "Semantic Textual Similarity": apps.sts,
    "Natural Language Inference": apps.nli,
    "Text Generation": apps.turna,
}

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)"
)