Utsav2001's picture
Update app.py
5085916 verified
import gradio as gr
from transformers import pipeline
# Load the sentiment analysis pipeline
sentiment_pipeline = pipeline("sentiment-analysis")
# Function to analyze sentiment
def sentiment_analysis(message, history):
result = sentiment_pipeline(message)[0]
label = result["label"]
score = result["score"]
response = f"Sentiment: {label}, Confidence: {score:.2f}"
return response
# Function to record feedback
feedback_store = [] # A list to store feedback
def record_feedback(response, feedback):
feedback_store.append({"response": response, "feedback": feedback})
return f"Thank you for your feedback! ({len(feedback_store)} recorded)"
# Gradio Interface
with gr.Blocks() as demo:
chat = gr.ChatInterface(fn=sentiment_analysis, type="messages")
# Additional components for feedback
with gr.Row():
feedback_input = gr.Textbox(placeholder="Enter your feedback here", label="Feedback")
record_button = gr.Button("Submit Feedback")
# Feedback submission functionality
feedback_status = gr.Textbox(interactive=False, label="Feedback Status")
record_button.click(
fn=record_feedback,
inputs=[chat.output_component, feedback_input],
outputs=feedback_status,
)
demo.launch()