Update app.py

app.py

@@ -14,18 +14,21 @@ cities = taraf_info['City'].unique().tolist()
 min_year = int(taraf_info['Year'].min())
 max_year = int(taraf_info['Year'].max())
 
-def subset_city_year( city , year ):
-    edges = taraf_info[(taraf_info['Year'] == year) & (taraf_info['City'].isin(city))]
-    return edges
+def subset_city_year(city, year):
+  edges = taraf_info[(taraf_info['Year'] == year) & (taraf_info['City'].isin(city))]
+  return edges
 
 def subset_year(year = 50):
     edges = taraf_info[(taraf_info['Year'] == year)]
     return edges
 
-def get_narrators(city, year ):
+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().drop('index', axis = 1).rename(columns = {'Edge_Name': 'Teacher To Student'})
+  except Exception as e:
+    return str(e)
 
 
 
@@ -35,4 +38,4 @@ with gr.Blocks() as demo:
   Last_Year = gr.Slider(min_year, max_year, value = 50, label = 'End', info = 'Choose the  year to display Narrators')
   btn = gr.Button('Submit')
   btn.click(fn = get_narrators, inputs = [Places, Last_Year], outputs = gr.DataFrame())
-  demo.launch()
+  demo.launch()