Update app.py

app.py

@@ -25,12 +25,13 @@ 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!')
-    if citi == ['All']:
-        filtered = taraf_s.copy()[(taraf_s['Year'] >= min_year) & (taraf_s['Year'] <= max_year)].groupby(['Year']).sum().reset_index()
-        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' )
+    
+    filtered = taraf_s[taraf_s['City'].isin(citi) & (taraf_s['Year'] >= min_year) & (taraf_s['Year'] <= max_year)].drop(['Unnamed: 0', 'Ranking'], axis = 1)
+    if 'All' in citi:
+        tot = taraf_s.groupby(['Year']).sum().reset_index().drop(['Unnamed: 0', 'Ranking'], axis = 1)
+        tot['City'] = 'All'
+        filtered = pd.concat([filtered, tot])
+    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