Python-Rock-Paper-Scissors-AI/main.py at main · mo-tem/Python-Rock-Paper-Scissors-AI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
from modules.game import Game # /modules/game.py
from modules.actors import Player, AiPlayer # /modules/actors.py
from modules.draw import DrawData # /modules/draw.py


# main function
def main():
    game = Game()  # initialize the Game object
    human = Player()  # initialize the Player object
    ai = AiPlayer()  # initialize the AiPlayer object

    (player_data, ai_data, game_data, loaded) = game.load_game() # ask to load the previous game (load_game() returns a tuple of 4 elements)

    if loaded == True: # only load data if game.load_game() ran succesfully
        human.name = player_data['name'] # get the human name from the player data dict
        human.wins = player_data['wins'] # get the human wins from the player data dict
        human.win_rate = player_data['win_rate'] # get the win rate from the player data dict
        ai.name = ai_data['name'] # get the ai name from the ai data dict
        ai.wins = ai_data['wins'] # get the ai wins from the ai data dict
        ai.win_rate = ai_data['win_rate'] # get the ai win rate from the ai data dict
        game.sessions = game_data['sessions'] # get the game sessions data from game data dict
        print('Loaded succesfully')

    while (game.get_state() == True):  # keep looping if the game state is True

        human.set_choice()  # set human choice

        if human.choice not in game.options:  # check the input and break the loop if the input is not what we want
            print('Error 1: Incorrect input')
            continue

        human.history.append(human.choice) # add player choices to a list

        if human.choice == 'exit':
            game.close()
            continue

        game.sessions += 1  # track game sessions

        if game.sessions == 1: # if it's the first round
            ai.set_choice()  # set the ai choice (randomly)

        else:
            if game.sessions % 5 == 0: # every 5 rounds
                if human.get_last_choice() == 'rock': # set the ai choice to the last player choice
                    ai.choice = 'paper'
                elif human.get_last_choice() == 'paper':
                    ai.choice = 'scissors'
                elif human.get_last_choice() == 'scissors':
                    ai.choice = 'rock'

                human.history = [] # wipe the player player history to get fresh data for next 5 rounds

            else:
                if human.frequent_choice() == 'rock': # set the ai choice to the most frequent player choice
                    ai.choice = 'paper'
                elif human.frequent_choice() == 'paper':
                    ai.choice = 'scissors'
                elif human.frequent_choice() == 'scissors':
                    ai.choice = 'rock'

        print('------------------------')

        print(f'{human.name} selected: {human.choice}') # display the player choice
        print(f'{ai.name} selected: {ai.choice}') # display the ai choice

        # compare the choices and display the winner
        if ai.choice == human.choice:
            print('[Draw]')
            ai.ties += 1 # increment tie by 1

        elif ai.choice == 'rock' and human.choice == 'scissors':
            print(f'[{ai.name} won]')
            ai.wins += 1 # increment ai win by 1

        elif ai.choice == 'paper' and human.choice == 'rock':
            print(f'[{ai.name} won]')
            ai.wins += 1 # increment ai win by 1

        elif ai.choice == 'scissors' and human.choice == 'paper':
            print(f'[{ai.name} won]')
            ai.wins += 1 # increment ai win by 1

        else:
            print(f'[{human.name} won]')
            human.wins += 1 # increment player win by 1

        print('------------------------')

        ai.win_rate.append(ai.wins / game.sessions) # track the ai win rate
        human.win_rate.append(human.wins / game.sessions) # track the player win rate

    # player, ai and game data is saved in dictionaries for the purpose of saving
    player_data = {"name": human.name, "wins": human.wins, "win_rate": human.win_rate}
    ai_data = {"name": ai.name, "wins": ai.wins, "win_rate": ai.win_rate}
    game_data = {"sessions": game.sessions}

    print(f'SCORES ---- Humans: {human.wins} AI: {ai.wins}') # display the individual wins
    print(f'AI Win Rate: {ai.get_win_rate()}%') # display the ai win rate
    print(f'HUMAN Win Rate: {human.get_win_rate()}%') # display the player win rate

    save_status = game.save_game(player_data, ai_data, game_data) # asks to save the game
    if save_status == True: # print a message if saved succesfully
        print('Saved game succesfully')

    #Post game figure display feature
    draw = DrawData()
    draw.bar_graph(human.wins,ai.ties,ai.wins) #Bar graph for wins ties and losses(In player perspective)
    draw.winrate_line_graph(human.win_rate,ai.win_rate,game.sessions)#Line graph for player winrate and AI winrate

# entry point
if __name__ == '__main__':
    main()