import streamlit as st
from st_pages import Page, show_pages, add_page_title
from transformers import pipeline
from datasets import load_dataset


class TextClassifier:
    bertPipe = pipeline( model="nlp-pw/test-model-3")


class AnalysisData:
    ds = load_dataset("nlp-pw/Disaster-Tweets-Normalized", revision="main", split='train')


def main():
    st.session_state["shared"] = True
    st.set_page_config(
        page_title="NLP - Project",
        page_icon="🤖",
        )
    add_page_title("NLP - Project")
    show_pages(
        [
            Page("pages/introduction.py", "Intro", "😊"),
            Page("pages/data_analysis.py", "Data and data", "📊"),
            Page("pages/run_models.py", "Run models", "🤖"),
        ]
    )

if __name__ == "__main__":
    main()