From baac9b3e74a621f620f0a1a9bc5507b4e34b7da9 Mon Sep 17 00:00:00 2001
From: 0880 <98263509+0880880@users.noreply.github.com>
Date: Mon, 19 Jan 2026 18:30:09 +0330
Subject: [PATCH] Fix king protection

---
 app.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index 3c20cbd..45dbaac 100644
--- 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