app.py

import joblib
import gradio as gr

modelo = joblib.load("modelo_rf_con_pace.pkl")

PACE_PROMEDIO = 150  # Cambia este valor por el promedio real de tu dataset

def predecir(q1, q2, pace):
    pred = modelo.predict([[q1, q2, pace]])[0]
    return int(pred[0]), int(pred[1]), int(pred[2])

demo = gr.Interface(
    fn=predecir,
    inputs=[
        gr.Number(label="Q1"),
        gr.Number(label="Q2"),
        gr.Number(label="Ritmo estimado (pace)", value=PACE_PROMEDIO)
    ],
    outputs=[
        gr.Number(label="Q3"),
        gr.Number(label="Q4"),
        gr.Number(label="Total")
    ],
    title="Predicción de Puntos de Baloncesto con Ritmo",
    description=f"Ingrese los puntos de Q1, Q2 y el ritmo estimado (pace). El valor sugerido de pace es {PACE_PROMEDIO}."
)

demo.launch()