Update app.py

app.py

@@ -1,9 +1,6 @@
-from fastapi import FastAPI, Request, HTTPException
-from huggingface_hub import HfApi
 import os
-
-app = FastAPI()
-api = HfApi()
+import gradio as gr
+from huggingface_hub import WebhooksServer, WebhookPayload, HfApi
 
 # Configuration
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -13,33 +10,102 @@ SPACES_TO_RESTART = [
     "OrganizedProgrammers/SpecSplitter"
 ]
 
-@app.post("/webhook")
-async def handle_webhook(request: Request):
-    # Vérification du secret
-    received_secret = request.headers.get("X-Webhook-Secret")
-    if received_secret != WEBHOOK_SECRET:
-        raise HTTPException(status_code=400, detail="Invalid secret")
+# Initialiser l'API Hugging Face
+api = HfApi()
+
+# Interface utilisateur personnalisée (optionnelle)
+with gr.Blocks(title="Dataset Update Webhook Server") as ui:
+    gr.Markdown("""
+    # 🔄 Dataset Update Webhook Server
     
-    payload = await request.json()
+    Ce serveur redémarre automatiquement les Spaces configurés lorsqu'un dataset est mis à jour.
+    
+    ## Spaces surveillés :
+    - OrganizedProgrammers/DpcFinder
+    - OrganizedProgrammers/SpecSplitter
+    
+    ## Configuration
+    Assurez-vous que votre webhook Hugging Face pointe vers `/webhooks/dataset_update`
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### Status")
+            status_text = gr.Textbox(
+                value="🟢 Serveur en attente de webhooks...", 
+                label="État du serveur",
+                interactive=False
+            )
+        
+        with gr.Column():
+            gr.Markdown("### Dernière activité")
+            activity_log = gr.Textbox(
+                value="Aucune activité récente", 
+                label="Log d'activité",
+                lines=5,
+                interactive=False
+            )
+
+# Créer le serveur de webhooks avec l'interface personnalisée
+app = WebhooksServer(ui=ui, webhook_secret=WEBHOOK_SECRET)
+
+@app.add_webhook
+async def dataset_update(payload: WebhookPayload):
+    """
+    Webhook déclenché lors de la mise à jour d'un dataset.
+    """
+    print(f"🔔 Webhook reçu - Type: {payload.repo.type}, Action: {payload.event.action}")
+    print(f"📦 Repo: {payload.repo.name}")
     
     # Vérifier si c'est une mise à jour de dataset
-    if (payload.get("repo", {}).get("type") == "dataset" and 
-        payload.get("event", {}).get("action") == "update"):
+    if payload.repo.type == "dataset" and payload.event.action == "update":
+        print(f"✅ Dataset {payload.repo.name} mis à jour!")
         
-        print(f"Dataset {payload['repo']['name']} mis à jour!")
+        results = []
         
         # 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}")
+                success_msg = f"✅ Space redémarré: {space_id}"
+                print(success_msg)
+                results.append(success_msg)
             except Exception as e:
-                print(f"❌ Erreur redémarrage {space_id}: {e}")
+                error_msg = f"❌ Erreur redémarrage {space_id}: {str(e)}"
+                print(error_msg)
+                results.append(error_msg)
         
-        return {"message": "Spaces mis à jour avec succès"}
-    
-    return {"message": "Aucune action nécessaire"}
+        return {
+            "message": "Spaces mis à jour avec succès",
+            "dataset": payload.repo.name,
+            "spaces_restarted": SPACES_TO_RESTART,
+            "results": results
+        }
+    else:
+        print(f"ℹ️ Événement ignoré - Type: {payload.repo.type}, Action: {payload.event.action}")
+        return {
+            "message": "Aucune action nécessaire",
+            "reason": f"Type: {payload.repo.type}, Action: {payload.event.action}"
+        }
+
+# Webhook de test pour vérifier que le serveur fonctionne
+@app.add_webhook("/health")
+async def health_check(payload: WebhookPayload):
+    """
+    Endpoint de santé pour vérifier que le serveur fonctionne.
+    """
+    return {
+        "status": "healthy",
+        "message": "Webhook server is running",
+        "configured_spaces": SPACES_TO_RESTART
+    }
 
-@app.get("/")
-async def root():
-    return {"message": "Webhook server is running"}
+# Démarrer le serveur
+if __name__ == "__main__":
+    print("🚀 Démarrage du serveur de webhooks...")
+    print(f"📋 Spaces configurés: {SPACES_TO_RESTART}")
+    print(f"🔐 Secret configuré: {'✅' if WEBHOOK_SECRET else '❌'}")
+    print(f"🔑 Token HF configuré: {'✅' if HF_TOKEN else '❌'}")
+    
+    # Le serveur se lance automatiquement
+    app.launch()