Chain reaction game in python.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a495bff7",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pygame\n",
    "import sys\n",
    "import random\n",
    "from math import *\n",
    "\n",
    "# Initialization of Pygame\n",
    "pygame.init()\n",
    "\n",
    "width = 400\n",
    "height = 400\n",
    "display = pygame.display.set_mode((width, height))\n",
    "clock = pygame.time.Clock()\n",
    "\n",
    "# Colors\n",
    "background = (21, 67, 96)\n",
    "border = (208, 211, 212)\n",
    "red = (231, 76, 60)\n",
    "white = (244, 246, 247)\n",
    "violet = (136, 78, 160)\n",
    "yellow = (244, 208, 63)\n",
    "green = (88, 214, 141)\n",
    "\n",
    "playerColor = [red, green, violet, yellow]\n",
    "\n",
    "font = pygame.font.SysFont(\"Times New Roman\", 30)\n",
    "\n",
    "blocks = 40\n",
    "noPlayers = 2  # Adjusted for 2 players\n",
    "\n",
    "pygame.display.set_caption(\"Chain Reaction %d Player\" % noPlayers)\n",
    "\n",
    "score = []\n",
    "for i in range(noPlayers):\n",
    "    score.append(0)\n",
    "\n",
    "players = []\n",
    "for i in range(noPlayers):\n",
    "    players.append(playerColor[i])\n",
    "\n",
    "d = blocks//2 - 2\n",
    "\n",
    "cols = int(width//blocks)\n",
    "rows = int(height//blocks)\n",
    "\n",
    "grid = []\n",
    "\n",
    "# Quit or Close the Game Window\n",
    "def close():\n",
    "    pygame.quit()\n",
    "    sys.exit()\n",
    "\n",
    "# Class for Each Spot in Grid\n",
    "class Spot():\n",
    "    def __init__(self):\n",
    "        self.color = border\n",
    "        self.neighbors = []\n",
    "        self.noAtoms = 0\n",
    "\n",
    "    def addNeighbors(self, i, j):\n",
    "        if i > 0:\n",
    "            self.neighbors.append(grid[i - 1][j])\n",
    "        if i < rows - 1:\n",
    "            self.neighbors.append(grid[i + 1][j])\n",
    "        if j < cols - 1:\n",
    "            self.neighbors.append(grid[i][j + 1])\n",
    "        if j > 0:\n",
    "            self.neighbors.append(grid[i][j - 1])\n",
    "\n",
    "# Initializing the Grid with \"Empty or 0\"\n",
    "def initializeGrid():\n",
    "    global grid, score, players\n",
    "    score = []\n",
    "    for i in range(noPlayers):\n",
    "        score.append(0)\n",
    "\n",
    "    players = []\n",
    "    for i in range(noPlayers):\n",
    "        players.append(playerColor[i])\n",
    "\n",
    "    grid = [[]for _ in range(cols)]\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            newObj = Spot()\n",
    "            grid[i].append(newObj)\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            grid[i][j].addNeighbors(i, j)\n",
    "\n",
    "# Draw the Grid in Pygame Window   \n",
    "def drawGrid(currentIndex):\n",
    "    r = 0\n",
    "    c = 0\n",
    "    for i in range(width//blocks):\n",
    "        r += blocks\n",
    "        c += blocks\n",
    "        pygame.draw.line(display, players[currentIndex], (c, 0), (c, height))\n",
    "        pygame.draw.line(display, players[currentIndex], (0, r), (width, r))\n",
    "\n",
    "# Draw the Present Situation of Grid\n",
    "def showPresentGrid(vibrate = 1):\n",
    "    r = -blocks\n",
    "    c = -blocks\n",
    "    padding = 2\n",
    "    for i in range(cols):\n",
    "        r += blocks\n",
    "        c = -blocks \n",
    "        for j in range(rows):\n",
    "            c += blocks\n",
    "            if grid[i][j].noAtoms == 0:\n",
    "                grid[i][j].color = border\n",
    "            elif grid[i][j].noAtoms == 1:\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (r + blocks/2 - d/2 + vibrate, c + blocks/2 - d/2, d, d))\n",
    "            elif grid[i][j].noAtoms == 2:\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (r + 5, c + blocks/2 - d/2 - vibrate, d, d))\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (r + d/2 + blocks/2 - d/2 + vibrate, c + blocks/2 - d/2, d, d))\n",
    "            elif grid[i][j].noAtoms == 3:\n",
    "                angle = 90\n",
    "                x = r + (d/2)*cos(radians(angle)) + blocks/2 - d/2\n",
    "                y = c + (d/2)*sin(radians(angle)) + blocks/2 - d/2\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (x - vibrate, y, d, d))\n",
    "                x = r + (d/2)*cos(radians(angle + 90)) + blocks/2 - d/2\n",
    "                y = c + (d/2)*sin(radians(angle + 90)) + 5\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (x + vibrate, y, d, d))\n",
    "                x = r + (d/2)*cos(radians(angle - 90)) + blocks/2 - d/2\n",
    "                y = c + (d/2)*sin(radians(angle - 90)) + 5\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (x - vibrate, y, d, d))\n",
    "\n",
    "    pygame.display.update()\n",
    "\n",
    "# Increase the Atom when Clicked\n",
    "def addAtom(i, j, color):\n",
    "    grid[i][j].noAtoms += 1\n",
    "    grid[i][j].color = color\n",
    "    if grid[i][j].noAtoms >= len(grid[i][j].neighbors):\n",
    "        overFlow(grid[i][j], color)\n",
    "    \n",
    "# Split the Atom when it Increases the \"LIMIT\"\n",
    "def overFlow(cell, color):\n",
    "    showPresentGrid()\n",
    "    cell.noAtoms = 0\n",
    "    for m in range(len(cell.neighbors)):\n",
    "        cell.neighbors[m].noAtoms += 1\n",
    "        cell.neighbors[m].color = color\n",
    "        if cell.neighbors[m].noAtoms >= len(cell.neighbors[m].neighbors):\n",
    "            overFlow(cell.neighbors[m], color)\n",
    "\n",
    "# Checking if Any Player has WON!\n",
    "def isPlayerInGame():\n",
    "    global score\n",
    "    playerScore = []\n",
    "    for i in range(noPlayers):\n",
    "        playerScore.append(0)\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            for k in range(noPlayers):\n",
    "                if grid[i][j].color == players[k]:\n",
    "                    playerScore[k] += grid[i][j].noAtoms\n",
    "    score = playerScore[:]\n",
    "\n",
    "# GAME OVER\n",
    "def gameOver(playerIndex):\n",
    "    while True:\n",
    "        for event in pygame.event.get():\n",
    "            if event.type == pygame.QUIT:\n",
    "                close()\n",
    "            if event.type == pygame.KEYDOWN:\n",
    "                if event.key == pygame.K_q:\n",
    "                    close()\n",
    "                if event.key == pygame.K_r:\n",
    "                    gameLoop()\n",
    "\n",
    "        text = font.render(\"Player %d Won!\" % (playerIndex + 1), True, white)\n",
    "        text2 = font.render(\"Press \\'r\\' to Reset!\", True, white)\n",
    "\n",
    "        display.blit(text, (width/3, height/3))\n",
    "        display.blit(text2, (width/3, height/2 ))\n",
    "\n",
    "        pygame.display.update()\n",
    "        clock.tick(60)\n",
    "\n",
    "\n",
    "def checkWon():\n",
    "    num = 0\n",
    "    for i in range(noPlayers):\n",
    "        if score[i] == 0:\n",
    "            num += 1\n",
    "    if num == noPlayers - 1:\n",
    "        for i in range(noPlayers):\n",
    "            if score[i]:\n",
    "                return i\n",
    "\n",
    "    return 9999\n",
    "\n",
    "# Function to make AI move\n",
    "def aiMove():\n",
    "    ownCells = []\n",
    "    opponentCells = []\n",
    "\n",
    "    # Collect AI player's cells and opponent cells\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            cell = grid[i][j]\n",
    "            if cell.color == players[1]:\n",
    "                ownCells.append((i, j))\n",
    "            elif cell.color != players[1]:\n",
    "                opponentCells.append((i, j))\n",
    "\n",
    "    chosenCell = None\n",
    "    threshold = 0.3 * cols * rows\n",
    "\n",
    "    # If AI hasn't reached the threshold yet, play randomly on own cells and empty cells\n",
    "    if len(ownCells) < threshold:\n",
    "        emptyCells = [(i, j) for i in range(cols) for j in range(rows) if grid[i][j].color == border]\n",
    "        chosenCell = random.choice(ownCells + emptyCells)\n",
    "    else:\n",
    "        # Once the threshold is reached, prioritize capturing opponent spots based on potential chain reactions\n",
    "        maxOpponentCapture = -1\n",
    "        for cell in opponentCells:\n",
    "            i, j = cell\n",
    "            potentialOpponentCapture = 0\n",
    "            for neighbor in grid[i][j].neighbors:\n",
    "                if neighbor.color == players[1]:\n",
    "                    potentialOpponentCapture += 1\n",
    "            if potentialOpponentCapture > maxOpponentCapture:\n",
    "                maxOpponentCapture = potentialOpponentCapture\n",
    "                chosenCell = (i, j)\n",
    "\n",
    "    if chosenCell:\n",
    "        addAtom(chosenCell[0], chosenCell[1], players[1])\n",
    "\n",
    "# Main Loop\n",
    "def gameLoop():\n",
    "    initializeGrid()\n",
    "    loop = True\n",
    "\n",
    "    turns = 0\n",
    "    \n",
    "    currentPlayer = 0\n",
    "\n",
    "    vibrate = .5\n",
    "\n",
    "    while loop:\n",
    "        for event in pygame.event.get():\n",
    "            if event.type == pygame.QUIT:\n",
    "                close()\n",
    "            if event.type == pygame.KEYDOWN:\n",
    "                if event.key == pygame.K_q:\n",
    "                    close()\n",
    "            if currentPlayer == 0:  # Human player's turn\n",
    "                if event.type == pygame.MOUSEBUTTONDOWN:\n",
    "                    x, y = pygame.mouse.get_pos()\n",
    "                    i = x//blocks\n",
    "                    j = y//blocks\n",
    "                    if grid[i][j].color == players[currentPlayer] or grid[i][j].color == border:\n",
    "                        turns += 1\n",
    "                        addAtom(i, j, players[currentPlayer])\n",
    "                        currentPlayer += 1\n",
    "                        if currentPlayer >= noPlayers:\n",
    "                            currentPlayer = 0\n",
    "                    if turns >= noPlayers:\n",
    "                        isPlayerInGame()\n",
    "        \n",
    "        display.fill(background)\n",
    "        # Vibrate the Atoms in their Cells\n",
    "        vibrate *= -1\n",
    "        \n",
    "        drawGrid(currentPlayer)\n",
    "        showPresentGrid(vibrate)\n",
    "\n",
    "        # AI's turn\n",
    "        if currentPlayer == 1:\n",
    "            aiMove()\n",
    "            currentPlayer = 0  # Change back to human player's turn\n",
    "        \n",
    "        pygame.display.update()\n",
    "\n",
    "        # Check for winner outside the ^loop\n",
    "        res = checkWon()\n",
    "        if res < 9999:\n",
    "            gameOver(res)\n",
    "\n",
    "        clock.tick(20)\n",
    "\n",
    "gameLoop()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "08c9f62a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting pygame\n",
      "  Obtaining dependency information for pygame from https://files.pythonhosted.org/packages/4d/84/5a03c47c76020c4f46354cf8c3f2d7926592c1ff9370f0ad91a3bb577b54/pygame-2.5.2-cp311-cp311-macosx_11_0_arm64.whl.metadata\n",
      "  Downloading pygame-2.5.2-cp311-cp311-macosx_11_0_arm64.whl.metadata (13 kB)\n",
      "Downloading pygame-2.5.2-cp311-cp311-macosx_11_0_arm64.whl (12.2 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m12.2/12.2 MB\u001b[0m \u001b[31m13.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
      "\u001b[?25hInstalling collected packages: pygame\n",
      "Successfully installed pygame-2.5.2\n",
      "Note: you may need to restart the kernel to use updated packages.\n"
     ]
    }
   ],
   "source": [
    "pip install pygame\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5e6a03cd",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pygame\n",
    "import sys\n",
    "import random\n",
    "from math import *\n",
    "\n",
    "# Initialization of Pygame\n",
    "pygame.init()\n",
    "\n",
    "width = 400\n",
    "height = 400\n",
    "display = pygame.display.set_mode((width, height))\n",
    "clock = pygame.time.Clock()\n",
    "\n",
    "# Colors\n",
    "background = (21, 67, 96)\n",
    "border = (208, 211, 212)\n",
    "red = (231, 76, 60)\n",
    "white = (244, 246, 247)\n",
    "violet = (136, 78, 160)\n",
    "yellow = (244, 208, 63)\n",
    "green = (88, 214, 141)\n",
    "\n",
    "playerColor = [red, green, violet, yellow]\n",
    "\n",
    "font = pygame.font.SysFont(\"Times New Roman\", 30)\n",
    "\n",
    "blocks = 40\n",
    "noPlayers = 2  # Adjusted for 2 players\n",
    "\n",
    "pygame.display.set_caption(\"Chain Reaction %d Player\" % noPlayers)\n",
    "\n",
    "score = []\n",
    "for i in range(noPlayers):\n",
    "    score.append(0)\n",
    "\n",
    "players = []\n",
    "for i in range(noPlayers):\n",
    "    players.append(playerColor[i])\n",
    "\n",
    "d = blocks//2 - 2\n",
    "\n",
    "cols = int(width//blocks)\n",
    "rows = int(height//blocks)\n",
    "\n",
    "grid = []\n",
    "\n",
    "# Quit or Close the Game Window\n",
    "def close():\n",
    "    pygame.quit()\n",
    "    sys.exit()\n",
    "\n",
    "# Class for Each Spot in Grid\n",
    "class Spot():\n",
    "    def __init__(self):\n",
    "        self.color = border\n",
    "        self.neighbors = []\n",
    "        self.noAtoms = 0\n",
    "\n",
    "    def addNeighbors(self, i, j):\n",
    "        if i > 0:\n",
    "            self.neighbors.append(grid[i - 1][j])\n",
    "        if i < rows - 1:\n",
    "            self.neighbors.append(grid[i + 1][j])\n",
    "        if j < cols - 1:\n",
    "            self.neighbors.append(grid[i][j + 1])\n",
    "        if j > 0:\n",
    "            self.neighbors.append(grid[i][j - 1])\n",
    "\n",
    "# Initializing the Grid with \"Empty or 0\"\n",
    "def initializeGrid():\n",
    "    global grid, score, players\n",
    "    score = []\n",
    "    for i in range(noPlayers):\n",
    "        score.append(0)\n",
    "\n",
    "    players = []\n",
    "    for i in range(noPlayers):\n",
    "        players.append(playerColor[i])\n",
    "\n",
    "    grid = [[]for _ in range(cols)]\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            newObj = Spot()\n",
    "            grid[i].append(newObj)\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            grid[i][j].addNeighbors(i, j)\n",
    "\n",
    "# Draw the Grid in Pygame Window   \n",
    "def drawGrid(currentIndex):\n",
    "    r = 0\n",
    "    c = 0\n",
    "    for i in range(width//blocks):\n",
    "        r += blocks\n",
    "        c += blocks\n",
    "        pygame.draw.line(display, players[currentIndex], (c, 0), (c, height))\n",
    "        pygame.draw.line(display, players[currentIndex], (0, r), (width, r))\n",
    "\n",
    "# Draw the Present Situation of Grid\n",
    "def showPresentGrid(vibrate = 1):\n",
    "    r = -blocks\n",
    "    c = -blocks\n",
    "    padding = 2\n",
    "    for i in range(cols):\n",
    "        r += blocks\n",
    "        c = -blocks \n",
    "        for j in range(rows):\n",
    "            c += blocks\n",
    "            if grid[i][j].noAtoms == 0:\n",
    "                grid[i][j].color = border\n",
    "            elif grid[i][j].noAtoms == 1:\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (r + blocks/2 - d/2 + vibrate, c + blocks/2 - d/2, d, d))\n",
    "            elif grid[i][j].noAtoms == 2:\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (r + 5, c + blocks/2 - d/2 - vibrate, d, d))\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (r + d/2 + blocks/2 - d/2 + vibrate, c + blocks/2 - d/2, d, d))\n",
    "            elif grid[i][j].noAtoms == 3:\n",
    "                angle = 90\n",
    "                x = r + (d/2)*cos(radians(angle)) + blocks/2 - d/2\n",
    "                y = c + (d/2)*sin(radians(angle)) + blocks/2 - d/2\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (x - vibrate, y, d, d))\n",
    "                x = r + (d/2)*cos(radians(angle + 90)) + blocks/2 - d/2\n",
    "                y = c + (d/2)*sin(radians(angle + 90)) + 5\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (x + vibrate, y, d, d))\n",
    "                x = r + (d/2)*cos(radians(angle - 90)) + blocks/2 - d/2\n",
    "                y = c + (d/2)*sin(radians(angle - 90)) + 5\n",
    "                pygame.draw.ellipse(display, grid[i][j].color, (x - vibrate, y, d, d))\n",
    "\n",
    "    pygame.display.update()\n",
    "\n",
    "# Increase the Atom when Clicked\n",
    "def addAtom(i, j, color):\n",
    "    grid[i][j].noAtoms += 1\n",
    "    grid[i][j].color = color\n",
    "    if grid[i][j].noAtoms >= len(grid[i][j].neighbors):\n",
    "        overFlow(grid[i][j], color)\n",
    "    \n",
    "# Split the Atom when it Increases the \"LIMIT\"\n",
    "def overFlow(cell, color):\n",
    "    showPresentGrid()\n",
    "    cell.noAtoms = 0\n",
    "    for m in range(len(cell.neighbors)):\n",
    "        cell.neighbors[m].noAtoms += 1\n",
    "        cell.neighbors[m].color = color\n",
    "        if cell.neighbors[m].noAtoms >= len(cell.neighbors[m].neighbors):\n",
    "            overFlow(cell.neighbors[m], color)\n",
    "\n",
    "# Checking if Any Player has WON!\n",
    "def isPlayerInGame():\n",
    "    global score\n",
    "    playerScore = []\n",
    "    for i in range(noPlayers):\n",
    "        playerScore.append(0)\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            for k in range(noPlayers):\n",
    "                if grid[i][j].color == players[k]:\n",
    "                    playerScore[k] += grid[i][j].noAtoms\n",
    "    score = playerScore[:]\n",
    "\n",
    "# GAME OVER\n",
    "def gameOver(playerIndex):\n",
    "    while True:\n",
    "        for event in pygame.event.get():\n",
    "            if event.type == pygame.QUIT:\n",
    "                close()\n",
    "            if event.type == pygame.KEYDOWN:\n",
    "                if event.key == pygame.K_q:\n",
    "                    close()\n",
    "                if event.key == pygame.K_r:\n",
    "                    gameLoop()\n",
    "\n",
    "        text = font.render(\"Player %d Won!\" % (playerIndex + 1), True, white)\n",
    "        text2 = font.render(\"Press \\'r\\' to Reset!\", True, white)\n",
    "\n",
    "        display.blit(text, (width/3, height/3))\n",
    "        display.blit(text2, (width/3, height/2 ))\n",
    "\n",
    "        pygame.display.update()\n",
    "        clock.tick(60)\n",
    "\n",
    "\n",
    "def checkWon():\n",
    "    num = 0\n",
    "    for i in range(noPlayers):\n",
    "        if score[i] == 0:\n",
    "            num += 1\n",
    "    if num == noPlayers - 1:\n",
    "        for i in range(noPlayers):\n",
    "            if score[i]:\n",
    "                return i\n",
    "\n",
    "    return 9999\n",
    "\n",
    "# Function to make AI move\n",
    "def aiMove():\n",
    "    ownCells = []\n",
    "    opponentCells = []\n",
    "\n",
    "    # Collect AI player's cells and opponent cells\n",
    "    for i in range(cols):\n",
    "        for j in range(rows):\n",
    "            cell = grid[i][j]\n",
    "            if cell.color == players[1]:\n",
    "                ownCells.append((i, j))\n",
    "            elif cell.color != players[1]:\n",
    "                opponentCells.append((i, j))\n",
    "\n",
    "    chosenCell = None\n",
    "    threshold = 0.3 * cols * rows\n",
    "\n",
    "    # If AI hasn't reached the threshold yet, play randomly on own cells and empty cells\n",
    "    if len(ownCells) < threshold:\n",
    "        emptyCells = [(i, j) for i in range(cols) for j in range(rows) if grid[i][j].color == border]\n",
    "        chosenCell = random.choice(ownCells + emptyCells)\n",
    "    else:\n",
    "        # Once the threshold is reached, prioritize capturing opponent spots based on potential chain reactions\n",
    "        maxOpponentCapture = -1\n",
    "        for cell in opponentCells:\n",
    "            i, j = cell\n",
    "            potentialOpponentCapture = 0\n",
    "            for neighbor in grid[i][j].neighbors:\n",
    "                if neighbor.color == players[1]:\n",
    "                    potentialOpponentCapture += 1\n",
    "            if potentialOpponentCapture > maxOpponentCapture:\n",
    "                maxOpponentCapture = potentialOpponentCapture\n",
    "                chosenCell = (i, j)\n",
    "\n",
    "    if chosenCell:\n",
    "        addAtom(chosenCell[0], chosenCell[1], players[1])\n",
    "\n",
    "# Main Loop\n",
    "def gameLoop():\n",
    "    initializeGrid()\n",
    "    loop = True\n",
    "\n",
    "    turns = 0\n",
    "    \n",
    "    currentPlayer = 0\n",
    "\n",
    "    vibrate = .5\n",
    "\n",
    "    while loop:\n",
    "        for event in pygame.event.get():\n",
    "            if event.type == pygame.QUIT:\n",
    "                close()\n",
    "            if event.type == pygame.KEYDOWN:\n",
    "                if event.key == pygame.K_q:\n",
    "                    close()\n",
    "            if currentPlayer == 0:  # Human player's turn\n",
    "                if event.type == pygame.MOUSEBUTTONDOWN:\n",
    "                    x, y = pygame.mouse.get_pos()\n",
    "                    i = x//blocks\n",
    "                    j = y//blocks\n",
    "                    if grid[i][j].color == players[currentPlayer] or grid[i][j].color == border:\n",
    "                        turns += 1\n",
    "                        addAtom(i, j, players[currentPlayer])\n",
    "                        currentPlayer += 1\n",
    "                        if currentPlayer >= noPlayers:\n",
    "                            currentPlayer = 0\n",
    "                    if turns >= noPlayers:\n",
    "                        isPlayerInGame()\n",
    "        \n",
    "        display.fill(background)\n",
    "        # Vibrate the Atoms in their Cells\n",
    "        vibrate *= -1\n",
    "        \n",
    "        drawGrid(currentPlayer)\n",
    "        showPresentGrid(vibrate)\n",
    "\n",
    "        # AI's turn\n",
    "        if currentPlayer == 1:\n",
    "            aiMove()\n",
    "            currentPlayer = 0  # Change back to human player's turn\n",
    "        \n",
    "        pygame.display.update()\n",
    "\n",
    "        # Check for winner outside the ^loop\n",
    "        res = checkWon()\n",
    "        if res < 9999:\n",
    "            gameOver(res)\n",
    "\n",
    "        clock.tick(20)\n",
    "\n",
    "gameLoop()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "14b752c3",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}