best minimax

import chess

def simplify_fen_string(fen):
    """Odstraní z FEN řetězce části týkající se počtu tahů, zachovává pozici, hráče na tahu, rošády a en passant."""
    parts = fen.split(' ')
    simplified_fen = ' '.join(parts[:4])  # Zachováváme informace o pozici, hráči na tahu, rošádách a en passant
    return simplified_fen

def minimax(board, depth, alpha, beta, maximizing_player):
    if depth == 0 or board.is_game_over():
        return None, evaluate_board(board)

    best_move = None

    if maximizing_player:
        max_eval = float('-inf')
        for move in board.legal_moves:
            board_copy = board.copy()
            board_copy.push(move)
            _, eval = minimax(board_copy, depth - 1, alpha, beta, False)
            if eval > max_eval:
                max_eval = eval
                best_move = move
            alpha = max(alpha, eval)
            if beta <= alpha:
                break
        return best_move, max_eval
    else:
        min_eval = float('inf')
        for move in board.legal_moves:
            board_copy = board.copy()
            board_copy.push(move)
            _, eval = minimax(board_copy, depth - 1, alpha, beta, True)
            if eval < min_eval:
                min_eval = eval
                best_move = move
            beta = min(beta, eval)
            if beta <= alpha:
                break
        return best_move, min_eval

def evaluate_board(board):
    if board.is_checkmate():
        # Pokud je král matován a na tahu je bílý, vrátí -1000, pokud je na tahu černý, vrátí 1000
        return -1000 if board.turn == chess.WHITE else 1000
    elif board.is_stalemate() or board.is_insufficient_material():
        # Remíza nebo nedostatek materiálu pro mat
        return 0
    # Základní neutrální hodnota pro nekoncové pozice
    return 0

# Nastavení startovní pozice
start_fen = "6k1/3Q4/8/8/4K3/8/8/8 w - - 0 1"
board = chess.Board(start_fen)

# Spuštění minimaxu s maximální hloubkou 3
best_move, best_score = minimax(board, 6, float('-inf'), float('inf'), True)
print(f"Nejlepší tah z pozice {start_fen} je {board.san(best_move)} s hodnocením {best_score}.")