from fastapi import FastAPI
from pydantic import BaseModel
from inference import preprocess_and_predict

# Définir l'API
app = FastAPI()

# Définir un modèle d'entrée
class InputData(BaseModel):
    puiss_admin_98: int
    conso_urb: float
    conso_exurb: float
    masse_ordma_max: float
    marque: str
    typ_boite: str
    champ_v9 : str
    carrosserie: str
    gamme: str

# Endpoint pour prédiction
@app.post("/predict")
def predict(data: InputData):
    input_json = data.dict()
    result = preprocess_and_predict(input_json)
    return result