Update app.py

app.py

@@ -67,10 +67,11 @@ def network_visualizer(yaxis, city, fstyear,lastyr, num_nodes):
         edge_color = value_to_hex(np.log10(attribute_value))
         teacher_info = narrator_bios[narrator_bios['Rawi ID'] == row['Teacher_ID']]
         student_info = narrator_bios[narrator_bios['Rawi ID'] == row['Student_ID']]
-        print(teacher_info)
+        teacher_narrations = teacher_info['Number of Narrations'].to_list()[0]
+        student_narrations = student_info['Number of Narrations'].to_list()[0]
         
-        net.add_node(source, color=value_to_hex(teacher_info['Number of Narrations'][0]), font = {'size':30, 'color': 'orange'})#, label = f"{source}\n{yaxis}: {hadith_count}")
-        net.add_node(target, color=value_to_hex(student_info['Number of Narrations'][0]) , font = {'size': 20, 'color': 'red'})#, label = f"{target}\n{yaxis}: {student_count}")
+        net.add_node(source, color=value_to_hex(teacher_narrations), font = {'size':30, 'color': 'orange'})#, label = f"{source}\n{yaxis}: {hadith_count}")
+        net.add_node(target, color=value_to_hex(student_narrations) , font = {'size': 20, 'color': 'red'})#, label = f"{target}\n{yaxis}: {student_count}")
         net.add_edge(source, target, color=edge_color, value=attribute_value, label = f"{source} to {target}\n{yaxis}: {attribute_value}")