import streamlit as st

from models.toxicity_MODEL import text2toxicity


st.title('Toxicity Detection')
st.write('This tool classifies text as toxic or non-toxic using RuBERT.')

user_input = st.text_area("Enter text to classify", "Type your text here...")

if st.button('Classify'):
    toxicity_score = text2toxicity(user_input)
    st.write('Toxicity score:', toxicity_score)

    # Optional: Interpret the score for the user
    if toxicity_score > 0.5:
        st.write("This text is likely to be considered toxic.")
    else:
        st.write("This text is likely to be considered non-toxic.")