Update app.py

app.py

@@ -192,11 +192,17 @@ with tab1:
                                            split_portfolio['UTIL'].map(player_own_dict)])
                                  
                             display_portfolio = split_portfolio[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'UTIL', 'Salary', 'Projection', 'Ownership']]
+                            st.session_state.display_portfolio = display_portfolio
                             hold_portfolio = display_portfolio.sort_values(by='Projection', ascending=False)
+                            
+                            st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:9].values, return_counts=True)),
+                                                        columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
+                            st.session_state.player_freq['Freq'] = st.session_state.player_freq['Freq'].astype(int)
+                            
                             gc.collect() 
                             
 with tab2:
-    col1, col2, col3 = st.columns([2, 6, 2])
+    col1, col2, col3 = st.columns([1, 8, 1])
     with col1:
         hold_container = st.empty()
         if st.button("Load/Reset Data", key='reset1'):
@@ -204,6 +210,9 @@ with tab2:
                   del st.session_state[key]
             display_portfolio = hold_portfolio
             st.session_state.display_portfolio = display_portfolio
+            st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:9].values, return_counts=True)),
+                                        columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
+            st.session_state.player_freq['Freq'] = st.session_state.player_freq['Freq'].astype(int)
         if st.button("Trim Lineups", key='trim1'):
             max_proj = 10000
             max_own = display_portfolio['Ownership'].iloc[0]
@@ -305,10 +314,6 @@ with tab2:
                 final_Proj['Player_swap'] = player_check
                 st.session_state.final_Proj = final_Proj
                 
-                st.session_state.player_freq = pd.DataFrame(np.column_stack(np.unique(st.session_state.display_portfolio.iloc[:,0:9].values, return_counts=True)),
-                                            columns=['Player','Freq']).sort_values('Freq', ascending=False).reset_index(drop=True)
-                st.session_state.player_freq['Freq'] = st.session_state.player_freq['Freq'].astype(int)
-                
                 hold_container = st.empty()
                     
     with col2: