modif app pour recéuperer dernier version du modele

app.py

@@ -88,21 +88,58 @@ S3_BUCKET = os.getenv("S3_BUCKET")
 MODEL_KEY = os.getenv("MODEL_KEY")
 S3_PREFIX = "mlflow/models/"
 s3 = boto3.client("s3")
-def latest_model(bucket:str, prefix: str):
-    response = s3.list_objects_v2(Bucket=bucket, Prefix=prefix)
-    if "content" not in response:
-        raise FileNotFoundError(f"Aucun modèle trouvé dans s3://{bucket}/{prefix}")
-    latest = max(response["content"], key=lambda x: x["LastModified"])
-    return latest["Key"]
+# def latest_model(bucket:str, prefix: str):
+#     response = s3.list_objects_v2(Bucket=bucket, Prefix=prefix)
+#     if "content" not in response:
+#         raise FileNotFoundError(f"Aucun modèle trouvé dans s3://{bucket}/{prefix}")
+#     latest = max(response["content"], key=lambda x: x["LastModified"])
+#     return latest["Key"]
+
+def get_latest_model_key():
+    """
+    Récupère automatiquement le dernier fichier modèle dans S3.
+    """
+    try:
+        # Liste tous les objets sous le préfixe
+        response = s3.list_objects_v2(
+            Bucket=S3_BUCKET,
+            Prefix=S3_PREFIX
+        )
+
+        if "Contents" not in response:
+            raise ValueError("Aucun modèle trouvé dans le bucket S3.")
+
+        # Filtrer uniquement les .joblib
+        models = [
+            obj for obj in response["Contents"]
+            if obj["Key"].endswith(".joblib")
+        ]
+
+        if not models:
+            raise ValueError("Aucun fichier .joblib trouvé.")
+
+        # Trier par LastModified (date d'upload dans S3)
+        models.sort(key=lambda x: x["LastModified"], reverse=True)
+
+        latest_key = models[0]["Key"]
+        print(f"📌 Dernier modèle détecté : {latest_key}")
+        return latest_key
+
+    except Exception as e:
+        raise RuntimeError(f"Erreur récupération modèle S3 : {e}")
+
+
+# === Chargement automatique du dernier modèle ===
+
 
 # === Chargement du modèle depuis S3 au démarrage ===
-model = None
+# model = None
 
 @app.on_event("startup")
 def load_model():
     global model
     try:
-        latest_model_key=latest_model(S3_BUCKET,S3_PREFIX)
+        latest_model_key = get_latest_model_key()
         print(f"Téléchargement du dernier modèle depuis s3://{S3_BUCKET}/{latest_model_key}")
 
         # print(f"Téléchargement du modèle depuis s3://{S3_BUCKET}/{MODEL_KEY}")