import gradio as gr
from transformers import pipeline

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

def puntuacion_sentimientos(texto):
    resultado = clasificador(texto)
    print(resultado)
    etiqueta = resultado[0]["label"]
    if(etiqueta == "POS"):
        respuesta = "Tu frase muy positiva"
    elif etiqueta == "NEG":
        respuesta = "Tu frase muy negativa"
    else:
        respuesta = "ni fu ni fa"
    return respuesta

demo = gr.Interface(
    fn=puntuacion_sentimientos,
    inputs=gr.Textbox(),
    outputs=gr.Textbox(),
    title="Ejemplo Sentimientos de las frases",
    description="Esta es nuestra interfaz para probar <strong>modelos de UA</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(theme=gr.themes.Soft())