Update app.py

app.py

@@ -127,40 +127,22 @@ def main():
 
 
 
-
         if sensors:  # Check if there are sensors to display
             st.header("Building Sensor Network")
-            sensor_store = TripleStore()
+            graph_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
+            # Central node for the building - adding as both subject and object to ensure it appears in the graph
             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=""))
+                sensor_id = f"Sensor: {sensor}"  # Label for sensor nodes
+                # Add triple: each sensor node is connected to the central building node
+                graph_store.add_triple(subject=building_name, predicate="has_sensor", object=sensor_id)
         
             # 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)
-        
-        
-
-
+            agraph_config = Config(height=600, width=600, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True, collapsible=True)
         
+            # Display the graph
+            agraph(nodes=list(graph_store.getNodes()), edges=list(graph_store.getEdges()), config=agraph_config)
 
         """