Minor tweaks after making video

README.md

@@ -23,6 +23,7 @@ It has been great fun making this Arena and watching LLMs duke it out!
 Quick links:
 - The [Live Arena](https://edwarddonner.com/connect-four/)  courtesy of amazing HuggingFace Spaces
 - The [GitHub repo](https://github.com/ed-donner/connect) for the code
+- My [video walkthrough](https://youtu.be/0OF-ChlKOQY) of the code
 - My [LinkedIn](https://www.linkedin.com/in/eddonner/) - I love connecting!
 
 If you'd like to learn more about this:  

prototype.ipynb

@@ -67,7 +67,7 @@
     "class Board:\n",
     "\n",
     "    def __init__(self):\n",
-    "        self.cells = [[0 for _ in range(7)] for _ in range(6)]\n",
+    "        self.cells = [[EMPTY for _ in range(7)] for _ in range(6)]\n",
     "        self.player = RED\n",
     "        self.winner = EMPTY\n",
     "\n",
@@ -245,7 +245,7 @@
     "    def system(self, board):\n",
     "        legal_moves = \", \".join(board.legal_moves())\n",
     "        return f\"\"\"You are an expert player of the board game Connect 4.\n",
-    "Players take turns to drop counters into one of 6 columns labelled A, B, C, D, E, F.\n",
+    "Players take turns to drop counters into one of 7 columns labelled A, B, C, D, E, F, G.\n",
     "The winner is the first player to get 4 coins in a row in a straight or diagonal line.\n",
     "You are playing with the {pieces[self.color]} coins.\n",
     "And your opponent is playing with the {pieces[self.color * -1]} coins.\n",
@@ -305,7 +305,9 @@
     "            messages=[\n",
     "                {\"role\": \"system\", \"content\": system},\n",
     "                {\"role\": \"user\", \"content\": user}\n",
-    "            ]\n",
+    "            ],\n",
+    "            response_format={\"type\": \"json_object\"}\n",
+    "            \n",
     "        )\n",
     "        self.process_move(reply.choices[0].message.content)"
    ]
@@ -329,7 +331,7 @@
    "source": [
     "board = Board()\n",
     "red = Player(\"gpt-4o-mini\", RED)\n",
-    "yellow = Player(\"gpt-4o-mini\", YELLOW)\n",
+    "yellow = Player(\"gpt-4o\", YELLOW)\n",
     "while not board.winner:\n",
     "    red.move(board)\n",
     "    print(board)\n",
@@ -337,6 +339,14 @@
     "        yellow.move(board)\n",
     "        print(board)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f73d9d16-9658-4e85-8d46-f75e6adf7051",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {