Update app.py

app.py

@@ -4,7 +4,6 @@ from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
 import numpy as np
 from typing import List
-import re
 
 model = Llama(
     model_path=hf_hub_download(
@@ -42,6 +41,7 @@ def extract_xml_move(text: str) -> str:
     Extracts the move (a single column letter a–g) from the XML format
     using an improved regex. This function is kept simple for reuse.
     """
+    import re
     match = re.search(r'<move>\s*([a-g])\s*</move>', text)
     if match:
         return match.group(1)
@@ -102,10 +102,18 @@ def parse_coordinate_list(board_str: str) -> List[List[str]]:
             grid[row][col] = piece
     return grid
 
-def get_available_positions(board_str: str) -> str:
+def get_available_positions(board_moves: List[str]) -> str:
     """Returns the next available position (lowest empty spot) for each column."""
-    # Parse the board state
-    grid = parse_coordinate_list(board_str)
+    # Initialize empty grid (. means empty)
+    grid = [['.' for _ in range(7)] for _ in range(6)]
+    
+    # Fill in taken positions from the moves
+    for move in board_moves:
+        if len(move) >= 2:
+            col = ord(move[0]) - ord('a')
+            row = int(move[1]) - 1
+            if 0 <= row < 6 and 0 <= col < 7:
+                grid[row][col] = 'X' if len(board_moves) % 2 == 1 else 'O'
     
     # Find next available position in each column
     available = []
@@ -135,12 +143,12 @@ class ConnectFour:
             return False, -1
             
         # Find the lowest empty row in the selected column
-        for row in range(5, -1, -1):
-            if self.board[5-row][col] == 0:
-                self.board[5-row][col] = self.current_player
+        for row in range(6):
+            if self.board[row][col] == 0:
+                self.board[row][col] = self.current_player
                 # Store the move
                 col_letter = chr(ord('a') + col)
-                row_num = row + 1
+                row_num = row + 1  # Converting to 1-based indexing for the coordinate system
                 move = f"{col_letter}{row_num}"
                 
                 if self.current_player == 1:
@@ -148,7 +156,7 @@ class ConnectFour:
                 else:
                     self.ai_moves.append(move)
                     
-                return True, 5-row
+                return True, row
         return False, -1
 
     def check_winner(self):
@@ -192,7 +200,7 @@ class ConnectFour:
             for col in range(7):
                 if self.board[row][col] != 0:
                     col_letter = chr(ord('a') + col)
-                    row_num = str(6 - row)  # Convert to 1-based indexing
+                    row_num = str(row + 1)  # Convert to 1-based indexing
                     piece = "X" if self.board[row][col] == 1 else "O"
                     moves.append(f"{col_letter}{row_num}({piece})")
         return ", ".join(moves) if moves else "Empty Board"
@@ -452,7 +460,8 @@ def render_board(board=None):
         
     html = '<div class="connect4-board">'
     
-    for row in range(6):
+    # Render from top to bottom to display the board correctly
+    for row in range(5, -1, -1):
         for col in range(7):
             cell_class = "connect4-cell"
             content = "⚪"