Update elo.py

elo.py

@@ -1,5 +1,7 @@
 import pandas as pd
 
+import math
+
 def update_elo_ratings(ratings_dict, winner, loser):
     # Check if the ratings_dict is empty
     if not ratings_dict:
@@ -26,6 +28,22 @@ def update_elo_ratings(ratings_dict, winner, loser):
             return 20
         else:
             return 10
+    def elo(winner_rating, loser_rating, k_factor_winner=32, k_factor_loser=32):
+        # Calculate the expected scores for each player
+        winner_expected = 1 / (1 + 10 ** ((loser_rating - winner_rating) / 400))
+        loser_expected = 1 / (1 + 10 ** ((winner_rating - loser_rating) / 400))
+
+        # Calculate the new ratings for each player
+        winner_new_rating = winner_rating + k_factor_winner * (1 - winner_expected)
+        loser_new_rating = loser_rating + k_factor_loser * (0 - loser_expected)
+
+        return winner_new_rating, loser_new_rating
+
+    # ...
+
+    # Calculate new ratings
+    winner_new_rating, loser_new_rating = elo(winner_old_rating, loser_old_rating, k_factor_winner=winner_k_factor, k_factor_loser=loser_k_factor)
+
 
     # Update games played
     ratings_df.loc[ratings_df['bot_name'] == winner, 'games_played'] += 1