Update app.py

app.py

@@ -43,6 +43,44 @@ color_codes = {
 }
 
 # Function to draw the grid with optional highlighting
+
+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 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
+
+
+"""
 def draw_grid(data, highlight_coords=None):
     fig, ax = plt.subplots(figsize=(12, 12))
     nrows, ncols = data['size']['rows'], data['size']['columns']
@@ -70,6 +108,8 @@ def draw_grid(data, highlight_coords=None):
     ax.set_title('Village Layout with Color Coding')
     return fig
 
+    """
+
 # Main Streamlit app function
 def main():
     st.title('Green Smart Village Application')