Update app.py

app.py

@@ -60,33 +60,15 @@ narrator_bios['Generation'] = narrator_bios['Generation'].replace([None], [-1])
 narrator_bios['Generation'] = narrator_bios['Generation'].astype(int)
 
 
- def get_node_info(node):
+def get_node_info(node):
     node_info = narrator_bios[narrator_bios['Rawi ID'] == int(node)]
-    student_narrations = node_info['Number of Narrations'].to_list()
-    if len(student_narrations):
-        student_narrations = student_narrations[0]
-    else:
-        student_narrations = 1
-    student_gen = node_info['Generation'].to_list()
-    if len(student_gen):
-        student_gen = student_gen[0]
-    else:
-        student_gen = -1
-    student_rank = node_info["Narrator Rank"].to_list()
-    if len(student_rank):
-        student_rank = student_rank[0]
-    else:
-        student_rank = 'فلان'
-    node_name = node_info['Famous Name'].to_list()
-    if len(node_name):
-        node_name = node_name[0]
-    else:
-        node_name = 'فلان'
+    # Extract values with defaults, using `.iloc` if available, otherwise default
+    student_narrations = node_info['Number of Narrations'].iloc[0] if not node_info.empty else 1
+    student_gen = node_info['Generation'].iloc[0] if not node_info.empty else -1
+    student_rank = node_info['Narrator Rank'].iloc[0] if not node_info.empty else 'فلان'
+    node_name = node_info['Famous Name'].iloc[0] if not node_info.empty else 'فلان'
     return node_info,student_narrations,student_gen, student_rank, node_name
 
-
-
-
 def network_narrator(narrator_id):
     edge_narrator = edge_info[(edge_info['Teacher_ID'] == str(narrator_id)) | (edge_info['Student_ID'] == str(narrator_id))]
     edge_full = edge_narrator[['Taraf Count', 'Hadith Count', 'Isnad Count', 'Source', 'Book Count', 'Destination']]