feat(prediction):correct single prediction error

src/controllers/prediction_controller.py

@@ -48,9 +48,8 @@ def global_prediction_on_csv(file: UploadFile):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-def single_prediction(data: dict):
+def single_prediction_controller(data: dict):
     try:
-    # log dans la console
         if not isinstance(data, dict):
             raise HTTPException(status_code=400, detail="Invalid JSON payload")
 
@@ -61,12 +60,11 @@ def single_prediction(data: dict):
                 detail=f"Missing required columns: {missing_columns}"
             )
 
-        # Ici tu peux faire la prédiction avec data
+        # Ici tu peux faire la prédiction
         print("Received data for single prediction:", data)
-
         return {"message": "All required columns present", "to_do": "Not yet implemented"}
+
     except HTTPException:
         raise
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
-

src/routes/prediction_routes.py

@@ -1,5 +1,10 @@
 from fastapi import APIRouter, UploadFile, File
 from src.controllers import prediction_controller
+from pydantic import BaseModel
+from typing import Dict
+
+class SinglePredictionRequest(BaseModel):
+    __root__: Dict[str, float]
 
 router = APIRouter(prefix="/predict", tags=["predict"])
 
@@ -11,6 +16,6 @@ def predict_from_csv(file: UploadFile = File(...)):
     return prediction_controller.global_prediction_on_csv(file)
 
 @router.post("/single-prediction")
-def list_users():
-    return prediction_controller.single_prediction()
+def single_prediction(req: SinglePredictionRequest):
+    return prediction_controller.single_prediction(req.__root__)