engine.py

import sys
import chess
import keras
from keras.callbacks import ModelCheckpoint
import os
from ocean_eater_network import create_model
from decision_tree import make_probabilistic_decision

# There's a lot of code that is repeated and could be better
# implemented as functions, but I don't have a firm grasp of
# scope so I left it like this.


# log file:
f = open('uci_log.txt', 'a')
f.write('\n\n\nStarting new log!\n\n')
f.flush()


# For logging purposes
def print_log(str):
    f.write(str + '\n')
    f.flush()
    print(str)


def log(str):
    f.write('COMMAND: ' + str + '\n')
    f.flush()


ENGINE_NAME = 'Ocean Eater 0.1'
AUTHOR_NAME = 'Ryan Peterson and Joshua Marsh'

# TODO: currently this doesn't actually change search cap
DEFAULT_STATE_CAP = 1000
MIN_STATE_CAP = 50
MAX_STATE_CAP = 10000000


MODEL_CHECKPOINTER = ModelCheckpoint("GameHistories/model.h5", monitor='loss')

if os.path.exists(MODEL_CHECKPOINTER.filepath):
    print("Using checkpoint model.")
    model = keras.models.load_model(MODEL_CHECKPOINTER.filepath)
else:
    model = create_model()


# initialize to empty string
# wait for uci command to be provided
while input().strip() != 'uci':
    pass  # do nothing

# Now, identify
print_log('id name ' + ENGINE_NAME)
print_log('id author ' + AUTHOR_NAME)

# Specify options
# State Cap is number of states
print_log('option name state_cap type spin default ' + str(DEFAULT_STATE_CAP) +
          ' min ' + str(MIN_STATE_CAP) +
          ' max ' + str(MAX_STATE_CAP))

# Done specifying options
print_log('uciok')

# get next command
command = input().strip()
log(command)
state_cap = DEFAULT_STATE_CAP
ready = False
while not ready:
    if command.startswith('quit'):
        sys.exit()

    elif command.startswith('setoption'):
        # check what the 3rd word is
        if command.split()[2].strip() == 'state_cap':
            # 4th word is the value
            state_cap = int(command.split()[3].strip())

    elif command.startswith('isready'):
        ready = True
        print_log('readyok')

    command = input().strip()
    log(command)

have_board = False

board = chess.Board()
while not have_board:
    if command.startswith('quit'):
        sys.exit()

    elif command.startswith('setoption'):
        # check what the 3rd word is
        if command.split()[2].strip() == 'state_cap':
            # 4th word is the value
            state_cap = int(command.split()[4].strip())

    elif command.startswith('ucinewgame'):
        pass

    elif command.startswith('position'):
        # if second word is startpos
        if command.split()[1].strip() == 'startpos':
            have_board = True
            board.reset()
            # if it has a list of moves
            if 'moves' in command:
                moves = command.split(' moves')[1]
                # apply moves to board
                for move in moves.split():
                    board.push(chess.Move.from_uci(move))

        # set from fen, command will be 'position fen <FEN> moves <MOVES>
        elif command.split()[1].strip() == 'fen':
            # get everything after 'fen' and before 'moves'
            fen = command[13:].split(' moves')[0]
            have_board = True
            board.set_fen(fen)  # set starting position
            # if it has a list of moves
            if 'moves' in command:
                moves = command.split(' moves')[1]
                # apply moves to board
                for move in moves.split():
                    board.push(chess.Move.from_uci(move))

    command = input().strip()
    log(command)

while True:
    if command.startswith('quit'):
        sys.exit()

    elif command.startswith('setoption'):
        # check what the 3rd word is
        if command.split()[2].strip() == 'state_cap':
            # 4th word is the value
            state_cap = int(command.split()[4].strip())

    elif command.startswith('ucinewgame'):
        pass

    elif command.startswith('isready'):
        print_log('readyok')

    elif command.startswith('position'):
        # if second word is startpos
        if command.split()[1].strip() == 'startpos':
            have_board = True
            board.reset()
            # if it has a list of moves
            if 'moves' in command:
                moves = command.split(' moves')[1]
                # apply moves to board
                for move in moves.split():
                    board.push(chess.Move.from_uci(move))

        # set from fen, command will be 'position fen <FEN> moves <MOVES>
        elif command.split()[1].strip() == 'fen':
            # get everything after 'fen' and before 'moves'
            fen = command[13:].split(' moves')[0]
            have_board = True
            board.set_fen(fen)  # set starting position
            # if it has a list of moves
            if 'moves' in command:
                moves = command.split(' moves')[1]
                # apply moves to board
                for move in moves.split():
                    board.push(chess.Move.from_uci(move))

    elif command.startswith('go'):
        move = chess.Move.uci(make_probabilistic_decision(model, board))
        print_log('bestmove ' + move)

    command = input().strip()
    log(command)