Update app.py

app.py

@@ -101,11 +101,14 @@ def visualize_isnad(taraf_num, yaxis):
         isnad = isnad_info[(isnad_info['Source'] == row['Source']) & (isnad_info['Destination'] == row['Destination'])]
         teacher_narrations = teacher_info['Number of Narrations'].to_list()[0]
         student_narrations = student_info['Number of Narrations'].to_list()[0]
+
+        teacher_gen = teacher_info['Generation'].to_list()[0]
+        student_gen = student_info['Generation'].to_list()[0]
         if row['Source'] == '99999':
             net.add_node(source, font = {'size':50, 'color': 'Black'}, color = '#000000')
         else:
-            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{source} \n {teacher_info["Narrator Rank"].to_list()[0]} \n ID: {row["Source"]}')
-        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"]}')
+            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{source} \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(isnad[f'{yaxis} Count'].to_list()[0])), label = f"{isnad[f'{yaxis} Count'].to_list()[0]}")
     net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
     html = net.generate_html()
@@ -141,11 +144,14 @@ def visualize_subTaraf(df, yaxis):
         isnad = isnad_info[(isnad_info['Source'] == row['Source']) & (isnad_info['Destination'] == row['Destination'])]
         teacher_narrations = teacher_info['Number of Narrations'].to_list()[0]
         student_narrations = student_info['Number of Narrations'].to_list()[0]
+        teacher_gen = teacher_info['Generation'].to_list()[0]
+        student_gen = student_info['Generation'].to_list()[0]
+        
         if row['Source'] == '99999':
             net.add_node(source, font = {'size':50, 'color': 'Black'}, color = '#000000')
         else:
-            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{source} \n {teacher_info["Narrator Rank"].to_list()[0]}  \n ID: {row["Source"]}')
-        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"]}')
+            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{source} \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(isnad[f'{yaxis} Count'].to_list()[0])), label = f"{isnad[f'{yaxis} Count'].to_list()[0]}")
     net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
     html = net.generate_html()