Update app.py

app.py

@@ -25,25 +25,21 @@ max_year = int(taraf_s['Year'].max())
 def plot_timeline(yaxis, citi = ['المدينه', 'بغداد', 'كوفة', 'بصرة'], min_year = 0, max_year = 400):
     if min_year >= max_year:
         raise gr.error('Min Year Cannot be Bigger than Max Year!')
-    filtered = taraf_s[taraf_s['City'].isin(citi) & (taraf_s['Year'] >= min_year) & (taraf_s['Year'] <= max_year)]
-    fig = px.line(data_frame = filtered, x = 'Year', y = yaxis, title = f"{yaxis} per Year", color = 'City', template = 'plotly_dark' )
+    if citi = ['All']:
+        filtered = taraf_s.copy()[(taraf_s['Year'] >= min_year) & (taraf_s['Year'] <= max_year)].groupby(['Year']).sum()
+        fig = px.line(data_frame = filtered, x = 'Year', y = yaxis, title = f"{yaxis} per Year", color = 'City', template = 'plotly_dark' )
+    else:
+        filtered = taraf_s[taraf_s['City'].isin(citi) & (taraf_s['Year'] >= min_year) & (taraf_s['Year'] <= max_year)]
+        fig = px.line(data_frame = filtered, x = 'Year', y = yaxis, title = f"{yaxis} per Year", color = 'City', template = 'plotly_dark' )
     fig.update_layout(legend_x=1, legend_y=0, title = {'x':.5})
     return fig
 
-# app = gradio.Interface(plot_timeline,
-#              [gradio.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Ranking'], value = 'Taraf', label = 'Y Axis'), gradio.Dropdown(choices = cities, value = ['المدينه', 'بغداد', 'كوفة', 'بصرة'], multiselect=True),
-#                         gradio.Slider(min_year, max_year, value = 0, label = 'Begining', info = 'Choose the first year to display Tarafs'),
-#                         gradio.Slider(min_year, max_year, value = 400, label = 'End', info = 'Choose the last year to display Tarafs')
-
-#                         ],
-#                        gradio.Plot()).launch()
-
 
 demo = gr.Blocks()
 with demo:
     with gr.Row():
         y_value = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Ranking'], value = 'Taraf', label = 'Y Axis')
-        city = gr.Dropdown(choices = cities, value = ['المدينه', 'بغداد', 'كوفة', 'بصرة'], multiselect=True, label = 'City')
+        city = gr.Dropdown(choices = cities + ['All'], value = ['المدينه', 'بغداد', 'كوفة', 'بصرة'], multiselect=True, label = 'City')
     begining = gr.Slider(min_year, max_year, value = 0, label = 'First Year', info = 'Choose the first year to display Tarafs') 
     end = gr.Slider(min_year, max_year, value = 400, label = 'Last Year', info = 'Choose the last year to display Tarafs')
     btn = gr.Button("Plot")