Update engine.py

engine.py

@@ -52,6 +52,7 @@ class GameState():
         self.current_castling_rights = Castling_Rights(True,True,True,True)
         self.castle_rights_log=[Castling_Rights(self.current_castling_rights.wks,self.current_castling_rights.wqs,self.current_castling_rights.bks,self.current_castling_rights.bqs)]
         self.empassant_possible_log=[self.empassant_moves]
+        self.update_state_variables()
         # when current castling rights modified it creates new object and pt it in log 
         
         
@@ -150,18 +151,83 @@ class GameState():
             print("this is our starting move ")
     #if u move then it might be check to u so need to check these possiblities 
     #so we need to generate possible moves in next turn abd based on that we need to move
+    def update_state_variables(self):
+        # Ensure move log is empty for a fresh board
+        self.moveLog = []
+        
+        # Initialize game status flags
+        self.whiteToMove = True
+        self.checkmate = False
+        self.stalemate = False
+        self.in_check = False
+        self.pins = []
+        self.checks = []
+        self.protects = []
+        self.threatens = []
+        self.pieces_can_move_to = []
+        
+        # Update king locations by scanning the board
+        self.white_king_location = None
+        self.black_king_location = None
+        for row in range(8):
+            for col in range(8):
+                if self.board[row][col] == 'wK':
+                    self.white_king_location = (row, col)
+                if self.board[row][col] == 'bK':
+                    self.black_king_location = (row, col)
+        
+        # Set default king locations if not found (for standard board)
+        if not self.white_king_location:
+            self.white_king_location = (7, 4)
+        if not self.black_king_location:
+            self.black_king_location = (0, 4)
+        
+        # Reset en passant
+        self.empassant_moves = ()
+        self.empassant_possible_log = [()]
+        
+        # Set castling rights based on piece positions
+        self.current_castling_rights = Castling_Rights(False, False, False, False)
+        if self.white_king_location == (7, 4):
+            if self.board[7][0] == 'wR':
+                self.current_castling_rights.wqs = True
+            if self.board[7][7] == 'wR':
+                self.current_castling_rights.wks = True
+        if self.black_king_location == (0, 4):
+            if self.board[0][0] == 'bR':
+                self.current_castling_rights.bqs = True
+            if self.board[0][7] == 'bR':
+                self.current_castling_rights.bks = True
+        self.castle_rights_log = [Castling_Rights(
+            self.current_castling_rights.wks, self.current_castling_rights.wqs,
+            self.current_castling_rights.bks, self.current_castling_rights.bqs
+        )]
+        
+        # Calculate pins, checks, and in_check
+        self.incheck,self.pins,self.checks = self.check_for_pins_and_checks()
     
-    def is_valid_board(self,board):
-    # must be list of 8 rows
+    def is_valid_board(self, board):
         if len(board) != 8:
             return False
-        for row in board:
-            # each row must have 8 columns
-            if len(row) != 8:
-                return False
-            # check no element is empty
-            if any(cell in [None, "", "-"] for cell in row):
+        valid_pieces = {'bR', 'bN', 'bB', 'bQ', 'bK', 'bp', 'wR', 'wN', 'wB', 'wQ', 'wK', 'wp', '--'}
+        white_kings = 0
+        black_kings = 0
+        for row in range(8):
+            if len(board[row]) != 8:
                 return False
+            for col in range(8):
+                cell = board[row][col]
+                if cell not in valid_pieces:
+                    return False
+                if cell == 'wK':
+                    white_kings += 1
+                if cell == 'bK':
+                    black_kings += 1
+                # No pawns on promotion ranks
+                if row in (0, 7) and cell[1] == 'p':
+                    return False
+        if white_kings != 1 or black_kings != 1:
+            return False
         return True
  
     '''