eval.py

import pickle
from console_player import ConsolePlayer
from pypokerengine.api.emulator import Emulator
from bot import PGBot
from pypokerengine.utils.game_state_utils import attach_hole_card_from_deck, attach_hole_card, replace_community_card
from pypokerengine.utils.card_utils import gen_cards

SAVE_PATH = 'saved_models/second_test/policy_net_after'

STACK = 1000
N_PLAYER = 2
MAX_ROUND = 2
SMALL_BLIND_AMOUNT = 5
ANTE_AMOUNT = 1

if __name__ == '__main__':
    emul = Emulator()
    emul.set_game_rule(N_PLAYER, MAX_ROUND, SMALL_BLIND_AMOUNT, ANTE_AMOUNT)
    console = ConsolePlayer()
    bot = PGBot("bot", pickle.load(open(SAVE_PATH + "105980", "rb")))
    emul.register_player("target", bot)
    emul.register_player("baseline", console)
    players_info = {
        "target": {"name": "PGBot" ,"stack": STACK},
        "baseline": {"name": "console", "stack": STACK}
    }
    game_state = emul.generate_initial_game_state(players_info)
    while True:
        game_state, events = emul.start_new_round(game_state)
       
        if events[-1]["type"] == "event_game_finish":
            break
        for player in game_state["table"].seats.players:
            if player.uuid == "target":
                card1 = input("Enter first hole card > ")
                card2 = input("Enter second hold card > ")
                hole_card = gen_cards([card1, card2])
                bot.add_hole_card(hole_card)
                game_state = attach_hole_card(game_state, player.uuid, hole_card)
            else:
                game_state = attach_hole_card_from_deck(game_state, player.uuid)
        while game_state["street"] != Const.Street.FINISHED:
            print("action hitories: ", events[0]["round_state"]['action_histories'])
            print("community cards: ", events[0]["round_state"]["community_card"])
            cur_street = game_state["street"]
            print("valid actions: ", events[-1]['valid_actions'])
            if events[0]['round_state']['next_player'] == 1:
                action = input("Enter a valid action > ")
                amount = int(input("Enter a valid amount > "))
                game_state, events = emul.apply_action(game_state, action, amount)
            else:
                action, amount = bot.declare_action(events[-1]['valid_actions'], bot.hole_card_obj, events[0]['round_state'])
                game_state, events = emul.apply_action(game_state, action, amount)
            print("action hitories: ", events[0]["round_state"]['action_histories'])
            if game_state["street"] != cur_street:
                if events[0]["round_state"]["street"] == "flop":
                    flop1 = input("Enter first flop > ")
                    flop2 = input("Enter second flop > ")
                    flop3  = input("Enter third flop > ")
                    game_state = replace_community_card(game_state, gen_cards([flop1, flop2, flop3]))
                elif events[0]["round_state"]["street"] == "turn":
                    flop1 = input("Enter first flop > ")
                    flop2 = input("Enter second flop > ")
                    flop3  = input("Enter third flop > ")
                    turn = input("Enter turn card > ")
                    game_state = replace_community_card(game_state, gen_cards([flop1, flop2, flop3, turn]))
                elif events[0]["round_state"]["street"] == "river":
                    flop1 = input("Enter first flop > ")
                    flop2 = input("Enter second flop > ")
                    flop3  = input("Enter third flop > ")
                    turn = input("Enter turn card > ")
                    river = input("Enter river card > ")
                    game_state = replace_community_card(game_state, gen_cards([flop1, flop2, flop3, turn, river]))
        if events[0]["round_state"]["round_count"] >= MAX_ROUND:
            break

    print(events[0]["round_state"]["seats"])