Update app.py

app.py

@@ -60,8 +60,8 @@ def init_baselines():
     raw_display = raw_display.reset_index(drop=True)
     raw_display.replace('', np.nan, inplace=True)
     raw_display = raw_display.apply(pd.to_numeric, errors='coerce').fillna(raw_display)
-    raw_display['Team Date'] = raw_display['Team'] + " " + raw_display['Date'].astype(str)
     game_model = raw_display[raw_display['Date'] != ""]
+    game_model['Team Date'] = game_model['Team'] + " " + game_model['Date'].astype(str)
 
     worksheet = sh.worksheet('SeasonExport')
     raw_display = pd.DataFrame(worksheet.get_values())
@@ -89,9 +89,9 @@ with tab1:
     if split_var1 == 'Specific Teams':
         team_var1 = st.multiselect('Which teams would you like to include in the tables?', options = game_model['Team'].unique(), key='team_var1')
     elif split_var1 == 'All':
-        team_var1 = game_model.Team.values.tolist()
+        team_var1 = game_model.TC.values.tolist()
     game_model = game_model[game_model['Team'].isin(team_var1)]
-    game_display = game_model.set_index('Team')
+    game_display = game_model.set_index('Team Date')
     st.dataframe(game_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
     st.download_button(
         label="Export Game Model",