Adding titles to graphs

app.py

@@ -71,7 +71,7 @@ class AInstance:
             #if this move is valid then return it
             proposed_board = game_state + selected_move
             if(verify_move(proposed_board)):
-                return proposed_board
+                return proposed_board, countr
             countr+=1
             #goes fifty times until the AInstance object flags itself as "ended" (fundamentally unable to make a valid move)
             if(countr > 50):
@@ -109,7 +109,7 @@ def print_game(string):
 def make_move(instance, game_state):
     print("\n" + instance.type + "s's move")
     return_state = game_state
-    return_state = instance.move(game_state)
+    return_state, countr = instance.move(game_state)
     game_ongoing = True
     if(instance.check_if_end()):
         st.write("This player claims countr > 50: " + instance.type)
@@ -117,7 +117,7 @@ def make_move(instance, game_state):
     if(check_mate(return_state)):
         st.write("This player claims mates: " + instance.type)
         game_ongoing = False
-    return(return_state, game_ongoing)
+    return(return_state, game_ongoing, countr)
 
 
 def main():
@@ -141,12 +141,17 @@ def main():
 
     moves = 0
     while game_ongoing:
-        game_state, game_ongoing = make_move(white, game_state)
+        game_state, game_ongoing, white_cheat_count = make_move(white, game_state)
+        for i in range(white_cheat_count):
+           onetime_logger.add_cheat()
         onetime_logger.add_legal_move(game_state)
+        
         if not game_ongoing:
             print_game(game_state)
             break
-        game_state, game_ongoing = make_move(black, game_state)
+        game_state, game_ongoing, black_cheat_count = make_move(black, game_state)
+        for i in range(black_cheat_count):
+           onetime_logger.add_cheat()
         onetime_logger.add_legal_move(game_state)
         if not game_ongoing:
             print_game(game_state)
@@ -154,7 +159,7 @@ def main():
         
         #testing code
         moves += 1
-        if moves >= 3:
+        if moves >= 2:
            break
     
     onetime_logger.add_checkmate(player_1) #here for testing purposes

game_database.py

@@ -19,17 +19,20 @@ class GameDatabase:
         game = self.db[game_num]
 
         fig, axs = plt.subplots(2,2)
+        fig.tight_layout()
         width = 0.4
 
         def list_sets_graph(data_list: [], axs_pos: [], title, width = 0.4):
             white_data = data_list[::2]
             black_data = data_list[1::2]
-            x = np.arange(len(white_data))
+            x_w = np.arange(len(white_data))
+            x_b = np.arange(len(black_data))
 
-            axs[axs_pos[0], axs_pos[1]].set_xticks(x, [x for x in range(len(white_data))])
-            axs[axs_pos[0], axs_pos[1]].bar(x - (width/2), white_data, width, color = "navajowhite")
-            axs[axs_pos[0], axs_pos[1]].bar(x + (width/2), black_data, width, color = "saddlebrown")
+            axs[axs_pos[0], axs_pos[1]].set_xticks(x_w, [x for x in range(len(white_data))])
+            axs[axs_pos[0], axs_pos[1]].bar(x_w - (width/2), white_data, width, color = "navajowhite")
+            axs[axs_pos[0], axs_pos[1]].bar(x_b + (width/2), black_data, width, color = "saddlebrown")
             axs[axs_pos[0], axs_pos[1]].legend(["White", "Black"]) 
+            axs[axs_pos[0], axs_pos[1]].set_title(title)
 
 
 
@@ -46,7 +49,6 @@ class GameDatabase:
 
         #plt.show()
         st.pyplot(fig)
-        pass
 
 
     def display_tournament(self): #Displays analytics for the entire tournament
@@ -72,7 +74,7 @@ class GameDatabase:
 
 if __name__ == "__main__":
     test_logger = Logger("ChessGPT", "ChatGPT")
-    test_logger.add_cheat("ChessGPT")
+    test_logger.add_cheat()
     time.sleep(1)
     test_logger.add_legal_move("e4")
     time.sleep(1)

logger.py

@@ -30,7 +30,7 @@ class Logger:
             self.seconds_per_move.append(current_time - self.prev_end_time)
             self.prev_end_time = current_time
             
-    def add_cheat(self, cheater_name: str):
+    def add_cheat(self):
         if self.checkmate:
             raise RuntimeError("This game has already reached checkmate")
         else: