Update server.ts

server.ts

@@ -2,20 +2,18 @@ import * as express from "express";
 
 const PORT = 7860;
 const BOT_USERNAME = "@smolSWE";
-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`;
 
 if (!process.env.WEBHOOK_SECRET || !process.env.HF_TOKEN) {
 	console.error(
-		"This app needs those env variables to be defined",
+		"This app needs the following env variables to be defined:",
 		"WEBHOOK_SECRET, HF_TOKEN"
 	);
 	process.exit();
 }
 
 const app = express();
-// parse HTTP request bodies as json
+
+// Parse HTTP request bodies as JSON
 app.use(express.json());
 
 app.get("/", (req, res) => {
@@ -24,33 +22,27 @@ app.get("/", (req, res) => {
 
 app.post("/", async (req, res) => {
 	if (req.header("X-Webhook-Secret") !== process.env.WEBHOOK_SECRET) {
-		console.error("incorrect secret");
-		return res.status(400).json({ error: "incorrect secret" });
+		console.error("Incorrect 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,
-		// 		""
-		// 	);
+		// Simple static response instead of AI-generated text
+		const continuationText = `You mentioned ${BOT_USERNAME}. Here's a helpful response! ${req.body.comment.content}`;
 
-            const continuationText = `You asked for ${req.body.comment.content}`;
-      
-			console.log(continuationText);
-			/// Finally, let's post it as a comment in the same discussion
-			const commentUrl = req.body.discussion.url.api + "/comment";
+		console.log(continuationText);
 
+		const commentUrl = req.body.discussion.url.api + "/comment";
+
+		try {
 			const commentApiResponse = await fetch(commentUrl, {
 				method: "POST",
 				headers: {
@@ -61,15 +53,15 @@ app.post("/", async (req, res) => {
 			});
 
 			const apiOutput = await commentApiResponse.json();
-
 			console.log(apiOutput);
-		} else {
-			console.error(`API Error`);
+		} catch (error) {
+			console.error("Failed to post comment:", error);
 		}
 	}
+
 	res.json({ success: true });
 });
 
 app.listen(PORT, () => {
-	console.debug(`server started at http://localhost:${PORT}`);
+	console.debug(`Server started at http://localhost:${PORT}`);
 });