import streamlit as st
from .sa import predictor


def write():
    st.markdown(
        """
        # Arabic Sarcasm Detection

        This is a simple sarcasm detection app that uses the [MARBERT](https://huggingface.co/UBC-NLP/MARBERT) model trained on [ArSarcasm](https://github.com/iabufarha/ArSarcasm)
        """
    )

    input_text = st.text_input(
        "Enter your text here:",
    )
    if st.button("Predict"):
        with st.spinner("Predicting..."):
            prediction, scores = predictor.get_preds_from_sarcasm([input_text])
            st.write(f"Result: {prediction[0]}")
            detailed_score = {
                "Sarcastic": scores[0][0],
                "Not_Sarcastic": scores[0][1],
            }
            st.write("All scores:")
            st.write(detailed_score)