Reformat app.py + add package

app.py

@@ -1,44 +1,45 @@
-from fastapi import FastAPI, Request
-import requests
-from dotenv import load_dotenv
+from fastapi import FastAPI, Request, HTTPException
+from huggingface_hub import HfApi
 import os
 
-load_dotenv()
 app = FastAPI()
+api = HfApi()
 
 # 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}")
+HF_TOKEN = os.getenv("HF_TOKEN")
+WEBHOOK_SECRET = os.getenv("WEBHOOK_SECRET")
+SPACES_TO_RESTART = [
+    "OrganizedProgrammers/DpcFinder",
+    "OrganizedProgrammers/SpecSplitter"
+]
 
 @app.post("/webhook")
 async def handle_webhook(request: Request):
+    # Vérification du secret
     received_secret = request.headers.get("X-Webhook-Secret")
-    expected_secret = os.getenv("WEBHOOK_SECRET")
+    if received_secret != WEBHOOK_SECRET:
+        raise HTTPException(status_code=400, detail="Invalid secret")
     
-    # Vérifier l'authenticité
-    if received_secret != expected_secret:
-        return {"error": "incorrect secret"}, 400
     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)
+        print(f"Dataset {payload['repo']['name']} mis à jour!")
+        
+        # Redémarrer tous les Spaces liés
+        for space_id in SPACES_TO_RESTART:
+            try:
+                api.restart_space(space_id, token=HF_TOKEN)
+                print(f"✅ Space redémarré: {space_id}")
+            except Exception as e:
+                print(f"❌ Erreur redémarrage {space_id}: {e}")
         
         return {"message": "Spaces mis à jour avec succès"}
     
-    return {"message": "Aucune action nécessaire"}
+    return {"message": "Aucune action nécessaire"}
+
+@app.get("/")
+async def root():
+    return {"message": "Webhook server is running"}

requirements.txt

@@ -1,4 +1,5 @@
 fastapi
 uvicorn[standard]
 python-dotenv
-requests
+requests
+huggingface_hub