Update app.py

app.py

@@ -18,10 +18,11 @@ X_LABELS = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']  # Labels for x-axis (a to h
 Y_LABELS = [8, 7, 6, 5, 4, 3, 2, 1]  # Reversed labels for y-axis (8 to 1)
 
 # Functions
-def get_grid_coordinate(pixel_x, pixel_y):
+def get_grid_coordinate(pixel_x, pixel_y, perspective):
     """
     Function to determine the grid coordinate of a pixel, considering a 10px border and
     the grid where bottom-left is (a, 1) and top-left is (h, 8).
+    The perspective argument can adjust for white ('w') or black ('b') viewpoint.
     """
     # Grid settings
     border = 10  # 10px border
@@ -46,17 +47,17 @@ def get_grid_coordinate(pixel_x, pixel_y):
     if x_index < 0 or x_index >= len(x_labels) or y_index < 0 or y_index >= len(y_labels):
         return "Pixel outside grid bounds"
 
-    # Convert indices to grid coordinates
-    x_index = adjusted_x // block_size  # Determine the column index (0-7)
-    y_index = adjusted_y // block_size  # Determine the row index (0-7)
+    # Adjust labels based on perspective
+    if perspective == "b":
+        x_labels = x_labels[::-1]  # Flip x-axis for black's perspective
+        y_labels = y_labels[::-1]  # Flip y-axis for black's perspective
 
-    # Convert row index to the correct label, with '8' at the bottom
-    y_labeld = y_labels[y_index]  # Correct index directly maps to '8' to '1'
     x_label = x_labels[x_index]
-    y_label = 8 - y_labeld + 1
+    y_label = y_labels[y_index]
 
     return f"{x_label}{y_label}"
 
+
 def predict_next_move(fen, stockfish):
     """
     Predict the next move using Stockfish.