Update app.py

app.py

@@ -7,33 +7,22 @@ app = FastAPI()
 
 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"
+PROMPT = "Faites comme si vous étiez un robot qui répond aux discussions sur l'apprentissage automatique et répondez au commentaire suivant :\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.
+    Gère les requêtes GET et renvoie une réponse JSON simple sans utiliser dict().
     """
-    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
+    request_details = {
+        "method": request.method,
+        "host": request.headers.get("host", "unknown"),
+        "user_agent": request.headers.get("user-agent", "unknown"),
+        "query_test": request.query_params.get("test", None),
+        "message": "Webhook server is running. Use POST to interact with the bot."
+    }
+    print(f"GET request received: {request_details}")
+    return request_details
 
 @app.post("/")
 async def webhook(request: Request) -> Dict[str, Any]:
@@ -41,17 +30,14 @@ 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"
@@ -59,7 +45,6 @@ async def webhook(request: Request) -> Dict[str, Any]:
         ):
             print(f"Processing comment: {data['comment']['content']}")
 
-            # Appeler l'API BLOOM
             response = requests.post(
                 INFERENCE_URL,
                 headers={"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"},
@@ -73,7 +58,6 @@ async def webhook(request: Request) -> Dict[str, Any]:
                     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(
@@ -97,67 +81,16 @@ async def webhook(request: Request) -> Dict[str, Any]:
                 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"):
-        raise HTTPException(status_code=400, detail="Secret incorrect")
-
-    data = await request.json()
-    event = data.get("event")
-
-    if (
-        event.get("action") == "create"
-        and event.get("scope") == "discussion.comment"
-        and BOT_USERNAME in data.get("comment", {}).get("content", "")
-    ):
-        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()
-            continuation_text = output[0]["generated_text"].replace(
-                PROMPT + data["comment"]["content"], ""
-            )
-
-            comment_url = data["discussion"]["url"]["api"] + "/comment"
-            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:
-                return {"success": True}
-            else:
-                raise HTTPException(status_code=500, detail="Failed to post comment")
-        else:
-            raise HTTPException(status_code=500, detail="Inference API failed")
-
-    return {"success": False}
-
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)