import streamlit as st  #Web App
from main import classify

demo_phrases = """ Here are some examples:
this is a phrase
is it neutral
nothing else to say
man I'm so damn angry
sarcasm lol
I love this product
"""

#title
st.title("Sentiment Analysis")

#subtitle
st.markdown("## A selection of popular sentiment analysis models -  hosted on 🤗 Spaces")

model_name = st.selectbox(
    'Select a pre-trained model',
    [
        'finiteautomata/bertweet-base-sentiment-analysis',
        'ahmedrachid/FinancialBERT-Sentiment-Analysis',
        'finiteautomata/beto-sentiment-analysis'
    ],
)

input_sentences = st.text_area("Sentences", value=demo_phrases, height=200)

data = input_sentences.split('\n')

if st.button("Classify"):
    st.write("Please allow a few minutes for the model to run/download")
    for i in range(len(data)):
        j = classify(model_name.strip(), data[i])[0]
        sentiment = j['label']
        confidence = j['score']
        st.write(f"{i}. {data[i]} :: Classification - {sentiment} with confidence {confidence}")
 

st.markdown("Link to the app - [image-to-text-app on 🤗 Spaces](https://huggingface.co/spaces/Amrrs/image-to-text-app)")