Update app.py

app.py

@@ -124,7 +124,7 @@ def visualize_isnad(taraf_num, yaxis):
         else:
             net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{row["Teacher"]} \n {teacher_info["Narrator Rank"].to_list()[0]} \n ID: {row["Source"]} - Gen {teacher_gen}')
         net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{target} \n{student_info["Narrator Rank"].to_list()[0]} \n ID: {row["Destination"]} - Gen {student_gen}')
-        net.add_edge(source, target, color = value_to_hex(int(row[f'{yaxis} Count'].to_list()[0])), label = f"{row[f'{yaxis} Count'].to_list()[0]}")
+        net.add_edge(source, target, color = value_to_hex(int(row[f'{yaxis} Count'])), label = f"{row[f'{yaxis} Count']}")
     net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
     html = net.generate_html()
     html = html.replace("'", "\"")