Update app.py

app.py

@@ -23,13 +23,13 @@ min_year = int(taraf_s['Year'].min())
 max_year = int(taraf_s['Year'].max())
 
 
-def plot_taraf_map(min_year = 0, max_year = 400):
+def plot_taraf_map(yaxis,min_year = 0, max_year = 400):
     if min_year > max_year:
         raise gr.Error('Minimum Year cannot be bigger than final ear!')
     filtered = geodf[(geodf['Year'] >= min_year) & (geodf['Year'] <= max_year)]
-    temp = filtered[['City', 'Taraf']].groupby('City').sum().join(filtered[['City', 'Coordinates']].set_index('City'))
+    temp = filtered[['City', yaxis]].groupby('City').sum().join(filtered[['City', 'Coordinates']].set_index('City'))
     filtered = geopandas.GeoDataFrame(temp, geometry= 'Coordinates').reset_index()
-    fig = px.scatter_mapbox(data_frame = filtered, lat = filtered.geometry.y, lon = filtered.geometry.x,size = filtered.Taraf,color = filtered.Taraf, title = 'Number of Tarafs in Place', opacity = .5, zoom = 0, hover_data = 'City')
+    fig = px.scatter_mapbox(data_frame = filtered, lat = filtered.geometry.y, lon = filtered.geometry.x,size = yaxis,color = yaxis, title = 'Number of Tarafs in Place', opacity = .5, zoom = 0, hover_data = 'City')
     fig.update_layout(title_font_color = 'red', title_x = .5, mapbox_style="stamen-terrain")
     return fig
 
@@ -37,8 +37,9 @@ def plot_taraf_map(min_year = 0, max_year = 400):
 
 
 with gr.Blocks() as demo:
+  Yaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad'], value = 'Taraf', label = 'Variable to Display')  
   First_Year = gr.Slider(min_year, max_year, value = 0, label = 'Begining', info = 'Choose the first year to display Tarafs')
   Last_Year = gr.Slider(min_year, max_year, value = 400, label = 'End', info = 'Choose the last year to display Tarafs')
   btn = gr.Button('Submit')
-  btn.click(fn = plot_taraf_map, inputs = [First_Year, Last_Year], outputs = gr.Plot())
+  btn.click(fn = plot_taraf_map, inputs = [Yaxis, First_Year, Last_Year], outputs = gr.Plot())
   demo.launch()