Update app.py

app.py

@@ -57,7 +57,19 @@ def draw_grid(data):
         # Plotting the building on the grid with color
         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')
-    
+
+    # Draw roads
+    for road in data.get('roads', []):  # Check for roads in the data, default to empty list if not found
+        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']))
+
+        
     # Setting labels and title
     ax.set_xlabel('Columns')
     ax.set_ylabel('Rows')