Update app.py

app.py

@@ -8,17 +8,17 @@ import numpy as np
 
 model = Llama(
     model_path=hf_hub_download(
-        repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-GRPO"),
+        repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.2b"),
         filename=os.environ.get("MODEL_FILE", "unsloth.Q8_0.gguf"),
     )
 )
 
-SYSTEM_PROMPT = """You are a Connect Four player[Connect Four is played on a 6 x 7 grid (with 6 rows and 7 columns]. Given the current board state, predict the next move. Respond in the following format:
+SYSTEM_PROMPT = """You are a Connect Four player.  Connect Four is played on a 6x7 grid. Given the current board state, predict the next *column* to play in.  Moves are specified by the column letter (a-g). Respond in the following format:
 <reasoning>
-Explain your reasoning for choosing the move, considering the current board state and potential future moves.
+Explain your reasoning, considering the board state and potential future moves.
 </reasoning>
 <move>
-The column and row of your move in the format 'a1', 'b3', 'g5', 'c6', etc. (column letter followed by row number).
+The column letter (a-g) of your move.
 </move>
 """
 
@@ -81,8 +81,8 @@ class ConnectFour:
                 if self.board[row][col] != 0:
                     col_letter = chr(ord('a') + col)
                     row_num = str(6 - row)  # Convert to 1-based indexing
-                    player = "X" if self.board[row][col] == 1 else "O"
-                    moves.append(f"{col_letter}{row_num}={player}")
+                    #player = "X" if self.board[row][col] == 1 else "O"
+                    moves.append(f"{col_letter}{row_num}")#={player}")
         return ", ".join(moves)
 
     def parse_ai_move(self, move_str):
@@ -146,7 +146,7 @@ def create_interface():
         padding: 0 15px;
     }
     #column-buttons button {
-      margin: 0px 4px;
+      margin: 0px 7px;
     }
     div.svelte-1nguped {
       display: block;
@@ -163,7 +163,7 @@ def create_interface():
                 status = gr.Markdown("Your turn! Click a button to drop your piece!", elem_id="ai-status")
                 
                 # Column buttons
-                with gr.Group(elem_id="column-buttons"):
+                with gr.Group(elem_id="column-buttons", elem_classes=["fitter"]):
                     col_buttons = []
                     for i in range(7):
                         btn = gr.Button(f"⬇️ {i+1}", scale=1)
@@ -211,7 +211,9 @@ def create_interface():
             # AI move
             game.current_player = 2
             board_state = game.board_to_string()
+
             prompt = f"Current Board: {board_state}. Make a move."
+            #print(prompt)
             
             # Get AI response
             response = model.create_chat_completion(
@@ -321,4 +323,4 @@ def render_board(board=None):
 # Launch the interface
 if __name__ == "__main__":
     interface = create_interface()
-    interface.launch(debug=True)
+    interface.launch()