more feature-complete app

.prettierrc

@@ -0,0 +1,3 @@
+{
+  "useTabs": true
+}

.vscode/settings.json

@@ -1,3 +1,4 @@
 {
-	"editor.formatOnSave": true
+	"editor.formatOnSave": true,
+	"editor.defaultFormatter": "esbenp.prettier-vscode",
 }

server.ts

@@ -1,5 +1,11 @@
 import * as express from "express";
 
+const PORT = 7860;
+const BOT_USERNAME = "@discussion-bot";
+const INFERENCE_URL =
+	"https://api-inference.huggingface.co/models/bigscience/bloom";
+const PROMPT = `Pretend that you are a bot that replies to discussions about machine learning, and reply to the following comment:\n`;
+
 const app = express();
 // parse HTTP request bodies as json
 app.use(express.json());
@@ -8,18 +14,37 @@ app.get("/", (req, res) => {
 	res.json({ hello: "world" });
 });
 
-app.post("/", (req, res) => {
-
+app.post("/", async (req, res) => {
 	if (req.header("X-Webhook-Secret") !== process.env.WEBHOOK_SECRET) {
 		return res.status(400).json({ error: "incorrect secret" });
 	}
 	console.log(req.body);
+	const event = req.body.event;
+	if (
+		event.action === "create" &&
+		event.scope === "discussion.comment" &&
+		req.body.comment.content.includes(BOT_USERNAME)
+	) {
+		const response = await fetch(INFERENCE_URL, {
+			method: "POST",
+			body: JSON.stringify({ inputs: PROMPT + req.body.comment.content }),
+		});
+		if (response.ok) {
+			const output = await response.json();
+			const continuationText = output[0].generated_text.replace(
+				PROMPT + req.body.comment.content,
+				""
+			);
 
+			console.log(continuationText);
+			/// Finally, let's post it as a comment in the same discussion
+		} else {
+			console.error(`API Error`, await response.json());
+		}
+	}
 	res.json({ success: true });
 });
 
-
-const PORT = 7860;
 app.listen(PORT, () => {
 	console.debug(`server started at http://localhost:${PORT}`);
 });