Update app.py

app.py

@@ -8,6 +8,124 @@ BOT_USERNAME = "@DiscussionBot"
 INFERENCE_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
 PROMPT = "Pretend you're a robot responding to machine learning discussions and reply to the following comment :\n"
 
+@app.get("/")
+async def root(request: Request) -> Dict[str, Any]:
+    """
+    Gère les requêtes GET pour vérifier l'état du serveur et afficher les détails de la requête.
+    Inclut une gestion d'erreurs robuste pour capturer et signaler tout problème.
+    """
+    try:
+        # Préparer les détails de la requête
+        request_details = {
+            "method": request.method,
+            "headers": dict(request.headers),
+            "query_params": dict(request.query_params),
+            "message": "Webhook server is running. Use POST to interact with the bot."
+        }
+
+        # Validation des paramètres de requête (exemple : vérifier si un paramètre spécifique existe)
+        if "test_error" in request.query_params:
+            if request.query_params["test_error"] == "simulate_error":
+                # Simuler une erreur pour tester la gestion des exceptions
+                raise ValueError("Simulated error triggered by query parameter 'test_error=simulate_error'")
+
+        # Log de la requête GET reçue
+        print(f"GET request received: headers={request_details['headers']}, query_params={request_details['query_params']}")
+
+        return request_details
+
+    except ValueError as ve:
+        # Gérer les erreurs spécifiques (par exemple, paramètres invalides)
+        error_message = f"ValueError in GET request: {str(ve)}"
+        print(error_message)
+        raise HTTPException(status_code=400, detail=error_message)
+
+    except Exception as e:
+        # Gérer toutes les autres erreurs imprévues
+        error_message = f"Unexpected error in GET request: {str(e)}"
+        print(error_message)
+        raise HTTPException(status_code=500, detail=error_message)
+
+
+@app.post("/")
+async def webhook(request: Request) -> Dict[str, Any]:
+    """
+    Gère les requêtes POST des Webhooks pour traiter les commentaires mentionnant le bot.
+    """
+    try:
+        # Vérifier le secret du Webhook
+        if request.headers.get("X-Webhook-Secret") != os.getenv("WEBHOOK_SECRET"):
+            error_message = "Secret incorrect"
+            print(error_message)
+            raise HTTPException(status_code=400, detail=error_message)
+
+        # Extraire les données de la requête
+        data = await request.json()
+        event = data.get("event", {})
+
+        # Vérifier si l'événement est un commentaire de discussion mentionnant le bot
+        if (
+            event.get("action") == "create"
+            and event.get("scope") == "discussion.comment"
+            and BOT_USERNAME in data.get("comment", {}).get("content", "")
+        ):
+            print(f"Processing comment: {data['comment']['content']}")
+
+            # Appeler l'API BLOOM
+            response = requests.post(
+                INFERENCE_URL,
+                headers={"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"},
+                json={"inputs": PROMPT + data["comment"]["content"]},
+            )
+
+            if response.status_code == 200:
+                output = response.json()
+                print(f"BLOOM response: {output}")
+                continuation_text = output[0]["generated_text"].replace(
+                    PROMPT + data["comment"]["content"], ""
+                ).strip()
+
+                # Publier la réponse dans la discussion
+                comment_url = data["discussion"]["url"]["api"] + "/comment"
+                print(f"Posting to comment URL: {comment_url}")
+                comment_response = requests.post(
+                    comment_url,
+                    headers={
+                        "Authorization": f"Bearer {os.getenv('HF_TOKEN')}",
+                        "Content-Type": "application/json",
+                    },
+                    json={"comment": continuation_text},
+                )
+
+                if comment_response.status_code == 200:
+                    print("Comment posted successfully")
+                    return {"success": True}
+                else:
+                    error_message = f"Failed to post comment: {comment_response.text}"
+                    print(error_message)
+                    raise HTTPException(status_code=500, detail=error_message)
+            else:
+                error_message = f"Inference API failed: {response.text}"
+                print(error_message)
+                raise HTTPException(status_code=500, detail=error_message)
+
+        # Si les conditions ne sont pas remplies, retourner un succès sans action
+        print("Comment does not meet criteria, no action taken")
+        return {"success": False}
+
+    except HTTPException as he:
+        # Propager les exceptions HTTP (par exemple, secret incorrect)
+        raise he
+    except Exception as e:
+        # Gérer les erreurs inattendues
+        error_message = f"Unexpected error in POST request: {str(e)}"
+        print(error_message)
+        raise HTTPException(status_code=500, detail=error_message)
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+
 @app.post("/")
 async def webhook(request: Request):
     if request.headers.get("X-Webhook-Secret") != os.getenv("WEBHOOK_SECRET"):