from huggingface_hub import from_pretrained_fastai
import gradio as gr
from fastai.text.all import *

# Cargar modelo
repo_id = "nohamdou/emociones"
learner = from_pretrained_fastai(repo_id)
labels = learner.dls.vocab[1]

# Mostrar clases (sin que sea input)
class_info = "Clases: -0: sadness -1: joy -2: love -3: anger -4: fear -5: surprise\n"

# Función de predicción (igual)
def predict(text):
    pred, pred_idx, probs = learner.predict(text)
    return {labels[i]: float(probs[i]) for i in range(len(labels))}

# Interfaz corregida (mínimos cambios)
gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=2, placeholder="Escribe una frase..."),  # Solo Textbox
    outputs=gr.Label(num_top_classes=3),
    examples=[  # Ejemplos como listas simples
        "I'm feeling great today!",
        "This is terrifying", 
        "I'm heartbroken"
    ],
    title="Clasificador de emociones",
    description=class_info  # Info de clases aquí
).launch(share=True)