Update app.py

app.py

@@ -70,7 +70,7 @@ def get_node_info(node):
     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_narrator = edge_info[(edge_info['Source'] == str(narrator_id)) | (edge_info['Destination'] == 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] )