Update app.py

app.py

@@ -37,7 +37,7 @@ color_codes = {
 }
 
 # Function to draw the grid with optional highlighting and roads
-def draw_grid(data, highlight_element=None):
+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)
@@ -92,30 +92,18 @@ def main():
         st.header("My Incentive")
         st.write("Total incentive for HIN optimization")
     
-    
     with col2:
         st.header("Green Smart Village Layout")
-        data = load_data('grid.json')
-        
-        # Create a combined list of buildings and roads for the dropdown
-        building_options = [f"Building: {bld['type']} at ({bld['coords'][0]}, {bld['coords'][1]})" for bld in data['buildings']]
-        road_options = [f"Road: {road['id']}" for road in data['roads']]
-        all_options = building_options + road_options
+        data = load_data('grid.json')  # Ensure this path is correct
         
-        selected_option = st.selectbox("Select a building or road to highlight:", options=all_options)
-        
-        # Determine if a building or road was selected
-        if "Building" in selected_option:
-            selected_index = building_options.index(selected_option)
-            selected_building_coords = data['buildings'][selected_index]['coords']
-            highlight_element = {'type': 'building', 'coords': selected_building_coords}
-        else:
-            selected_index = road_options.index(selected_option) 
-            selected_road_coords = data['roads'][selected_index]['start'], data['roads'][selected_index]['end']
-            highlight_element = {'type': 'road', 'coords': selected_road_coords}
+        # Dropdown for selecting a building
+        building_options = [f"{bld['type']} at ({bld['coords'][0]}, {bld['coords'][1]})" for bld in data['buildings']]
+        selected_building = st.selectbox("Select a building to highlight:", options=building_options)
+        selected_index = building_options.index(selected_building)
+        selected_building_coords = data['buildings'][selected_index]['coords']
 
-        # Draw the grid with the selected element highlighted
-        fig = draw_grid(data, highlight_element=highlight_element)
+        # Draw the grid with the selected building highlighted and roads
+        fig = draw_grid(data, highlight_coords=selected_building_coords)
         st.pyplot(fig)
 
     with col3: