Update app.py

app.py

@@ -2,27 +2,10 @@ import streamlit as st
 import json
 import matplotlib.pyplot as plt
 import time
-
-
-st.set_page_config(layout="wide")
-
-
-
-
-# HIN Number +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 from SPARQLWrapper import SPARQLWrapper, JSON
 from streamlit_agraph import agraph, TripleStore, Node, Edge, Config
-import json
-
-
-# Green Village ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-
 
-# Function to load and parse the JSON data
-def load_data(filename):
-    with open(filename, 'r') as file:
-        data = json.load(file)
-    return data
+st.set_page_config(layout="wide")
 
 # Color codes updated with hexadecimal values for compatibility
 color_codes = {
@@ -49,35 +32,39 @@ color_codes = {
     "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
+# Function to load and parse the JSON data
+def load_data(filename):
+    with open(filename, 'r') as file:
+        data = json.load(file)
+    return data
+
+# Function to draw the grid layout with color coding and to highlight a selected building
+def draw_grid(data, highlight=None):
     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
+
+    highlight_coords = None
+    if highlight is not None:
+        highlight_type, highlight_coords_str = highlight.split(' at ')
+        highlight_coords = tuple(map(int, highlight_coords_str.strip('()').split(',')))
+
     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))
+        color = color_codes.get(b_type, '#FFFFFF')
+        edgecolor = 'red' if coords == highlight_coords else 'black'
+        linewidth = 2 if coords == highlight_coords else 1
+        ax.add_patch(plt.Rectangle((coords[1], nrows-coords[0]-size), size, size, color=color, edgecolor=edgecolor, linewidth=linewidth))
         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
+    for road in data.get('roads', []):
         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']))
@@ -85,76 +72,59 @@ def draw_grid(data):
             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']))
 
-
-    # Reverse the y-axis numbers
-    # ax.invert_yaxis()            
-    # 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():
     st.title('Green Smart Village Application')
-    
-    # Creating three columns
+
     col1, col2, col3 = st.columns(3)
-    
+
     with col1:
         st.header("Today's Agenda")
-        # 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")
         st.header("Agent Advisors")
-        st.write("Would you like to optimize your HIN number")
+        st.write("Would you like to optimize your HIN number?")
         st.header("My Incentive")
         st.write("Total incentive for HIN optimization")
-    
+
     with col2:
         st.header("Green Smart Village Layout")
-        # Load and display the data with color coding
-        data = load_data('grid.json')  # Ensure this path is correct
-        fig = draw_grid(data)
-        st.pyplot(fig)
-
-        # Interactivity: Selecting a building to display sensor data
+        data = load_data('grid.json')
         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 view sensors:", options=building_options)
+        
+        fig = draw_grid(data, highlight=selected_building)
+        st.pyplot(fig)
+
         selected_index = building_options.index(selected_building)
         sensors = data['buildings'][selected_index]['sensors']
         st.write(f"Sensors in selected building: {', '.join(sensors)}")
 
-
-
     with col3:
         st.header("Check Your HIN Number")
-        config = Config(height=400, width=400, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True,
-                  collapsible=True)
+        config = Config(height=400, width=400, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True, collapsible=True)
 
-        #based on Insurance Fraud
         with open("data/fraud.json", encoding="utf8") as f:
-          fraud_file = json.loads(f.read())
-          st.session_state['fraud_topic'] = fraud_file
-          fraud_store = TripleStore()
-          for sub_graph in fraud_file["children"]:
-            fraud_store.add_triple(fraud_file["name"], "has_subgroup", sub_graph["name"], picture=fraud_file["img"])
-            for node in sub_graph["children"]:
-              node1 = node["role"]
-              link = "blongs_to"
-              node2 = sub_graph["name"]
-              pic = node["img"]
-              fraud_store.add_triple(node1, link, node2, picture=pic)
-          agraph(list(fraud_store.getNodes()), (fraud_store.getEdges()), config)
-
+            fraud_file = json.loads(f.read())
+            st.session_state['fraud_topic'] = fraud_file
+            fraud_store = TripleStore()
+            for sub_graph in fraud_file["children"]:
+                fraud_store.add_triple(fraud_file["name"], "has_subgroup", sub_graph["name"], picture=fraud_file["img"])
+                for node in sub_graph["children"]:
+                    node1 = node["role"]
+                    link = "belongs_to"
+                    node2 = sub_graph["name"]
+                    pic = node["img"]
+                    fraud_store.add_triple(node1, link, node2, picture=pic)
+            agraph(list(fraud_store.getNodes()), list(fraud_store.getEdges()), config)
 
         hin_number = st.text_input("Enter your HIN number:")
         if hin_number:
-            st.write("HIN number details...")  # Placeholder for actual HIN number check
-
-    
+            st.write("HIN number details...")
 
 if __name__ == "__main__":
-    main()
+    main()