Update app.py

app.py

@@ -535,11 +535,15 @@ async def get_video_frames(file_name: str, frame_id: int) -> dict:
         frames = []
         frame_size = (128, 128)
         csv_filename = Path(file_name).stem + ".csv"
-        csv_path = UPLOAD_DIR / csv_filename
         
+        # Télécharger le fichier CSV depuis Hugging Face
+        csv_url = f"https://huggingface.co/datasets/{DATASET_REPO}/resolve/main/csv/{csv_filename}"
+        csv_response = requests.get(csv_url, headers=headers)
         df_annotations = None
-        if csv_path.exists():
-            df_annotations = pd.read_csv(csv_path)
+        
+        if csv_response.status_code == 200:
+            csv_bytes = io.BytesIO(csv_response.content)
+            df_annotations = pd.read_csv(csv_bytes)
 
         for i in range(start_frame, end_frame):
             cap.set(cv2.CAP_PROP_POS_FRAMES, i)
@@ -557,7 +561,7 @@ async def get_video_frames(file_name: str, frame_id: int) -> dict:
             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["class"], row["player"]
+                class_label, player = row.get("class", None), row.get("player", None)
 
             frames.append({"frame_index": i, "image": img_base64, "class": class_label, "player": player})