Update app.py

app.py

@@ -84,28 +84,7 @@ def draw_grid(data, highlight_coords=None):
     ax.set_title('Village Layout with Color Coding')
     return fig
 
-    # Assuming sensors are defined in your data, display them
-    sensors = data['buildings'][selected_index].get('sensors', [])
-    st.write(f"Sensors in selected building: {', '.join(sensors)}")
-
-    # Only attempt to display the graph if sensors are available
-    if sensors:
-        st.header("Building Sensor Network")
-        sensor_store = TripleStore()
-        building_name = f"{data['buildings'][selected_index]['type']} ({selected_building_coords[0]}, {selected_building_coords[1]})"
-        
-        # Create a node for the building
-        sensor_store.add_node(Node(id=building_name, label=building_name, color="#FFD700"))
-
-        # Create nodes for each sensor and add edges from the building to the sensors
-        for sensor in sensors:
-            sensor_id = f"{sensor}-{selected_building_coords}"  # Unique identifier for each sensor
-            sensor_store.add_node(Node(id=sensor_id, label=sensor, color="#90EE90"))
-            sensor_store.add_edge(Edge(source=building_name, target=sensor_id, label="contains"))
-
-        agraph_config = Config(height=400, width=400, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True, collapsible=True)
-        agraph(list(sensor_store.getNodes()), list(sensor_store.getEdges()), config=agraph_config)
-
+  
 # Main Streamlit app function
 def main():
     st.title('Green Smart Village Application')
@@ -146,6 +125,45 @@ def main():
         config = Config(height=400, width=400, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True,
                   collapsible=True)
 
+
+
+
+        if sensors:  # Check if there are sensors to display
+            st.header("Building Sensor Network")
+            sensor_store = TripleStore()
+            building_name = f"{data['buildings'][selected_index]['type']} ({selected_building_coords[0]}, {selected_building_coords[1]})"
+            
+            # Create a central node for the building
+            central_node_color = "#FFD700"  # Gold color for the central node (building)
+            sensor_store.add_node(Node(id=building_name, label=building_name, color=central_node_color, size=400))
+        
+            # Define a color for sensor nodes
+            sensor_node_color = "#90EE90"  # Light green for sensor nodes
+        
+            # Create nodes for each sensor and link them to the central building node
+            for sensor in sensors:
+                sensor_id = f"{sensor}-{selected_building_coords}"  # Ensuring uniqueness
+                sensor_label = f"Sensor: {sensor}"  # Label for the sensor nodes
+                
+                # Add sensor node to the store
+                sensor_store.add_node(Node(id=sensor_id, label=sensor_label, color=sensor_node_color, size=300))
+                
+                # Connect each sensor node to the central building node
+                sensor_store.add_edge(Edge(source=building_name, target=sensor_id, label=""))
+        
+            # Configuration for the graph visualization
+            agraph_config = Config(height=600, width=600, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True, collapsible=True, node={'labelProperty':'label'})
+            
+            # Display the graph
+            agraph(nodes=list(sensor_store.getNodes()), edges=list(sensor_store.getEdges()), config=agraph_config)
+        
+        
+
+
+        
+
+        """
+    
         #based on Insurance Fraud
         with open("data/fraud.json", encoding="utf8") as f:
           fraud_file = json.loads(f.read())
@@ -160,7 +178,7 @@ def main():
               pic = node["img"]
               fraud_store.add_triple(node1, link, node2, picture=pic)
           agraph(list(fraud_store.getNodes()), (fraud_store.getEdges()), config)
-
+         """
 
         hin_number = st.text_input("Enter your HIN number:")
         if hin_number: