Update app.py

app.py

@@ -1,3 +1,4 @@
+User
 import streamlit as st
 import json
 import matplotlib.pyplot as plt
@@ -41,14 +42,9 @@ 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)  # Keep the original limits
+    ax.set_ylim(0, nrows)
     ax.set_xticks(range(ncols+1))
     ax.set_yticks(range(nrows+1))
-    
-    # Relabel y-axis with reversed order
-    y_labels = list(range(nrows, -1, -1))  # Generates labels from nrows to 0
-    ax.set_yticklabels(y_labels)
-    
     ax.grid(True)
     
     for building in data['buildings']:
@@ -57,19 +53,29 @@ def draw_grid(data, highlight_coords=None):
         size = building['size']
         color = color_codes.get(b_type, '#FFFFFF')  # Default color is white if not specified
         
-        # Plotting logic remains unchanged
-        ax.add_patch(plt.Rectangle((coords[1], coords[0]), size, size, color=color, edgecolor='black', linewidth=1))
-        ax.text(coords[1]+0.5*size, coords[0]+0.5*size, b_type, ha='center', va='center', fontsize=8, color='black')
+        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')
+
+    # Drawing roads
+    for road in data.get('roads', []):  # Ensures compatibility if 'roads' key is missing
+        start, end = road['start'], road['end']
+        road_color = road.get('color', '#696969')  # Dark gray as default road color
+        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))
 
-    # No change in drawing roads, assuming your original logic fits here
-    
     ax.set_xlabel('Columns')
     ax.set_ylabel('Rows')
     ax.set_title('Village Layout with Color Coding')
     return fig
 
-
-
 # Main Streamlit app function
 def main():
     st.title('Green Smart Village Application')
@@ -108,4 +114,4 @@ def main():
             st.write("HIN number details will be displayed here...")  # Placeholder for actual HIN number check
 
 if __name__ == "__main__":
-    main()
+    main()