import streamlit as st
from transformers import pipeline

# Title and Subtitle
st.title('Sentiment Analysis App')
st.subheader('Enter text to analyze sentiment')

# Text input box
text = st.text_area('Enter your text')

if text:
    # Analyze sentiment using the model
    pipe = pipeline(model="dancingninjas/sentiment-analysis-nlp")
    results = pipe(text)

    # Determine sentiment label
    sentiment_label = "Negative" if results[0]['label'] == 'LABEL_1' else "Positive"

    # Display sentiment label and probability
    st.subheader('Sentiment Analysis Result:')
    st.write(f'Sentiment: {sentiment_label}')
    st.write(f'Probability: {results[0]["score"]:.4f}')