Tested game_database, began analytics

requirements.txt

@@ -1,2 +1,4 @@
 #To download requirements, use 'python -m pip install -r requirements.txt'
-#python-chess==1.10.0
+python-chess==1.10.0
+matplotlib==3.8.2
+requests==2.28.2

src/game_database.py

@@ -1,10 +1,44 @@
 import chess
+import matplotlib.pyplot as pyplot
+import pandas as pd
+from typing import Dict
+from logger import Logger
 
-class game_database:
+class GameDatabase:
     def __init__(self):
+        self.db = []
         pass
     
+    def add_game(self, game: Dict[str, str]):
+        self.db.append(game)
+
     def display_game(self, game_num: int): #Displays analytics for a specific game
         pass
     def display_tournament(self): #Displays analytics for the entire tournament
-        pass
+
+        df = pd.DataFrame(self.db)
+        print(df)
+
+        #heatmap of tournament winrates
+        
+        #bar chart of tournament winrates 
+
+        pass
+
+if __name__ == "__main__":
+    test_logger = Logger("ChessGPT", "ChatGPT")
+    test_logger.add_cheat("ChessGPT")
+    test_logger.add_legal_move("e4")
+    test_logger.add_checkmate("ChessGPT")
+    #test_logger.add_legal_move("e4 e5")
+    formatted = test_logger.return_formatted_game()
+
+    test_logger_2 = Logger("ChessGPT", "BERT")
+    test_logger_2.add_checkmate("BERT")
+    formatted_2 = test_logger_2.return_formatted_game()
+
+    db = GameDatabase()
+    db.add_game(formatted)
+    db.add_game(formatted_2)
+
+    db.display_tournament()

src/logger.py

@@ -1,25 +1,36 @@
 import requests
 import re
 
-class logger:
+class Logger:
     def __init__(self, model_1: str, model_2: str):
         self.model_1 = model_1
         self.model_2 = model_2
-
-    current_moves = "" #UCI notation
-    cheat_attempts = [0] #logs the number of cheat attempts for every move in order 
-    winner = ""
+        self.current_moves = ""
+        self.cheat_attempts = [0]
+        self.winner = ""
+        self.checkmate = False
 
     #Interface with the Model Interface
     def add_legal_move(self, current_moves: str): #current_moves should be all moves so far, in UCI notation
-        self.current_moves = current_moves
-        self.cheat_attempts.append(0)
-        
+        if self.checkmate:
+            raise RuntimeError("This game has already reached checkmate")
+        else: 
+            self.current_moves = current_moves
+            self.cheat_attempts.append(0)
+            
     def add_cheat(self, cheater_name: str):
-        self.cheat_attempts[-1] += 1
+        if self.checkmate:
+            raise RuntimeError("This game has already reached checkmate")
+        else: 
+            self.cheat_attempts[-1] += 1
 
     def add_checkmate(self, winner_name: str):
-        self.winner = winner_name
+        if winner_name != self.model_1 and winner_name != self.model_2:
+            raise RuntimeError("Not a valid winner for this logger")
+        else: 
+            self.winner = winner_name
+            self.checkmate = True
+
 
 
     #Internal Work
@@ -54,12 +65,16 @@ class logger:
             pass
 
         else:
-            game = {"UCI": self.current_moves, "Cheat Attempts": self.cheat_attempts}
+            game = {"Model 1": self.model_1, 
+                    "Model 2": self.model_2, 
+                    "Winner": self.winner, 
+                    "UCI": self.current_moves, 
+                    "Cheat Attempts": self.cheat_attempts}
             return game
     
 #Testing section
 if __name__ == "__main__":
-    new_logger = logger("ChessGPT", "ChatGPT")
+    new_logger = Logger("ChessGPT", "ChatGPT")
     current_board = "r2q1rk1/1pp2ppp/3bbn2/p2p4/8/1B1P4/PPP2PPP/RNB1QRK1 w - - 5 11" #make sure to include the additional notation at the end
     prev_board = "r2q1rk1/ppp2ppp/3bbn2/3p4/8/1B1P4/PPP2PPP/RNB1QRK1 w - - 5 11"
     print(str(new_logger.get_stockfish_results(prev_board, current_board)))