import streamlit as st
import pandas as pd
from filter_dataframe import filter_dataframe
import plotly.express as px


@st.cache_data
def get_domain_transfer_data():
    return pd.read_parquet("data/domain_transfer.parquet")


TITLE = "Cross-domain Transfer"

st.set_page_config(page_title=TITLE, page_icon="📈")

st.markdown(f"# {TITLE}")
st.write(
    """TODO: Description"""
)

df = get_domain_transfer_data()

st.plotly_chart(
    px.box(
        df,
        x="train_domain",
        y="macro_f1_score",
        color="test_domain",
        category_orders={"train_domain": ["news", "reviews", "social_media"]},
        labels={
            "train_domain": "Train Domain",
            "test_domain": "Test Domain",
            "macro_f1_score": "Macro F1 Score",
        }
    )
)