Update app.py

app.py

@@ -0,0 +1,57 @@
+from fastapi import FastAPI, Request, HTTPException
+import requests
+import os
+
+app = FastAPI()
+
+BOT_USERNAME = "@DiscussionBot"
+INFERENCE_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
+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.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)