using isnad info as opposed to edge info

app.py

@@ -18,7 +18,7 @@ css = """
 
 Secret_token = os.getenv('HF_Token')
 
-dataset = load_dataset('FDSRashid/hadith_info',data_files = 'Basic_Edge_Information.csv', token = Secret_token, split = 'train')
+edge_info = load_dataset('FDSRashid/hadith_info',token = Secret_token, data_files = 'isnad_info.csv', split = 'train').to_pandas()
 
 lst = ['Rawi ID',
  'Gender',
@@ -60,19 +60,38 @@ narrator_bios['Generation'] = narrator_bios['Generation'].replace([None], [-1])
 narrator_bios['Generation'] = narrator_bios['Generation'].astype(int)
 
 
-edge_info = dataset.to_pandas()
+ 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 = 'فلان'
+    return node_info,student_narrations,student_gen, student_rank, node_name
 
 
-def splitIsnad(dataframe):
-    teacher_student =dataframe['Edge_Name'].str.split(' TO ')
-    dataframe['Teacher'] = teacher_student.apply(lambda x: x[0])
-    dataframe['Student'] = teacher_student.apply(lambda x: x[1])
-    return dataframe
 
 
 def network_narrator(narrator_id):
-    edge_narrator = edge_info[(edge_info['Teacher_ID'] == narrator_id) | (edge_info['Student_ID'] == narrator_id)]
-    edge_full = splitIsnad(edge_narrator[['Tarafs', 'Hadiths', 'Isnads', 'Edge_Name', 'Books']]).drop(['Edge_Name'], axis=1)
+    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']]
+    edge_full['Teacher'] = edge_full['Source'].apply(lambda x: get_node_info(x)[-1] )
+    edge_full['Student'] = edge_full['Destination'].apply(lambda x: get_node_info(x)[-1] )
     return edge_full
 
 def narrator_retriever(name):