Update app.py

app.py

@@ -500,7 +500,55 @@ async def list_videos():
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to fetch videos: {str(e)}")
 
+@app.get("/get-video-frames/{file_name}/")
+async def get_video_frames(file_name: str) -> List[dict]:
+    try:
+        # Récupérer l'URL du fichier vidéo dans le dataset
+        video_url = f"https://huggingface.co/datasets/{DATASET_REPO}/blob/main/{file_name}"
+
+        # Télécharger la vidéo
+        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+        response = requests.get(video_url, headers=headers)
+        
+        if response.status_code != 200:
+            raise HTTPException(status_code=404, detail="Video not found in the dataset")
+        
+        # Charger la vidéo depuis la réponse
+        video_bytes = io.BytesIO(response.content)
+        cap = cv2.VideoCapture(video_bytes)
+        
+        # Liste pour stocker les frames extraites
+        frames = []
 
+        # Extraire les 8 premières frames
+        for i in range(8):
+            ret, frame = cap.read()
+            if not ret:
+                break
+            
+            # Convertir le frame BGR en RGB pour Pillow
+            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            img = Image.fromarray(frame_rgb)
+            
+            # Sauvegarder l'image en format PNG dans un buffer
+            img_byte_arr = io.BytesIO()
+            img.save(img_byte_arr, format="PNG")
+            img_byte_arr.seek(0)
+            
+            # Ajouter l'image à la liste des frames avec une URL temporaire ou base64
+            frame_data = {
+                "frame_index": i,
+                "image": img_byte_arr.getvalue().encode("base64").decode("utf-8")  # image encodée en base64
+            }
+            frames.append(frame_data)
+
+        cap.release()
+
+        # Retourner les frames extraites
+        return {"status": "success", "frames": frames}
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to retrieve video frames: {str(e)}")