Update app.py

app.py

@@ -22,11 +22,20 @@ def subset_year(year = 50):
     edges = taraf_info[(taraf_info['Year'] == year)]
     return edges
 
+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 get_narrators( city , year):
   try:
     df = subset_city_year(city, year)
     narrators = edge_info[edge_info['Edge_ID'].isin(df['ID'])]
-    return narrators['Edge_Name'].reset_index().rename(columns = {'Edge_Name': 'Teacher To Student'})
+    fixed = narrators['Edge_Name'].reset_index()
+        return splitIsnad(fixed)
   except Exception as e:
     return str(e)