File size: 1,336 Bytes
f70fcdd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from fastapi import FastAPI, Request
import requests
from dotenv import load_dotenv
import os
load_dotenv()
app = FastAPI()
# Configuration
HF_TOKEN = os.getenv("HF_TOKEN") # Token avec droits d'écriture
SPACES = ["OrganizedProgrammers/DocFinder", "OrganizedProgrammers/SpecSplitter"] # Liste des Spaces à redémarrer
def restart_space(space_id):
"""Redémarre un Space Hugging Face via l'API."""
api_url = f"https://api.huggingface.co/spaces/{space_id}/restart"
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
response = requests.post(api_url, headers=headers, verify=False)
if response.status_code == 200:
print(f"Space redémarré avec succès: {space_id}")
else:
print(f"Échec du redémarrage: {space_id} - {response.text}")
@app.post("/webhook")
async def handle_webhook(request: Request):
payload = await request.json()
# Vérifier si c'est une mise à jour de dataset
if (payload.get("repo", {}).get("type") == "dataset" and
payload.get("event", {}).get("action") == "update"):
print("Dataset mis à jour, redémarrage des Spaces...")
for space_id in SPACES:
restart_space(space_id)
return {"message": "Spaces mis à jour avec succès"}
return {"message": "Aucune action nécessaire"} |