Update app.py

app.py

@@ -40,16 +40,31 @@ def network_visualizer(city, year, num_nodes):
         edge_15 = edges.copy()
     teacher_hadiths = edge_15[['Teacher', 'Hadiths']].groupby('Teacher').sum().reset_index()
     net = Network()
+     # Create a dictionary to store node colors based on attribute values
+    node_colors = {}
+     # Maintain a set to track source nodes
+    source_nodes = set()
+
+    
     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]
+        if source not in source_nodes:
+            node_colors[source] = value_to_hex(hadith_count)
+            source_nodes.add(source)
+            net.add_node(source, color=value_to_hex(hadith_count), 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'})
+        if target not in node_colors:
+            node_colors[target] = value_to_hex(attribute_value)
+            net.add_node(target, color=value_to_hex(attribute_value), font={'size': 20, 'color': 'red'})
+        edge_color = value_to_hex(attribute_value)
         net.add_edge(source, target, color=edge_color, value=attribute_value)
+        #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)
 
 
     net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)