Change

app.py

@@ -1,111 +1,78 @@
 import os
 import gradio as gr
-from huggingface_hub import WebhooksServer, WebhookPayload, HfApi
+from fastapi import FastAPI, Request, HTTPException
+from huggingface_hub import HfApi
+import uvicorn
+from threading import Thread
 
 # Configuration
 HF_TOKEN = os.getenv("HF_TOKEN")
 WEBHOOK_SECRET = os.getenv("WEBHOOK_SECRET")
 SPACES_TO_RESTART = [
-    "OrganizedProgrammers/DpcFinder",
+    "OrganizedProgrammers/DocFinder",
     "OrganizedProgrammers/SpecSplitter"
 ]
 
-# 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
-    
-    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
-            )
+# FastAPI pour les webhooks
+fastapi_app = FastAPI()
 
-# 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}")
+@fastapi_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=401, detail="Invalid secret")
+    
+    try:
+        payload = await request.json()
+        print(f"📦 Payload reçu: {payload}")
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Invalid JSON: {str(e)}")
     
     # Vérifier si c'est une mise à jour de dataset
-    if payload.repo.type == "dataset" and payload.event.action == "update":
-        print(f"✅ Dataset {payload.repo.name} mis à jour!")
+    if (payload.get("repo", {}).get("type") == "dataset" and 
+        payload.get("event", {}).get("action") == "update"):
         
-        results = []
+        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)
-                success_msg = f"✅ Space redémarré: {space_id}"
-                print(success_msg)
-                results.append(success_msg)
+                print(f"✅ Space redémarré: {space_id}")
             except Exception as e:
-                error_msg = f"❌ Erreur redémarrage {space_id}: {str(e)}"
-                print(error_msg)
-                results.append(error_msg)
+                print(f"❌ Erreur redémarrage {space_id}: {e}")
         
-        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}"
-        }
+        return {"message": "Spaces mis à jour avec succès"}
+    
+    return {"message": "Aucune action nécessaire"}
 
-# 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.
-    """
+@fastapi_app.get("/health")
+async def health_check():
     return {
         "status": "healthy",
-        "message": "Webhook server is running",
-        "configured_spaces": SPACES_TO_RESTART
+        "spaces": SPACES_TO_RESTART
     }
 
-# 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 '❌'}")
+# Interface Gradio
+with gr.Blocks() as gradio_app:
+    gr.Markdown("""
+    # 🔄 Dataset Update Webhook Server
+    
+    **Status**: Serveur actif et en attente de webhooks
+    
+    **URL du webhook**: `/webhook`
     
-    # Le serveur se lance automatiquement
-    app.launch()
+    **Spaces surveillés**:
+    - OrganizedProgrammers/DocFinder  
+    - OrganizedProgrammers/SpecSplitter
+    """)
+
+# Monter Gradio sur FastAPI
+fastapi_app = gr.mount_gradio_app(fastapi_app, gradio_app, path="/")
+
+# Lancer le serveur
+if __name__ == "__main__":
+    uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)

requirements.txt

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