Update app.py

app.py

@@ -575,3 +575,44 @@ async def get_video_frames(file_name: str, frame_id: int) -> dict:
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erreur lors de l'extraction des frames : {str(e)}")
 
+
+
+@app.post("/annotate/")
+async def annotate(
+    video_path: str,
+    start_frame: int,
+    end_frame: int,
+    class_label: str,
+    player: str
+):
+    try:
+        # Vérifier si le fichier vidéo est valide
+        #if not os.path.exists(video_path):
+        #    raise HTTPException(status_code=400, detail="Le fichier vidéo n'existe pas.")
+
+        # Construire le nom du fichier CSV
+        csv_path = os.path.splitext(video_path)[0] + ".csv"
+
+        # Générer les frames
+        frames = list(range(start_frame, end_frame + 1))
+        data = {"frame": frames, "class": [class_label] * len(frames), "player": [player] * len(frames)}
+        df_new = pd.DataFrame(data)
+
+        # Vérifier si le fichier existe déjà
+        if os.path.exists(csv_path):
+            df_existing = pd.read_csv(csv_path)
+
+            # Fusionner sans doublons
+            df_combined = pd.concat([df_existing, df_new]).drop_duplicates(subset=["frame"], keep="last")
+
+            # Sauvegarder les données mises à jour
+            df_combined.to_csv(csv_path, index=False)
+        else:
+            df_new.to_csv(csv_path, index=False)
+
+        return {"message": "Fichier CSV mis à jour avec succès.", "csv_path": csv_path}
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+