Update app.py

app.py

@@ -54,42 +54,28 @@ def draw_grid(data, highlight_coords=None):
     ax.grid(True)
 
     # Draw roads
+    """
     for road in data.get('roads', []):  # Default to empty list if no roads key
         start = road['start']
         end = road['end']
         color = color_codes.get("road", '#898989')  # Default color is gray for roads
         ax.plot([start[1], end[1]], [nrows-start[0], nrows-end[0]], color=color, linewidth=2)  # Adjust coordinates for matplotlib
+    """
 
-    # Draw buildings
-    for building in data['buildings']:
-        coords = building['coords']
-        b_type = building['type']
-        size = building['size']
-        color = color_codes.get(b_type, '#FFFFFF')  # Default color is white if not specified
-        
-        if highlight_coords and (coords[0], coords[1]) == tuple(highlight_coords):
-            highlighted_color = "#FFD700"  # Gold for highlighting
-            ax.add_patch(plt.Rectangle((coords[1], nrows-coords[0]-size), size, size, color=highlighted_color, edgecolor='black', linewidth=2))
-        else:
-            ax.add_patch(plt.Rectangle((coords[1], nrows-coords[0]-size), size, size, color=color, edgecolor='black', linewidth=1))
-            ax.text(coords[1]+0.5*size, nrows-coords[0]-0.5*size, b_type, ha='center', va='center', fontsize=8, color='black')
-
-    ax.set_xlabel('Columns')
-    ax.set_ylabel('Rows')
-    ax.set_title('Village Layout with Color Coding')
-    return fig
 
+    # Draw roads with a specified grey color
+    road_color = "#A0A0A0"  # Light grey; change to "#505050" for dark grey
+    for road in data.get('roads', []):  # Check for roads in the data
+        start, end = road['start'], road['end']
+        # Determine if the road is vertical or horizontal based on start and end coordinates
+        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))
 
-"""
-def draw_grid(data, highlight_coords=None):
-    fig, ax = plt.subplots(figsize=(12, 12))
-    nrows, ncols = data['size']['rows'], data['size']['columns']
-    ax.set_xlim(0, ncols)
-    ax.set_ylim(0, nrows)
-    ax.set_xticks(range(ncols+1))
-    ax.set_yticks(range(nrows+1))
-    ax.grid(True)
-    
+    # Draw buildings
     for building in data['buildings']:
         coords = building['coords']
         b_type = building['type']
@@ -108,7 +94,7 @@ def draw_grid(data, highlight_coords=None):
     ax.set_title('Village Layout with Color Coding')
     return fig
 
-    """
+
 
 # Main Streamlit app function
 def main():