Update app.py

app.py

@@ -482,19 +482,24 @@ async def upload_file(
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Upload failed: {str(e)}")
 
-@app.get("/list-files/")
-async def list_files():
+@app.get("/list-videos/")
+async def list_videos():
     try:
-        # 🔍 Récupérer la liste des fichiers du dataset
+        #  Récupérer la liste des fichiers du dataset
         files = api.list_repo_files(repo_id=DATASET_REPO, repo_type="dataset", token=HF_TOKEN)
 
-        # 🔄 Formater en JSON avec URLs complètes
-        files_list = [{"file_name": f, "url": f"https://huggingface.co/datasets/{DATASET_REPO}/blob/main/{f}"} for f in files]
+        #  Filtrer les fichiers pour ne garder que les vidéos (par exemple .mp4, .avi, .mov)
+        video_extensions = [".mp4", ".avi", ".mov", ".mkv", ".flv"]  # Ajouter d'autres extensions si nécessaire
+        video_files = [f for f in files if any(f.endswith(ext) for ext in video_extensions)]
 
-        return {"status": "success", "files": files_list}
+        #  Formater en JSON avec URLs complètes
+        videos_list = [{"file_name": f, "url": f"https://huggingface.co/datasets/{DATASET_REPO}/blob/main/{f}"} for f in video_files]
+
+        return {"status": "success", "videos": videos_list}
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to fetch files: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to fetch videos: {str(e)}")
+