| import gradio as gr |
| from transformers import pipeline |
|
|
| |
| def load_model(): |
| return pipeline('sentiment-analysis', model='cardiffnlp/twitter-roberta-base-sentiment') |
|
|
| sentiment_model = load_model() |
|
|
| |
| def analyze_sentiment(user_input): |
| result = sentiment_model(user_input)[0] |
| sentiment = result['label'] |
| |
| if sentiment in ['NEGATIVE', 'NEUTRAL']: |
| return "Stay positive! π You can handle anything that comes your way." |
| return "You're on the right track! Keep shining! π" |
|
|
| |
| def chatbot_ui(): |
| |
| interface = gr.Interface( |
| fn=analyze_sentiment, |
| inputs=gr.Textbox(label="Enter your text here:"), |
| outputs=gr.Textbox(label="Motivational Message"), |
| title="Student Sentiment Analysis Chatbot", |
| description="This chatbot detects your mood and provides positive or motivational messages." |
| ) |
|
|
| return interface |
|
|
| |
| if __name__ == "__main__": |
| chatbot_ui().launch() |
|
|