Update app.py

app.py

@@ -40,10 +40,9 @@ 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
+    # Create dictionaries to store node roles and colors based on attribute values
+    node_roles = {}
     node_colors = {}
-     # Maintain a set to track source nodes
-    source_nodes = set()
 
     
     for _, row in edge_15.iterrows():
@@ -52,12 +51,13 @@ 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]
-        if source not in source_nodes:
+        if source not in node_roles or node_roles[source] == "source":
+            node_roles[source] = "source"
             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'})
 
-        if target not in node_colors:
+        if target not in node_roles or node_roles[target] == "target":
+            node_roles[target] = "target"
             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)