import gradio as gr
import os

try:
    from setfit import SetFitModel
    print("✓ SetFit importado correctamente")
except ImportError as e:
    print(f"❌ Error importando SetFit: {e}")
    raise e

# Cargar el modelo con manejo de errores
def load_model():
    try:
        print("Cargando modelo...")
        model = SetFitModel.from_pretrained("rovargasc/modelopruebaUNAL2")
        print("✓ Modelo cargado correctamente")
        return model
    except Exception as e:
        print(f"❌ Error cargando el modelo: {e}")
        return None

model = load_model()

def predict_text(text):
    """Función para hacer predicciones con el modelo"""
    if model is None:
        return "Error: El modelo no se pudo cargar"
    
    if not text.strip():
        return "Por favor ingresa un texto"
    
    try:
        prediction = model(text)
        return f"Predicción: {prediction}"
    except Exception as e:
        return f"Error en la predicción: {str(e)}"

# Crear la interfaz
interface = gr.Interface(
    fn=predict_text,
    inputs=gr.Textbox(
        lines=3, 
        placeholder="Escribe tu texto aquí...",
        label="Texto de entrada"
    ),
    outputs=gr.Textbox(label="Resultado"),
    title="Clasificador de Texto - SetFit Model",
    description="Ingresa un texto para obtener una predicción del modelo SetFit",
    examples=[
        ["This movie is amazing!"],
        ["I don't like this product"],
        ["chokes on its own depiction of upper-crust decorum."]
    ]
)

if __name__ == "__main__":
    interface.launch()