from transformers import pipeline
import gradio as gr

sentiment = pipeline("sentiment-analysis")

#"I really enjoyed my stay !"
#"Worst rental I ever got"

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

print(results)

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

def get_sentiment(text):
  sentiment_result = sentiment(text)[0]
  return sentiment_result['label'], sentiment_result['score']


interface = gr.Interface(
        fn=get_sentiment, 
        inputs=gr.Textbox(lines=3, label="Enter the review"), 
        outputs=[
            gr.Textbox(label="Sentiment"),   # first return value
            gr.Textbox(label="Score")
          ]
        )


interface.launch()