agents/app/main.py

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from app.utils.predict import predict_quality
import uvicorn

app = FastAPI(title="CNN SP500 Predictor")

class InputData(BaseModel):
    data: list  # lista de listas com as features

@app.post("/predict")
def predict(input_data: InputData):
    try:
        prediction = predict_quality(input_data.data)
        return {"prediction": prediction}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

# Descomente se for rodar diretamente
# if __name__ == "__main__":
#     uvicorn.run(app, host="0.0.0.0", port=8000)