Update app.py

app.py

@@ -34,14 +34,16 @@ def splitIsnad(dataframe):
     
 def network_visualizer(city, year, num_nodes):
     edge_15 = splitIsnad(subsetEdges(city, year))[['Teacher', 'Student', 'Hadiths']].groupby(['Teacher', 'Student']).sum().reset_index().sample(num_nodes)
+    teacher_hadiths = edge_15[['Teacher', 'Hadiths']].groupby('Teacher').sum().reset_index()
     net = Network()
     for _, row in edge_15.iterrows():
         source = row['Teacher']
         target = row['Student']
         attribute_value = row['Hadiths']
         edge_color = value_to_hex(attribute_value)
+        hadith_count = teacher_hadiths[teacher_hadiths['Teacher'] == source]['Hadiths'].to_list()[0]
 
-        net.add_node(source, color=value_to_hex(attribute_value), font = {'size':30, 'color': 'orange'})
+        net.add_node(source, color=value_to_hex(hadith_count), font = {'size':30, 'color': 'orange'})
         net.add_node(target, color=value_to_hex(attribute_value) , font = {'size': 20, 'color': 'red'})
         net.add_edge(source, target, color=edge_color, value=attribute_value)