Update app.py

app.py

@@ -51,20 +51,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]
-        if source not in node_roles or node_roles[source] == "target":
-            node_roles[source] = "source"
-            node_colors[source] = value_to_hex(hadith_count)
-            net.add_node(source, color=value_to_hex(hadith_count), font={'size': 30, 'color': 'orange'}, label=f"{source}\nHadiths: {hadith_count}")
-
-        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'}, label=f"{target}")
         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.add_node(source, color=value_to_hex(hadith_count), font = {'size':30, 'color': 'orange'}, label = f"Hadiths: {hadith_count}")
+        net.add_node(target, color=value_to_hex(attribute_value) , font = {'size': 20, 'color': 'red'}, label = f"Hadith from Teacher: {attribute_value}")
+        net.add_edge(source, target, color=edge_color, value=attribute_value)
 
 
     net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)