Update chess-gpt-eval/main.py

-Instead of the play_games max_games parameter controlling the number of games played, it now controls the number of games *saved* -- an important distinction because...
-We now track the transcript of all played games (reset for each new player_ones) and record/save only unique games. So, play_games will continue playing new games until the desired number of games have been saved.
-Additionally, when random_opening is True, we check for game over after each added move, (stop adding moves and) discard the game (/ start over adding moves) if the random moves cause the game to end.

chess-gpt-eval/main.py

@@ -348,7 +348,7 @@ def add_random_moves(
     for i in range(num_moves * 2):  # Full moves to half moves
         legal_moves = list(board.legal_moves)
         if not legal_moves:
-            break
+            return None, None, 0  # Game over, discard the game
 
         move = board.san(random.choice(legal_moves))
         board.push(board.parse_san(move))
@@ -358,6 +358,9 @@ def add_random_moves(
         else:
             game_state += f" {move}"
 
+        if board.is_game_over():
+            return None, None, 0  # Game over, discard the game
+
     game_state = game_state.strip()
     return game_state, board, num_moves
 
@@ -452,8 +455,10 @@ def play_games(
     random_opening: bool = False,
     random_opening_moves: int = 20,
 ):
-    for z in range(max_games):
-        print(f"\nGame {z} of {max_games}\n")
+    unique_games = set()
+    games_saved = 0
+    while games_saved < max_games:
+        print(f"\nGame {games_saved} of {max_games}\n")
     
         with open("gpt_inputs/prompt.txt", "r") as f:
             game_state = f.read()
@@ -462,7 +467,10 @@ def play_games(
         if book_opening:
             game_state, board, opening_moves = random_book_opening(game_state, board)
         elif random_opening:
-            game_state, board, opening_moves = add_random_moves(game_state, board, random_opening_moves)
+            while True:
+                game_state, board, opening_moves = add_random_moves(game_state, board, random_opening_moves)
+                if game_state is not None:
+                    break
         else:
             opening_moves = 0
         player_one_illegal_moves = 0
@@ -545,26 +553,30 @@ def play_games(
         print(f"Result: {board.result()}")
         print(board)
         print()
-        record_results(
-            board,
-            player_one,
-            player_two,
-            game_state,
-            player_one_illegal_moves,
-            player_one_illegal_attempts,
-            player_two_illegal_moves,
-            player_one_legal_moves,
-            player_two_legal_moves,
-            total_time,
-            player_one_resignation,
-            player_two_resignation,
-            player_one_failed_to_find_legal_move,
-            player_two_failed_to_find_legal_move,
-            total_moves,
-            illegal_moves,
-            opening_moves,
-            illegal_move_numbers
-        )
+        if game_transcript not in unique_games:
+            unique_games.add(game_transcript)
+            record_results(
+                board,
+                player_one,
+                player_two,
+                game_state,
+                player_one_illegal_moves,
+                player_one_illegal_attempts,
+                player_two_illegal_moves,
+                player_one_legal_moves,
+                player_two_legal_moves,
+                total_time,
+                player_one_resignation,
+                player_two_resignation,
+                player_one_failed_to_find_legal_move,
+                player_two_failed_to_find_legal_move,
+                total_moves,
+                illegal_moves,
+                opening_moves,
+                illegal_move_numbers
+            )
+        else:
+            print("Duplicate game; not saved.")
     if isinstance(player_one, StockfishPlayer):
         player_one.close()
     if isinstance(player_two, StockfishPlayer):