Update app.py

app.py

@@ -4,6 +4,11 @@ import os
 import pandas as pd
 from datasets import load_dataset
 
+from datasets import load_dataset
+from datasets import Features
+from datasets import Value
+from datasets import Dataset
+
 Secret_token = os.getenv('token')
 
 dataset = load_dataset('FDSRashid/hadith_info',data_files = 'Basic_Edge_Information.csv', token = Secret_token, split = 'train')
@@ -14,6 +19,16 @@ cities = taraf_info['City'].unique().tolist()
 min_year = int(taraf_info['Year'].min())
 max_year = int(taraf_info['Year'].max())
 
+
+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 = HF_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()
+
+
 def subset_city_year(city, year1, year2):
   edges = taraf_info[(taraf_info['Year'] >= year1) & (taraf_info['City'].isin(city)) & (taraf_info['Year'] <= year2)]
   return edges
@@ -32,12 +47,14 @@ def splitIsnad(dataframe):
 
 def get_narrators( city , year1, year2):
   try:
-    df = subset_city_year(city, year1, year2)
-    narrators = edge_info[edge_info['Edge_ID'].isin(df['ID'])]
-    fixed = narrators['Edge_Name'].reset_index()
-    return splitIsnad(fixed).reset_index().drop(['index','Edge_Name'], axis = 1)
+      df = subset_city_year(city, year1, year2)
+      narrators = edge_info[edge_info['Edge_ID'].isin(df['ID'])]
+      fixed = splitIsnad(narrators)
+      fixed['Teacher Reports'] = narrator_bios[narrator_bios['Rawi ID'] == fixed['Teacher_ID']]
+      fixed['Teacher Reports'] = narrator_bios[narrator_bios['Rawi ID'] == fixed['Teacher_ID']]  
+      return fixed[['Teacher', 'Student', 'Teacher Reports', 'Student Reports']]
   except Exception as e:
-    return str(e)
+      return str(e)