Upload 3 files

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-import numpy as np
-import plotly.graph_objects as go
 
 from src.utils import select_board
 

src/boards.py

@@ -39,5 +39,14 @@ class GenerateBoard():
         image = env.render()
         image = Image.fromarray(image)
 
+        # Convert all white pixels to gray
+        image = image.convert("RGBA")  # Ensure the image has an alpha channel
+        data = np.array(image)  # Convert image to numpy array
+        # Create a mask for white pixels
+        white_pixels = (data[:, :, :3] == [255, 255, 255]).all(axis=2)
+        data[white_pixels] = [200, 200, 200, 20]
+
+        image = Image.fromarray(data)  # Convert back to image
+
         return image, target_options, info
 

src/utils.py

@@ -5,7 +5,7 @@ def select_board(level: str = 'easy', size: int = 18, board_number: int = 0):
     initial_board_image, target_positions, info = GenerateBoard(level, size, board_number).setup_initial_board() 
 
     # Convert initial_board_image to a Plotly figure
-    fig = px.imshow(initial_board_image)  # Use Plotly's imshow
+    fig = px.imshow(initial_board_image)  # Use Plotly's imshow with gray scale
     fig.update_xaxes(showticklabels=False)  # Hide x-axis ticks
     fig.update_yaxes(showticklabels=False)  # Hide y-axis ticks