import gradio as gr
from transformers import pipeline

sentiment_pipeline = pipeline("sentiment-analysis")

reviews = ["I really enjoyed my stay!", "Worst rental I ever got"]
results = sentiment_pipeline(reviews)

for review, result in zip(reviews, results):
    print(f"Review: {review}")
    print(f"Sentiment: {result['label']}, Confidence: {result['score']:.4f}\n")

def get_sentiment(text):
    result = sentiment_pipeline(text)[0]  # Extract first result
    sentiment = result["label"]
    confidence = result["score"]
    return sentiment, confidence

interface = gr.Interface(fn = get_sentiment, inputs=gr.Textbox(label = "Enter the review :"), outputs = [gr.Textbox(label = "Sentiment :"), gr.Textbox(label = "Score :")], title = "Sentiment analysis protoype")
interface.launch()