gmedin commited on
Commit
a1850db
·
verified ·
1 Parent(s): 4622e0f

node types

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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[int(start_node)].get('title', 'No title available')
117
- start_in_degree = graph.in_degree(int(start_node))
118
- start_out_degree = graph.out_degree(int(start_node))
119
  start_node_size = (start_in_degree + start_out_degree) * 0.15
120
- start_rank = rankings.index(int(start_node)) if rankings and int(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
- int(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,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(int(start_node))
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 = []