Update app.py

app.py

@@ -7,7 +7,7 @@ from typing import List
 
 model = Llama(
     model_path=hf_hub_download(
-        repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.8b"), #"Lyte/QuadConnect2.5-0.5B-v0.0.6b"), #"Lyte/QuadConnect-Llama-1B-v0.0.7b"),#"
+        repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.9b"),#"Lyte/QuadConnect2.5-0.5B-v0.0.8b"), #"Lyte/QuadConnect2.5-0.5B-v0.0.6b"), #"Lyte/QuadConnect-Llama-1B-v0.0.7b"),#"
         filename=os.environ.get("MODEL_FILE", "unsloth.Q8_0.gguf"), #"quadconnect.Q8_0.gguf"),
     ),
     n_ctx=16384
@@ -102,20 +102,34 @@ def parse_coordinate_list(board_str: str) -> List[List[str]]:
             grid[row][col] = piece
     return grid
 
-def get_available_positions(board: np.ndarray) -> str:
-    """
-    Returns the next available position (the lowest empty spot) for each column
-    directly from the board (a NumPy array). If a piece has been played at a1,
-    then the available spot in column a will be a2.
-    """
+def get_available_positions(board_moves: List[str]) -> str:
+    """Returns all available positions per column after simulating gravity."""
+    # Initialize empty grid ('.' means empty)
+    grid = [['.' for _ in range(7)] for _ in range(6)]
+    
+    # Place each move into the lowest available slot in its column
+    for i, move in enumerate(board_moves):
+        if not move:
+            continue
+        col = ord(move[0]) - ord('a')
+        for row in range(6):
+            if grid[row][col] == '.':
+                grid[row][col] = 'X' if i % 2 == 0 else 'O'
+                break
+
+    # For each column, list all empty positions (which will be above the placed pieces)
     available = []
     for col in range(7):
         col_letter = chr(ord('a') + col)
-        # Search from bottom (row 0) to top (row 5)
+        positions = []
         for row in range(6):
-            if board[row][col] == 0:  # Found empty spot
-                available.append(f"{col_letter}: {col_letter}{row + 1}")
-                break
+            if grid[row][col] == '.':
+                positions.append(f"{col_letter}{row + 1}")
+        if positions:
+            available.append(f"Column {col_letter}: {', '.join(positions)}")
+        else:
+            available.append(f"Column {col_letter}: Full")
+    
     return "\n  ".join(available)
 
 class ConnectFour: