update

src/ai_battle.py

@@ -8,6 +8,7 @@ from .chess_ML.model import ChessModel
 import torch
 import pickle
 import numpy as np
+import os
 class AIAgent:
     """Base class cho các AI agent"""
     
@@ -80,13 +81,14 @@ class MLAgent(AIAgent):
     
     def __init__(self, model=None, move_to_idx: Dict[str, int] = None, idx_to_move: Dict[int, str] = None):
         super().__init__("ML AI", "Machine Learning Agent")
-        with open("src/chess_ML/models/heavy_move_to_int_1", "rb") as file:
+        BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+        with open(os.path.join(BASE_DIR, "src/chess_ML/models/heavy_move_to_int_1"), "rb") as file:
             self.move_to_int = pickle.load(file) #load mapping
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # sử dụng GPU nếu có
 
         # Load the model
         self.model = ChessModel(num_classes=len(self.move_to_int))
-        self.model.load_state_dict(torch.load("src/chess_ML/models/TORCH_1_100EPOCHS.pth"), map_location=self.device)
+        self.model.load_state_dict(torch.load(os.path.join(BASE_DIR, 'src/chess_ML/models/TORCH_1_100EPOCHS.pth'), map_location=self.device))
         self.model.to(self.device)
         self.model.eval()  # Set the model to evaluation mode (it may be reductant)
 

src/streamlit_app.py

@@ -234,7 +234,7 @@ with col_ctrl:
             piece = board.piece_at(chess.parse_square(from_sq))
             if piece and piece.piece_type == chess.PAWN:
                 if (board.turn == chess.WHITE and to_sq[1] == "8") or \
-                   (board.turn == chess.BLACK and to_sq[1] == "1"):
+                (board.turn == chess.BLACK and to_sq[1] == "1"):
                     uci += "q"
             try:
                 move = chess.Move.from_uci(uci)