import gradio as gr
from transformers import pipeline

clasificador = pipeline("sentiment-analysis", model="pysentimiento/robertuito-sentiment-analysis")

def puntuacion_sentimiento(texto):
    resultado = clasificador(texto)
    print(resultado)

    etiqueta = resultado[0]['label']

    if(etiqueta == 'POS'):
        respuesta = "Tu frase es positiva."
    elif(etiqueta == 'NEG'):
        respuesta = "Tu frase es negativa."
    else:
        respuesta = "Ni fu ni fa."
    return respuesta

demo = gr.Interface(
    fn=puntuacion_sentimiento, 
    inputs=gr.Textbox(), 
    outputs=gr.Textbox(),
    title="Analizador de Sentimientos",
    description="Esta es una muestra para probar <strong>modelos de IA</strong>",
    article="En este modelo, escribe una frase y pulsa en **Comprobar** para ver si tiene sentimientos Positivos, Negativos o Neutros.",
    submit_btn="Comprobar",
    fill_width=True
    )
demo.launch()