Octasf/HCIP

1import joblib2import pandas as pd3from fastapi import FastAPI, Request4from pydantic import BaseModel5 6# Cargar el modelo entrenado7model = joblib.load("modelo_ensemble1.joblib")8 9# Definir la aplicación FastAPI10app = FastAPI()11 12# Esquema de entrada con Pydantic13class EntradaDatos(BaseModel):14    vidas: int15    puntaje: int16    tiempo: int17    nivel: int18 19@app.get("/")20def leer_raiz():21    return {"mensaje": "✅ API funcionando correctamente."}22 23@app.post("/predecir")24async def predecir_dificultad(datos: EntradaDatos):25    entrada_df = pd.DataFrame([{26        "vidas": datos.vidas,27        "puntaje": datos.puntaje,28        "tiempo": datos.tiempo,29        "nivel": datos.nivel30    }])31 32    prediccion = model.predict(entrada_df)[0]33    resultado_redondeado = max(0, min(round(prediccion), 2))  # limita entre 0 y 234    return {"resultado": int(resultado_redondeado)}35