from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import numpy as np
from typing import List

class InputData(BaseModel):
    data: List[float]  # Lista de características numéricas (flotantes)

app = FastAPI()

def sumar(n1, n2):
    return n1 + n2

# Ruta de predicción
@app.post("/predict/")
async def predict(data: InputData):
    print(f"Data: {data}")
    try:
        # Convertir la lista de entrada a un array de NumPy para la predicción
        input_data = np.array(data.data).reshape(
            1, -1
        )        
        n1 = int(input_data[0][0])
        n2 = int(input_data[0][1])
        prediccion = sumar(n1,n2)        
        return {"prediction": [prediccion]}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))