Update app.py

app.py

@@ -576,41 +576,35 @@ async def extract_and_store_frames(file_name: str = Form(...)):
         raise HTTPException(status_code=500, detail=f"Erreur lors de l'extraction des frames : {str(e)}")
 
 
-STEP=240
+STEP = 240
+GITHUB_REPO = "Og31330/FA_Dataset"
+GITHUB_BASE_URL = f"https://raw.githubusercontent.com/{GITHUB_REPO}/main/"
+
 @app.get("/get-video-frames/")
 async def get_video_frames(file_name: str, frame_id: int) -> dict:
     try:
         frames = []
         csv_filename = Path(file_name).stem + ".csv"
-        csv_url = f"https://huggingface.co/datasets/{DATASET_REPO}/resolve/main/csv/{csv_filename}"
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-        
-        # Récupérer la liste des fichiers dans le dossier frames/{file_name}/
-        dataset_url = f"https://huggingface.co/api/datasets/{DATASET_REPO}/tree/main/frames/{Path(file_name).stem}/"
-        file_list_response = requests.get(dataset_url, headers=headers)
-        available_frames = set()
-        
-        if file_list_response.status_code == 200:
-            file_list = file_list_response.json()
-            available_frames = {file["path"] for file in file_list}
+        csv_url = f"{GITHUB_BASE_URL}/csv/{csv_filename}"
         
         # Charger le CSV des annotations si disponible
-        csv_response = requests.get(csv_url, headers=headers)
         df_annotations = None
-
+        csv_response = requests.get(csv_url)
+        
         if csv_response.status_code == 200:
             csv_bytes = io.BytesIO(csv_response.content)
             df_annotations = pd.read_csv(csv_bytes)
-
+        
         for i in range(frame_id, frame_id + STEP):
-            frame_filename = f"frames/{Path(file_name).stem}/{Path(file_name).stem}_{i}.png"
-            print(frame_filename)
-            if frame_filename not in available_frames:
+            frame_filename = f"{Path(file_name).stem}/{Path(file_name).stem}_{i}.png"
+            image_url = f"{GITHUB_BASE_URL}/frames/{frame_filename}"
+            
+            # Vérifier si l'image existe
+            image_response = requests.head(image_url)
+            if image_response.status_code != 200:
                 continue
-
-            image_url = f"https://huggingface.co/datasets/{DATASET_REPO}/resolve/main/{frame_filename}"
-            class_label, player = None, None
             
+            class_label, player = None, None
             if df_annotations is not None and i in df_annotations["frame"].values:
                 row = df_annotations[df_annotations["frame"] == i].iloc[0]
                 class_label, player = row.get("class", None), row.get("player", None)