Update app.py

app.py

@@ -482,5 +482,18 @@ 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():
+    try:
+        # 🔍 Récupération des fichiers du dataset
+        files = 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]
+
+        return {"status": "success", "files": files_list}
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch files: {str(e)}")