#Importamos las librerias necesarias, en caso que quieras hacerlo en tu PC debes instar las librerias detalladas en requirements.txt
from transformers import pipeline
import gradio as gr
from transformers.utils import logging
logging.set_verbosity_error()


pipe = pipeline(
    "zero-shot-classification", 
    model="MoritzLaurer/xtremedistil-l6-h256-mnli-fever-anli-ling-binary"
)

# Etiquetas fijas (algunos rubros de consultas bancarias que se me ocurrieron)
etiquetas = [
    "Préstamos",
    "Tarjetas",
    "Transferencias",
    "Inversiones",
    "Seguros",
    "Beneficios"
    "Aplicación App"
    
]

def clasificar_consulta(texto):
    resultado = pipe(texto, candidate_labels=etiquetas)
    salida = "\n".join(
        [f"{label}: {round(score * 100, 2)}%" for label, score in zip(resultado["labels"], resultado["scores"])]
    )
    return salida


# Interfaz con Gradio
iface = gr.Interface(
    fn=clasificar_consulta,
    inputs=gr.Textbox(label="Consulta del cliente"),
    outputs=gr.Textbox(label="Clasificación de la consulta por rubro"),
    title="Clasificación de consultas hecha por clientes a su entidad financiera",
    description="Clasificá una consulta o reclamo y el modelo asignará un rubro."
)

iface.launch(share=True)