import streamlit as st
from transformers import pipeline

text_classifier = pipeline("text-classification", model="nlptown/bert-base-multilingual-uncased-sentiment")

def main():
    st.title("Know The Mood of Your Writing")
    
    # Text input area for the user
    text = st.text_area("Enter your text:")

    if st.button("Go!"):
        if text:
            analyze_text_tone(text)
        else:
            st.warning("Please enter some text.")

def analyze_text_tone(text):
    classification_result = text_classifier(text)
    predicted_label = classification_result[0]['label']
    
    st.subheader("Text Tone Analysis Result:")
    st.write(f"Predicted Tone: {predicted_label}")

if __name__ == "__main__":
    main()