import gradio as gr
from transformers import pipeline

# Cargar el modelo de clasificación de género literario
clasificador = pipeline("text-classification", model="TheBritishLibrary/bl-books-genre")

# Función para predecir el género del texto
def clasificar_genero(texto):
    resultado = clasificador(texto)
    print(resultado)
    etiqueta = resultado[0]["label"]
    puntuacion = resultado[0]["score"]
    respuesta = f"Género predicho: {etiqueta} (confianza: {puntuacion:.2f})"
    return respuesta

# Interfaz Gradio con personalización
demo = gr.Interface(
    fn=clasificar_genero,
    inputs=gr.Textbox(placeholder="Escribe tu Frase", label="Frase para Analizar"),
    outputs=gr.Textbox(label="Género del Texto"),
    title="Ejemplo Clasificación de Género Literario",
    description="Esta es nuestra interfaz para probar <strong>modelos de IA</strong> entrenados en textos literarios.",
    article="En este modelo, escribe una frase o párrafo y pulsa en <strong>¡Clasificar!</strong> para ver el género literario estimado.",
    theme="soft",
    submit_btn="¡Clasificar!",
    fill_width=True
)

demo.launch()