Update app.py

app.py

@@ -37,7 +37,7 @@ color_codes = {
     # Add other types with their corresponding colors
 }
 
-# Function to draw the grid with optional highlighting
+# Modified draw_grid function with highlighting based on coordinates
 def draw_grid(data, highlight=None):
     fig, ax = plt.subplots(figsize=(12, 12))
     nrows, ncols = data['size']['rows'], data['size']['columns']
@@ -51,34 +51,23 @@ def draw_grid(data, highlight=None):
         coords = building['coords']
         b_type = building['type']
         size = building['size']
-        color = color_codes.get(b_type, '#FFFFFF')  # Use default color if type is not found
+        color = color_codes.get(b_type, '#FFFFFF')
         # Highlight the selected building
         if coords == highlight:
-            edgecolor = 'red'  # Highlight color
-            linewidth = 3  # Make the border thicker for the selected building
+            edgecolor = 'red'
+            linewidth = 3
         else:
             edgecolor = 'black'
             linewidth = 1
         ax.add_patch(plt.Rectangle((coords[1], nrows-coords[0]-size), size, size, color=color, edgecolor=edgecolor, linewidth=linewidth))
         ax.text(coords[1]+0.5*size, nrows-coords[0]-0.5*size, b_type, ha='center', va='center', fontsize=8, color='black')
-
-    # Draw roads
-    for road in data.get('roads', []):
-        start, end = road['start'], road['end']
-        road_color = road.get('color', '#000000')  # Default road color is black
-        # Determine if the road is vertical or horizontal and draw it
-        if start[0] == end[0]:  # Vertical road
-            for y in range(min(start[1], end[1]), max(start[1], end[1]) + 1):
-                ax.add_patch(plt.Rectangle((start[0], nrows-y-1), 1, 1, color=road_color))
-        else:  # Horizontal road
-            for x in range(min(start[0], end[0]), max(start[0], end[0]) + 1):
-                ax.add_patch(plt.Rectangle((x, nrows-start[1]-1), 1, 1, color=road_color))
+    
+    # Add the roads drawing logic here, if applicable
 
     ax.set_xlabel('Columns')
     ax.set_ylabel('Rows')
     ax.set_title('Village Layout with Color Coding')
     return fig
-
     
 # Main Streamlit app function
 def main():