Update app.py

app.py

@@ -5,12 +5,27 @@ import numpy as np
 import pandas as pd
 import os
 from datasets import load_dataset
+from datasets import Features
+from datasets import Value
+from datasets import Dataset
 import matplotlib.pyplot as plt
 
 Secret_token = os.getenv('token')
 
 dataset = load_dataset('FDSRashid/hadith_info',data_files = 'Basic_Edge_Information.csv', token = Secret_token, split = 'train')
 dataset2 = load_dataset('FDSRashid/hadith_info',data_files = 'Taraf_Info.csv', token = Secret_token, split = 'train')
+
+
+
+features = Features({'Rawi ID': Value('int32'), 'Famous Name': Value('string'), 'Narrator Rank': Value('string'), 'Number of Narrations': Value('string')})
+narrator_bios = load_dataset("FDSRashid/hadith_info", data_files = 'Teacher_Bios.csv', token = Secret_token,features=features )
+narrator_bios = narrator_bios['train'].to_pandas()
+narrator_bios.loc[49845, 'Narrator Rank'] = 'رسول الله'
+narrator_bios.loc[49845, 'Number of Narrations'] = 0
+narrator_bios['Number of Narrations'] = narrator_bios['Number of Narrations'].astype(int)
+narrator_bios.loc[49845, 'Number of Narrations'] = narrator_bios['Number of Narrations'].sum()
+
+
 edge_info = dataset.to_pandas()
 taraf_info = dataset2.to_pandas()
 cities = taraf_info['City'].unique().tolist()
@@ -33,14 +48,14 @@ def splitIsnad(dataframe):
 
     
 def network_visualizer(yaxis, city, fstyear,lastyr, num_nodes):
-    edges = splitIsnad(subsetEdges(city, fstyear, lastyr))[['Teacher', 'Student', yaxis]].reset_index()
+    edges = subsetEdges(city, fstyear, lastyr)
+    edges_split = splitIsnad(edges).reset_index()
     #.groupby(['Teacher', 'Student']).sum()
-    if edges.shape[0] > num_nodes:
-        edge_15 = edges.sort_values(by=yaxis, ascending=False).head(num_nodes)
+    if edges_split.shape[0] > num_nodes:
+        edge_15 = edges_split.sort_values(by=yaxis, ascending=False).head(num_nodes)
     else:
-        edge_15 = edges.copy()
-    teacher_hadiths = edge_15[['Teacher', yaxis]].groupby('Teacher').sum().reset_index()
-    student_hadiths = edge_15[['Student', yaxis]].groupby('Student').sum().reset_index()
+        edge_15 = edges_split.copy()
+
     net = Network()
 
 
@@ -50,11 +65,12 @@ def network_visualizer(yaxis, city, fstyear,lastyr, num_nodes):
         target = row['Student']
         attribute_value = row[yaxis]
         edge_color = value_to_hex(np.log10(attribute_value))
-        hadith_count = teacher_hadiths[teacher_hadiths['Teacher'] == source][yaxis].to_list()[0]
-        student_count = student_hadiths[student_hadiths['Student'] == target][yaxis].to_list()[0]
+        teacher_info = narrator_bios[narrator_bios['Rawi ID'] == row['Teacher_ID']]
+        student_info = narrator_bios[narrator_bios['Rawi ID'] == row['Student_ID']]
+       
         
-        net.add_node(source, color=value_to_hex(hadith_count), font = {'size':30, 'color': 'orange'}, label = f"{source}\n{yaxis}: {hadith_count}")
-        net.add_node(target, color=value_to_hex(student_count) , font = {'size': 20, 'color': 'red'}, label = f"{target}\n{yaxis}: {student_count}")
+        net.add_node(source, color=value_to_hex(teacher_info.loc[0, 'Number of Narrations']), font = {'size':30, 'color': 'orange'})#, label = f"{source}\n{yaxis}: {hadith_count}")
+        net.add_node(target, color=value_to_hex(student_info.loc[0, 'Number of Narrations']) , font = {'size': 20, 'color': 'red'})#, label = f"{target}\n{yaxis}: {student_count}")
         net.add_edge(source, target, color=edge_color, value=attribute_value, label = f"{source} to {target}\n{yaxis}: {attribute_value}")