|
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() |
|
|
|
|