update app

app.py

@@ -235,6 +235,7 @@
 # # http://localhost:8000/docs#/
 # # (.venv) PS C:\Users\antoi\Documents\Work_Learn\Labeling-Deploy\FastAPI> uvicorn main:app --host 0.0.0.0 --port 8000 --workers 1
 
+
 from fastapi import FastAPI, UploadFile, File, HTTPException
 import cv2
 import torch
@@ -451,30 +452,42 @@ def convertir_sequences_en_json(dataframe):
 
 
 
-@app.post("/analyze_video/")
-async def analyze_video(user_name: str, video_url: Optional[str] = None):
+
+@app.post("/upload_video/")
+async def upload_video(user_name: str, file: UploadFile = File(...)):
     try:
-        # Log for debugging purposes
-        print(f"Received user_name: {user_name}")
-        print(f"Received video_url: {video_url}")
-
-        # Assurez-vous d'avoir une URL valide à ce stade
-        if not video_url:
-            raise HTTPException(status_code=400, detail="No valid video URL provided.")
-
-        # Analyser la vidéo via l'URL
-        dataframe_sequences = identifier_sequences_surfing(video_url, intervalle=1)
-        json_result = convertir_sequences_en_json(dataframe_sequences)
-
-        return {
-            "message": "Video analyzed successfully!",
-            "file_url": video_url,
-            "analysis": json_result
-        }
+        print(f"Received request to upload video for user: {user_name}")
+
+        # Sauvegarder le fichier temporairement
+        temp_file_path = f"/tmp/{file.filename}"
+        with open(temp_file_path, "wb") as buffer:
+            buffer.write(await file.read())
+
+        # Préparer l'upload sur Hugging Face
+        dataset_name = "2nzi/Video-Sequence-Labeling"
+        repo_path = f"{user_name}/raw/{file.filename}"
+
+        print(f"Uploading {temp_file_path} to Hugging Face at {repo_path}")
+
+        hf_api.upload_file(
+            path_or_fileobj=temp_file_path,
+            path_in_repo=repo_path,
+            repo_id=dataset_name,
+            repo_type="dataset",
+            token=api_key
+        )
+
+        # Générer l'URL finale
+        file_url = f"https://huggingface.co/datasets/{dataset_name}/resolve/main/{repo_path}"
+        print(f"File successfully uploaded to: {file_url}")
+
+        return {"status": "success", "file_url": file_url}
 
     except Exception as e:
-        print(f"Error during video analysis: {str(e)}")  # Log the error
-        raise HTTPException(status_code=500, detail=f"Failed to analyze video: {str(e)}")
+        print(f"Error during upload: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Upload failed: {str(e)}")
+
+
 
 
 # Fonction pour uploader une vidéo vers un dataset Hugging Face