Update app.py

app.py

@@ -39,6 +39,7 @@ def network_visualizer(city, year, num_nodes):
     else:
         edge_15 = edges.copy()
     teacher_hadiths = edge_15[['Teacher', 'Hadiths']].groupby('Teacher').sum().reset_index()
+    student_hadiths = edge_15[['Student', 'Hadiths']].groupby('Student').sum().reset_index()
     net = Network()
     # Create dictionaries to store node roles and colors based on attribute values
     node_roles = {}
@@ -51,10 +52,11 @@ def network_visualizer(city, year, num_nodes):
         attribute_value = row['Hadiths']
         edge_color = value_to_hex(attribute_value)
         hadith_count = teacher_hadiths[teacher_hadiths['Teacher'] == source]['Hadiths'].to_list()[0]
+        student_count = student_hadiths[student_hadiths['Student'] == target]['Hadiths'].to_list()[0]
         edge_color = value_to_hex(attribute_value)
         
         net.add_node(source, color=value_to_hex(hadith_count), font = {'size':30, 'color': 'orange'}, label = f"{source}\nHadiths: {hadith_count}")
-        net.add_node(target, color=value_to_hex(attribute_value) , font = {'size': 20, 'color': 'red'}, label = f"{target}\nHadiths: {attribute_value}")
+        net.add_node(target, color=value_to_hex(student_count) , font = {'size': 20, 'color': 'red'}, label = f"{target}\nHadiths: {student_count}")
         net.add_edge(source, target, color=edge_color, value=attribute_value)