import streamlit as st
import transformers as t
import plotly.express as px
import pandas as pd

st.title("Phrase Feeling Analysis")
with st.spinner():
    classifier = t.pipeline("zero-shot-classification",
                            model="facebook/bart-large-mnli",
                            multi_class=True)

x = st.text_input("Enter your title here:")
candidate_labels = ['anger', 'sadness', 'fear', 'joy', 'interest',
                    'surprise', 'disgust', 'shame', 'guilt', 'compassion', 'other']

if x != "":

    with st.spinner():
        output = classifier(x, candidate_labels)
    # st.write(str(output))

    df = pd.DataFrame(dict(r=output['scores'], theta=output['labels']))
    fig = px.line_polar(df, r='r', theta='theta', line_close=True)
    fig.update_traces(fill='toself')

    st.plotly_chart(fig)