Fix king protection

2026-01-19 18:30:09 +03:30
parent 4daf04d340
commit baac9b3e74
1 changed files with 9 additions and 1 deletions
--- a/app.py
+++ b/app.py
@@ -393,8 +393,16 @@ def generate_valid_moves(
                    )
                    if king not in enemy_moves:
                        continue
+                    ni = i
+                    nj = j
+                    if j == sx and i == sy:
+                        ni = m.y
+                        nj = m.x
                    new_enemy_moves = get_piece_moves(
-                        p.value.lower(), fake_board, not is_white, xy_to_pos_safe(j, i)
+                        p.value.lower(),
+                        fake_board,
+                        not is_white,
+                        xy_to_pos_safe(nj, ni),
                    )
                    if king in new_enemy_moves:
                        king_safe = False