import streamlit as st 
from transformers import pipeline

st.set_page_config(page_title="AI Project", page_icon=":tada:", layout="wide")

with st.container():
    st.title("Hello! Welcome to the Sentiment Analysis App :wave:")
    st.header("By: Cory Khalilollahi")

with st.container():
    st.write("---")
    text = st.text_input("Please enter any text to use for the sentiment analysis:",
                         value="Hello! It is a pleasure to meet you!")
    model = st.selectbox(
        "Please select one of the following pre-trained models:",
        ["finiteautomata/bertweet-base-sentiment-analysis",
        'bhadresh-savani/distilbert-base-uncased-emotion',
        "nlptown/bert-base-multilingual-uncased-sentiment"]
    )
    analysis = pipeline("sentiment-analysis", model=model)

with st.container():
    st.write("---")
    if st.button("Analyze!"):
        result = analysis(text)
        sentiment = result[0]["label"] 
        score = result[0]["score"]
        
        sentiment = "Sentiment: " + sentiment
        score = "Score: " + str(score * 100)[:7] + "%"
        
        st.title(sentiment)
        st.title(score)