Update pages/17_Graphs2.py

pages/17_Graphs2.py

@@ -37,9 +37,10 @@ def create_sample_graph():
     num_nodes = 10
     num_edges = 15
 
+    # Create a random graph
     graph = nx.gnm_random_graph(num_nodes, num_edges, directed=True)
 
-    # Create synthetic features
+    # Generate synthetic features
     years_published = np.random.randint(1990, 2022, size=num_nodes).astype(np.float32)
     num_authors = np.random.randint(1, 10, size=num_nodes).astype(np.float32)
     citation_weights = np.random.uniform(0.1, 5.0, size=num_edges).astype(np.float32)
@@ -48,6 +49,10 @@ def create_sample_graph():
     node_features = np.stack([years_published, num_authors], axis=-1)
     edge_features = citation_weights.reshape(-1, 1)
 
+    # Assign random titles to nodes
+    paper_titles = [f"Paper {i+1}" for i in range(num_nodes)]
+    nx.set_node_attributes(graph, {i: {'title': title} for i, title in enumerate(paper_titles)})
+
     graph_tensor = tfgnn.GraphTensor.from_pieces(
         node_sets={
             "papers": tfgnn.NodeSet.from_fields(
@@ -88,7 +93,8 @@ def main():
     st.subheader("Sample Graph Visualization")
     fig, ax = plt.subplots(figsize=(10, 8))
     pos = nx.spring_layout(nx_graph)
-    nx.draw(nx_graph, pos, with_labels=True, node_color='lightblue',
+    labels = nx.get_node_attributes(nx_graph, 'title')
+    nx.draw(nx_graph, pos, labels=labels, with_labels=True, node_color='lightblue',
             node_size=500, arrowsize=20, ax=ax)
     st.pyplot(fig)