Update app.py

app.py

@@ -118,12 +118,21 @@ def visualize_isnad(taraf_num, yaxis):
             student_gen = student_gen[0]
         else:
             student_gen = -1
-            
+        teacher_rank = teacher_info["Narrator Rank"].to_list()
+        if len(teacher_rank):
+            teacher_rank = teacher_rank[0]
+        else:
+            teacher_rank = 'فلان'
+            student_rank = student_info["Narrator Rank"].to_list()
+        if len(student_rank):
+            student_rank = student_rank[0]
+        else:
+            student_rank = 'فلان'
         if row['Source'] == '99999':
             net.add_node(source, font = {'size':50, 'color': 'Black'}, color = '#000000', label = f'{row["Teacher"]}')
         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'{row["Student"]} \n{student_info["Narrator Rank"].to_list()[0]} \n ID: {row["Destination"]} - Gen {student_gen}')
+            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{row["Teacher"]} \n {teacher_rank} \n ID: {row["Source"]} - Gen {teacher_gen}')
+        net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{row["Student"]} \n{student_rank} \n ID: {row["Destination"]} - Gen {student_gen}')
         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()