Déploiement automatique depuis GitLab CI - 2026-02-23 17:29:42

api.py

@@ -411,7 +411,7 @@ async def predict_from_file(file: UploadFile = File(...)):
         # Lire le contenu du fichier uploadé (en mémoire)
         contents = await file.read()
         # Convertir les octets en chaîne et lire le CSV avec séparateur ';'
-        df = pd.read_csv(io.StringIO(contents.decode('utf-8')), sep=';')
+        df = pd.read_csv(io.StringIO(contents.decode('utf-8')), sep=';', index_col='SK_ID_CURR')
         logger.info(f"Fichier CSV lu avec succès: {len(df)} lignes")
 
         # Vérifier et réordonner les colonnes selon l'ordre attendu
@@ -432,6 +432,10 @@ async def predict_from_file(file: UploadFile = File(...)):
         proba_defaults = [p[1] for p in probabilities]  # Probabilité de la classe 1 (défaut)
         predictions = [1 if p >= THRESHOLD else 0 for p in proba_defaults]
 
+        # Réintégrer l'index SK_ID_CURR avant l'enregistrement pour la détection de drift
+        X = X.copy()
+        X.insert(0, 'SK_ID_CURR', df.index)
+
         # Enregistrer les données pour la détection de drift
         save_and_upload_logs(X, predictions)