node types
Browse files
app.py
CHANGED
|
@@ -113,18 +113,18 @@ def dynamic_visualize_graph(graph, start_node, layers=3, top_k=5, show_titles=Fa
|
|
| 113 |
max_rank = len(rankings) if rankings else 0
|
| 114 |
|
| 115 |
# Add the starting node, color it red, and include a tooltip
|
| 116 |
-
start_title = graph.nodes[
|
| 117 |
-
start_in_degree = graph.in_degree(
|
| 118 |
-
start_out_degree = graph.out_degree(
|
| 119 |
start_node_size = (start_in_degree + start_out_degree) * 0.15
|
| 120 |
-
start_rank = rankings.index(
|
| 121 |
if rankings:
|
| 122 |
start_border_color = rank_to_color(start_rank, max_rank)
|
| 123 |
else:
|
| 124 |
start_border_color = 'darkblue'
|
| 125 |
label = str(start_node) if not show_titles else f"{str(start_node)}: {start_title[:15]}..."
|
| 126 |
net.add_node(
|
| 127 |
-
|
| 128 |
label=label,
|
| 129 |
color={"background": "darkblue", "border": start_border_color},
|
| 130 |
title=f"{start_title}, In-degree: {start_in_degree}, Out-degree: {start_out_degree}, Rank: {start_rank}",
|
|
@@ -132,7 +132,7 @@ def dynamic_visualize_graph(graph, start_node, layers=3, top_k=5, show_titles=Fa
|
|
| 132 |
borderWidth=3,
|
| 133 |
borderWidthSelected=6
|
| 134 |
)
|
| 135 |
-
visited_nodes.add(
|
| 136 |
|
| 137 |
for layer in range(layers):
|
| 138 |
next_nodes = []
|
|
|
|
| 113 |
max_rank = len(rankings) if rankings else 0
|
| 114 |
|
| 115 |
# Add the starting node, color it red, and include a tooltip
|
| 116 |
+
start_title = graph.nodes[str(start_node)].get('title', 'No title available')
|
| 117 |
+
start_in_degree = graph.in_degree(str(start_node))
|
| 118 |
+
start_out_degree = graph.out_degree(str(start_node))
|
| 119 |
start_node_size = (start_in_degree + start_out_degree) * 0.15
|
| 120 |
+
start_rank = rankings.index(str(start_node)) if rankings and str(start_node) in rankings else max_rank + 1
|
| 121 |
if rankings:
|
| 122 |
start_border_color = rank_to_color(start_rank, max_rank)
|
| 123 |
else:
|
| 124 |
start_border_color = 'darkblue'
|
| 125 |
label = str(start_node) if not show_titles else f"{str(start_node)}: {start_title[:15]}..."
|
| 126 |
net.add_node(
|
| 127 |
+
str(start_node),
|
| 128 |
label=label,
|
| 129 |
color={"background": "darkblue", "border": start_border_color},
|
| 130 |
title=f"{start_title}, In-degree: {start_in_degree}, Out-degree: {start_out_degree}, Rank: {start_rank}",
|
|
|
|
| 132 |
borderWidth=3,
|
| 133 |
borderWidthSelected=6
|
| 134 |
)
|
| 135 |
+
visited_nodes.add(str(start_node))
|
| 136 |
|
| 137 |
for layer in range(layers):
|
| 138 |
next_nodes = []
|