from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from typing import List

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

app = FastAPI()

def sumar(a,b):
    return a+b

# Ruta de predicción
@app.post("/predict/")
async def predict(data: InputData):
    print(f"Data: {data}")
    try:
        input_data = data.data
        print(input_data)
        a = input_data[0]
        b = input_data[1]
        s = sumar(a,b)
        prediction = s
        print(prediction)
        #return {"prediction": prediction.tolist()}
        return {"prediction": prediction}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))