modif app pour chrger automatiqueuement new model

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, HTTPException
+from fastapi import BackgroundTasks
 from pydantic import BaseModel
 from typing import Literal
 import pandas as pd
@@ -117,6 +118,17 @@ def get_latest_model_key():
     except Exception as e:
         raise RuntimeError(f"Erreur récupération modèle S3 : {e}")
 
+def load_latest_model():
+    global model
+    latest_model_key = get_latest_model_key()
+    print(f"Rechargement du modèle : {latest_model_key}")
+
+    response = s3.get_object(Bucket=S3_BUCKET, Key=latest_model_key)
+    model_bytes = io.BytesIO(response["Body"].read())
+
+    model = joblib.load(model_bytes)
+    print("Nouveau modèle chargé avec succès")
+    return latest_model_key
 
 @app.on_event("startup")
 def load_model():
@@ -156,4 +168,11 @@ def predict(data: InputData):
         print(f"Erreur prédiction : {e}")
         raise HTTPException(status_code=500, detail=str(e))
     
-    
+@app.post("/reload-model")
+def reload_model(background_tasks: BackgroundTasks):
+    """
+    Recharge le dernier modèle S3 sans redémarrer l'API.
+    """
+    background_tasks.add_task(load_latest_model)
+    latest_key = get_latest_model_key()
+    return {"status - Rechargement en arrière-plan modèle :", latest_key}