main.py

from fastapi import FastAPI, HTTPException

from ai.services import get_predictions
from core import config
from schemas.schemas import InputData, ResponseDataAPI

app = FastAPI(title=config.PROJECT_NAME, version=config.VERSION, openapi_url="/v1/openapi.json")


@app.get("/status")
def status():
    return "running"


@app.post("/v1/services/predict", tags=["services"], response_model=ResponseDataAPI)
def predict(data: InputData):
    
    X = data.data
    if len(X)==0:
        raise HTTPException(status_code=400, detail="Empty data")

    return {
        "inputData": data,
        "predictions": get_predictions(X),
    }