Update app.py

sauvegarde csv dans dataset hugging face

app.py

@@ -624,3 +624,29 @@ async def annotate(
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
+@app.post("/upload_csv/")
+async def upload_csv(
+    video_path: str = Form(...)
+):
+    try:
+        target_dir = UPLOAD_DIR
+        csv_filename = Path(video_path).stem + ".csv"
+        csv_path = target_dir / csv_filename
+        
+        if not csv_path.exists():
+            raise HTTPException(status_code=404, detail="Le fichier CSV n'existe pas.")
+        
+        # 🔥 Upload vers Hugging Face
+        api.upload_file(
+            path_or_fileobj=str(csv_path),
+            path_in_repo=f"csv/{csv_filename}",
+            repo_id=DATASET_REPO,
+            repo_type="dataset",
+            token=HF_TOKEN,
+        )
+        
+        return {"message": "Fichier CSV uploadé avec succès.", "csv_path": str(csv_path)}
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))