Update app.py

app.py

@@ -21,8 +21,8 @@ def value_to_hex(value):
     rgba_color = cmap(value)
     return "#{:02X}{:02X}{:02X}".format(int(rgba_color[0] * 255), int(rgba_color[1] * 255), int(rgba_color[2] * 255))
 
-def subsetEdges(city, year):
-  info = taraf_info[(taraf_info['Year'] == year) & (taraf_info['City'] == city)]
+def subsetEdges(city, fstyear, lstyear):
+  info = taraf_info[(taraf_info['Year'] >= fstyear) & (taraf_info['City'] == city) & (taraf_info['Year'] <= lstyear)]
   narrators = edge_info[edge_info['Edge_ID'].isin(info['ID'].unique())]
   return narrators
 def splitIsnad(dataframe):
@@ -32,8 +32,8 @@ def splitIsnad(dataframe):
   return dataframe
 
     
-def network_visualizer(yaxis, city, year, num_nodes):
-    edges = splitIsnad(subsetEdges(city, year))[['Teacher', 'Student', yaxis]].reset_index()
+def network_visualizer(yaxis, city, fstyear,lastyr, num_nodes):
+    edges = splitIsnad(subsetEdges(city, fstyear, lastyr))[['Teacher', 'Student', yaxis]].reset_index()
     #.groupby(['Teacher', 'Student']).sum()
     if edges.shape[0] > num_nodes:
         edge_15 = edges.sample(num_nodes)
@@ -74,9 +74,10 @@ def network_visualizer(yaxis, city, year, num_nodes):
 with gr.Blocks() as demo:
   Yaxis = gr.Dropdown(choices = ['Tarafs', 'Hadiths', 'Isnads', 'Books'], value = 'Tarafs', label = 'Variable to Display', info = 'Choose the variable to visualize.')  
   Places = gr.Dropdown(choices = cities, value = 'المدينه', label = 'Location')
-  Last_Year = gr.Slider(min_year, max_year, value = 9, label = 'End', info = 'Choose the  year to display Narrators')
+  FirstYear = gr.Slider(min_year, max_year, value = -11, label = 'Begining', info = 'Choose the first year to display Narrators')
+  Last_Year = gr.Slider(min_year, max_year, value = 9, label = 'End', info = 'Choose the  last year to display Narrators')
   num_narrators = gr.Slider(0, 700, value = 400, label = 'Narrators', info = 'Choose the  number of Narrators to display')  
     
   btn = gr.Button('Submit')
-  btn.click(fn = network_visualizer, inputs = [Yaxis, Places, Last_Year, num_narrators], outputs = gr.HTML())
+  btn.click(fn = network_visualizer, inputs = [Yaxis, Places, FirstYear, Last_Year, num_narrators], outputs = gr.HTML())
   demo.launch()