Merge branch 'main' of https://github.com/ItzyBitzySpider/pawnpawnpwn

backend/package.json

@@ -40,7 +40,7 @@
     "start:watch": "./node_modules/nodemon/bin/nodemon.js build/server.js",
     "client-connect": "node test/client-connect.js",
     "llm": "node test/llm.js",
-    "chess": "node test/chess.js"
+    "chess": "node test/illegalChess.js"
   },
   "devDependencies": {
     "@types/node": "^20.11.5",

backend/test/illegalChess.js

@@ -4,9 +4,4 @@ const FEN = "rnbqkb1r/1p2ppp1/3p4/p2n3p/3P4/3B1N2/PPP2PPP/RNBQK2R w KQkq - 0 7";
 const chess = new Chess(FEN)
 console.log(chess.ascii())
 
-while (!chess.isGameOver()) {
-  const moves = chess.moves()
-  const move = moves[Math.floor(Math.random() * moves.length)]
-  chess.move(move)
-}
-console.log(chess.pgn())
+chess.put({})

backend/test/llm.js

@@ -1,6 +1,7 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import dotenv from  'dotenv';
 dotenv.config();
+
 async function displayTokenCount(model, request) {
     const { totalTokens } = await model.countTokens(request);
     console.log("Token count: ", totalTokens);
@@ -12,38 +13,117 @@ async function displayChatTokenCount(model, chat, msg) {
     await displayTokenCount(model, { contents: [...history, msgContent] });
 }
 
-console.log(process.env.GEMINI_KEY)
 // Access your API key as an environment variable (see "Set up your API key" above)
 const genAI = new GoogleGenerativeAI(process.env.GEMINI_KEY);
+const FEN = "rnbqkb1r/1p2ppp1/3p4/p2n3p/3P4/3B1N2/PPP2PPP/RNBQK2R w KQkq - 0 7";
+
+async function run(prompt, fen) {
+    // For text-only input, use the gemini-pro model
+    const model = genAI.getGenerativeModel({ model: "gemini-pro" });
+    const chat = model.startChat({
+        history: [
+          {
+            role: "user",
+            parts: `Assume the role of an Next Generation Chess Interpreter. Players will describe their moves within 15 words and you are to parse them into valid chess moves. Your response can be one of the following:
+
+            1. (square, square), to move a piece from the first square to the second square. For example, ('e2', 'e4')
+            2. (square, square, piece), to promote a pawn to a piece. For example, ('e7', 'e8', 'q'). This promotes the pawn at e7 to a queen. The piece can be a 'q' (queen), 'r' (rook), 'b' (bishop), or 'n' (knight).
+            3. 'Invalid move', if the move does not make sense or is illegal.
+            If you understand, respond with 'Yes, I understand'. The current game state is provided by the following FEN: ${fen}`,
+          },
+          {
+            role: "model",
+            parts: "Yes, I understand.",
+          },
+        ],
+        generationConfig: {
+          maxOutputTokens: 1000,
+        },
+    });
+
+    try {
+        // const result = await model.generateContent(prompt);
+        const result = await chat.sendMessage(prompt)
+        const response = await result.response;
+        const text = response.text();
+        const safe = await check(text);
+        console.log(parseResponse(text), safe);
+    } catch (e) {
+        console.log(e);
+    }
+}
 
-async function run() {
+async function check(prompt, fen) {
     // For text-only input, use the gemini-pro model
     const model = genAI.getGenerativeModel({ model: "gemini-pro" });
-    const FEN = "rnbqkb1r/1p2ppp1/3p4/p2n3p/3P4/3B1N2/PPP2PPP/RNBQK2R w KQkq - 0 7";
     const chat = model.startChat({
         history: [
           {
             role: "user",
-            parts: `Assume the role of a Advanced Chess Interpreter. You will be given moves which are you are to interpret respond with a valid chess move. If you determine that the move is not possible, respond with 'Invalid'. If you understand, respond with 'Yes, I understand'. The current game state is provided by the following FEN: ${FEN}`,
+            parts: `Assume the role of an Next Generation Chess Interpreter. Given the FEN of the current game, you are to determine whether a move is legal. The input can be one of the following formats:
+
+            1. (square, square), to move a piece from the first square to the second square. For example, ('e2', 'e4') moves the piece at e2 to e4.
+            2. (square, square, piece), to promote a pawn to a piece. For example, ('e7', 'e8', 'q'), promotes the pawn at e7 to a queen. The piece can be a 'q' (queen), 'r' (rook), 'b' (bishop), or 'n' (knight).
+
+            If the move is legal, respond with 'True'. If the move is illegal, respond with 'False'. You should only have either 'True' or 'False' in your response.
+            If you understand, respond with 'Yes, I understand'. The current game state is provided by the following FEN: ${fen}`,
           },
           {
             role: "model",
             parts: "Yes, I understand.",
           },
         ],
         generationConfig: {
-          maxOutputTokens: 100,
+          maxOutputTokens: 1000,
         },
-      });
+    });
+
+    try {
+        // const result = await model.generateContent(prompt);
+        const result = await chat.sendMessage(prompt)
+        const response = await result.response;
+        const text = response.text();
+        return text === 'True';
+    } catch (e) {
+        console.log(e);
+    }
+}
+
+function parseResponse(response) {
+    console.log(response);
+    // accept only (square, square) or (square, square, piece) or 'Invalid move'
+
+    // if response contains 'Invalid move', return 'Invalid move'
+    if (response.includes('Invalid move')) {
+        return response;
+    }
 
-    // const prompt = 
-    const prompt = "advance the queen forward to make a check";
+    // check if response is in the format (square, square)
+    const moveRegex = /\(\'([abcdefgh]\d)\', \'([abcdefgh]\d)\'\)/;
+    const moveMatch = response.match(moveRegex);
+    if (moveMatch) {
+        const [_, square1, square2] = moveMatch;
+        return {square1, square2};
+    }
+
+    // check if response is in the format (square, square)
+    const promotionRegex = /\(\'([abcdefgh]\d)\', \'([abcdefgh])\d\', '([qrbn])'\)/;
+    const promotionMatch = response.match(promotionRegex);
+    if (promotionMatch) {
+        const [_, square1, square2, piece] = promotionMatch;
+        return {square1, square2, piece};
+    }
 
-    // const result = await model.generateContent(prompt);
-    const result = await chat.sendMessage(prompt)
-    const response = await result.response;
-    const text = response.text();
-    console.log(text);
+    return `Illegal Response: \n ${response}`; 
+
 }
+// user prompt
+const prompt1 = "capture the opponent's rook";
+const prompt2 = "advance and promote all my pawns";
+const prompt3 = "deliver a checkmate";
+const prompt4 = "('e2', 'e8', 'q')";
 
-run();
+run(prompt1, FEN);
+run(prompt2, FEN);
+run(prompt3, FEN);
+run(prompt4, FEN);