Update app.py

app.py

@@ -8,7 +8,73 @@ def load_data(filename):
         data = json.load(file)
     return data
 
-# Your existing color codes and draw_grid function go here
+# Color codes updated with hexadecimal values for compatibility
+color_codes = {
+    "residential": "#ADD8E6",  # Light Blue
+    "commercial": "#90EE90",  # Light Green
+    "community_facilities": "#FFFF00",  # Yellow
+    "school": "#FFFF00",  # Yellow
+    "healthcare_facility": "#FFFF00",  # Yellow
+    "green_space": "#90EE90",  # Light Green
+    "utility_infrastructure": "#90EE90",  # Light Green
+    "emergency_services": "#FF0000",  # Red
+    "cultural_facilities": "#D8BFD8",  # Light Purple
+    "recreational_facilities": "#D8BFD8",  # Light Purple
+    "innovation_center": "#90EE90",  # Light Green
+    "elderly_care_home": "#FFFF00",  # Yellow
+    "childcare_centers": "#FFFF00",  # Yellow
+    "places_of_worship": "#D8BFD8",  # Light Purple
+    "event_spaces": "#D8BFD8",  # Light Purple
+    "guest_housing": "#FFA500",  # Orange
+    "pet_care_facilities": "#FFA500",  # Orange
+    "public_sanitation_facilities": "#A0A0A0",  # Grey
+    "environmental_monitoring_stations": "#90EE90",  # Light Green
+    "disaster_preparedness_center": "#A0A0A0",  # Grey
+    "outdoor_community_spaces": "#90EE90"  # Light Green
+}
+
+# Function to draw the grid layout with color coding
+def draw_grid(data):
+    # Create a figure and a grid of subplots
+    fig, ax = plt.subplots(figsize=(12, 12))
+    
+    # Setting the grid size
+    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)
+    
+    # Plotting each building with its assigned color from the color_codes dictionary
+    for building in data['buildings']:
+        # Extracting the building details
+        coords = building['coords']
+        b_type = building['type']
+        size = building['size']
+        color = color_codes.get(b_type, '#FFFFFF')  # Default color is white if not specified
+        
+        # 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')
+    ax.set_title('Village Layout with Color Coding')
+    
+    return fig
 
 # Streamlit application starts here
 def main():
@@ -19,25 +85,21 @@ def main():
     
     with col1:
         st.header("Today's Agenda")
-        # You can add content to Today's Agenda here
+        # Example content for Today's Agenda
         st.write("1. Morning Meeting\n2. Review Project Plans\n3. Lunch Break\n4. Site Visit\n5. Evening Wrap-up")
     
     with col2:
         st.header("Green Smart Village Layout")
         # Load and display the data with color coding
-        data = load_data('grid.json')  # Make sure this path is correct
-        # Drawing and displaying the grid layout with color coding
+        data = load_data('grid.json')  # Ensure this path is correct
         fig = draw_grid(data)
         st.pyplot(fig)
     
     with col3:
         st.header("Check Your HIN Number")
-        # You can add an input form or information related to HIN numbers here
         hin_number = st.text_input("Enter your HIN number:")
         if hin_number:
-            # Perform HIN number validation or lookup here
-            st.write("HIN number details...")
-            # You might want to replace the placeholder text with actual code to check HIN numbers
+            st.write("HIN number details...")  # Placeholder for actual HIN number check
 
 if __name__ == "__main__":
     main()