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 es muy positiva"
    elif etiqueta == "NEG":
        respuesta = "Tu frase es muy negativa"
    else:
        respuesta = "ni fu ni fa"
    return respuesta

demo = gr.Interface(
    fn=puntuacion_sentimientos,
    inputs=gr.Textbox(placeholder="Escribe tu frase.", label="Frase para analizar"),
    outputs=gr.Textbox(label="Sentimiento de la frase"),
    title="Ejemplo Sentimientos de las frases",
    description="Esta es nuestra interfaz para probar <strong>modelos de IA</strong>",
    article="En este modelo, escribe una frase y pulsa en **Comprobar** para ver si tiene setimientos Positivos, Negativos o Neutros",
    submit_btn="Comprobar",
    fill_width=True
)
demo.launch(theme=gr.themes.Soft())