chess_bot_ai.ipynb

{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.14","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":86524,"databundleVersionId":9818394,"sourceType":"competition"}],"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"# FIDE & Google Efficient Chess AI Challenge\n\nWelcome! This notebook will familiarize you with using competition's environment, creating an agent, and submitting your first chess bot!","metadata":{}},{"cell_type":"code","source":"# first let's make sure you have internet enabled\nimport requests\nrequests.get('http://www.google.com',timeout=10).ok","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-11-18T23:09:25.039291Z","iopub.execute_input":"2024-11-18T23:09:25.040781Z","iopub.status.idle":"2024-11-18T23:09:25.387017Z","shell.execute_reply.started":"2024-11-18T23:09:25.040725Z","shell.execute_reply":"2024-11-18T23:09:25.385516Z"}},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"#### If you don't have internet access (it doesn't say \"True\" above)\n1. make sure your account is Phone Verified in [account settings](https://www.kaggle.com/settings)\n2. make sure internet is turned on in Settings -> Turn on internet","metadata":{}},{"cell_type":"code","source":"%%capture\n# ensure we are on the latest version of kaggle-environments\n!pip install --upgrade kaggle-environments","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-11-18T23:09:47.729366Z","iopub.execute_input":"2024-11-18T23:09:47.729796Z","iopub.status.idle":"2024-11-18T23:10:04.079868Z","shell.execute_reply.started":"2024-11-18T23:09:47.72976Z","shell.execute_reply":"2024-11-18T23:10:04.078192Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"# Now let's set up the chess environment!\nfrom kaggle_environments import make\nenv = make(\"chess\", debug=True)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-11-18T23:10:04.082077Z","iopub.execute_input":"2024-11-18T23:10:04.082495Z","iopub.status.idle":"2024-11-18T23:10:13.358685Z","shell.execute_reply.started":"2024-11-18T23:10:04.082457Z","shell.execute_reply":"2024-11-18T23:10:13.357384Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"# this should run a game in the environment between two random bots\n# NOTE: each game starts from a randomly selected opening\nresult = env.run([\"random\", \"random\"])\nenv.render(mode=\"ipython\", width=1000, height=1000) ","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-11-18T23:10:13.359742Z","iopub.execute_input":"2024-11-18T23:10:13.360259Z","iopub.status.idle":"2024-11-18T23:10:19.515866Z","shell.execute_reply.started":"2024-11-18T23:10:13.360223Z","shell.execute_reply":"2024-11-18T23:10:19.514427Z"}},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"### Creating your first agent\nNow let's create your first agent! The environment has the [Chessnut](https://github.com/cgearhart/Chessnut) pip package installed and we'll use that to parse the board state and generate moves.","metadata":{}},{"cell_type":"code","source":"%%writefile main.py\nfrom Chessnut import Game\nimport random\n\ndef chess_bot(obs):\n    \"\"\"\n    Simple chess bot that prioritizes checkmates, then captures, queen promotions, then randomly moves.\n\n    Args:\n        obs: An object with a 'board' attribute representing the current board state as a FEN string.\n\n    Returns:\n        A string representing the chosen move in UCI notation (e.g., \"e2e4\")\n    \"\"\"\n    # 0. Parse the current board state and generate legal moves using Chessnut library\n    game = Game(obs.board)\n    moves = list(game.get_moves())\n\n    # 1. Check a subset of moves for checkmate\n    for move in moves[:10]:\n        g = Game(obs.board)\n        g.apply_move(move)\n        if g.status == Game.CHECKMATE:\n            return move\n\n    # 2. Check for captures\n    for move in moves:\n        if game.board.get_piece(Game.xy2i(move[2:4])) != ' ':\n            return move\n\n    # 3. Check for queen promotions\n    for move in moves:\n        if \"q\" in move.lower():\n            return move\n\n    # 4. Random move if no checkmates or captures\n    return random.choice(moves)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-11-18T23:10:19.517943Z","iopub.execute_input":"2024-11-18T23:10:19.518322Z","iopub.status.idle":"2024-11-18T23:10:19.525943Z","shell.execute_reply.started":"2024-11-18T23:10:19.518286Z","shell.execute_reply":"2024-11-18T23:10:19.524861Z"}},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"### Testing your agent\n\nNow let's see how your agent does againt the random agent!","metadata":{}},{"cell_type":"code","source":"result = env.run([\"main.py\", \"random\"])\nprint(\"Agent exit status/reward/time left: \")\n# look at the generated replay.json and print out the agent info\nfor agent in result[-1]:\n    print(\"\\t\", agent.status, \"/\", agent.reward, \"/\", agent.observation.remainingOverageTime)\nprint(\"\\n\")\n# render the game\nenv.render(mode=\"ipython\", width=1000, height=1000) ","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2024-11-18T23:10:40.177748Z","iopub.execute_input":"2024-11-18T23:10:40.178165Z","iopub.status.idle":"2024-11-18T23:10:42.527222Z","shell.execute_reply.started":"2024-11-18T23:10:40.178129Z","shell.execute_reply":"2024-11-18T23:10:42.525985Z"}},"outputs":[],"execution_count":null},{"cell_type":"markdown","source":"# To Submit:\n1. Download (or save) main.py\n2. Go to the [submissions page](https://www.kaggle.com/competitions/fide-google-efficiency-chess-ai-challenge/submissions) and click \"Submit Agent\"\n3. Upload main.py\n4. Press Submit!\n\nNow doubt you are already thinking of ways this bot could be improved! Go ahead and fork this notebook and get started! ♟️","metadata":{}},{"cell_type":"markdown","source":"# Submitting Multiple files \n### (or compressing your main.py)\n\nSet up your directory structure like this:\n```\nkaggle_submissions/\n  main.py\n  <other files as desired>\n```\n\nYou can run `tar -czf submission.tar.gz -C kaggle_submissions .` and upload `submission.tar.gz`","metadata":{}}]}