Merge branch 'logger'

requirements.txt

@@ -1,4 +1,6 @@
 #To download requirements, use 'python -m pip install -r requirements.txt'
 python-chess==1.1.0
 matplotlib==3.8.2
-requests==2.28.2
+requests==2.28.2
+pandas==1.5.3
+transformers==4.36.2

src/game_database.py

@@ -1,5 +1,5 @@
 import chess
-import matplotlib.pyplot as pyplot
+import matplotlib.pyplot as plt
 import pandas as pd
 from typing import Dict
 from logger import Logger
@@ -17,12 +17,14 @@ class GameDatabase:
     def display_tournament(self): #Displays analytics for the entire tournament
 
         df = pd.DataFrame(self.db)
-        print(df)
 
         #heatmap of tournament winrates
         
-        #bar chart of tournament winrates 
-
+        #bar chart of tournament winrates
+        win_results = df["Winner"].value_counts()
+        print(win_results.rank())
+        win_results.plot.bar()
+        plt.show()
         pass
 
 if __name__ == "__main__":
@@ -37,8 +39,13 @@ if __name__ == "__main__":
     test_logger_2.add_checkmate("BERT")
     formatted_2 = test_logger_2.return_formatted_game()
 
+    test_logger_3 = Logger("ChessGPT", "BERT")
+    test_logger_3.add_checkmate("ChessGPT")
+    formatted_3 = test_logger_3.return_formatted_game()
+
     db = GameDatabase()
     db.add_game(formatted)
     db.add_game(formatted_2)
+    db.add_game(formatted_3)
 
     db.display_tournament()

src/logger.py

@@ -2,7 +2,7 @@ import requests
 import re
 
 class Logger:
-    def __init__(self, model_1: str, model_2: str):
+    def __init__(self, model_1: str, model_2: str): #Model 1 should be white
         self.model_1 = model_1
         self.model_2 = model_2
         self.current_moves = ""