import streamlit as st

from clarin_datasets.polemo_dataset import PolemoDataset
from clarin_datasets.abusive_clauses_dataset import AbusiveClausesDataset
from clarin_datasets.aspectemo_dataset import AspectEmoDataset

selected_dataset = st.sidebar.selectbox(
    "Choose a dataset to load",
    (
        "clarin-pl/polemo2-official",
        "laugustyniak/abusive-clauses-pl",
        "clarin-pl/aspectemo",
    ),
)

if selected_dataset == "clarin-pl/polemo2-official":
    dataset = PolemoDataset()
elif selected_dataset == "laugustyniak/abusive-clauses-pl":
    dataset = AbusiveClausesDataset()
elif selected_dataset == "clarin-pl/aspectemo":
    dataset = AspectEmoDataset()

dataset.load_data()
dataset.show_dataset()