Update app.py

app.py

@@ -279,6 +279,44 @@ with tab2:
                 if st.button('Make swaps'):
                     with hold_container:
                         st.session_state.display_portfolio.replace(player_check, player_swap, inplace=True)
+                        split_portfolio = st.session_state.display_portfolio
+                        
+                        split_portfolio['Salary'] = sum([split_portfolio['PG'].map(player_salary_dict),
+                                  split_portfolio['SG'].map(player_salary_dict),
+                                  split_portfolio['SF'].map(player_salary_dict),
+                                  split_portfolio['PF'].map(player_salary_dict),
+                                  split_portfolio['C'].map(player_salary_dict),
+                                  split_portfolio['G'].map(player_salary_dict),
+                                  split_portfolio['F'].map(player_salary_dict),
+                                  split_portfolio['UTIL'].map(player_salary_dict)])
+                        
+                        split_portfolio['Projection'] = sum([split_portfolio['PG'].map(player_proj_dict),
+                                  split_portfolio['SG'].map(player_proj_dict),
+                                  split_portfolio['SF'].map(player_proj_dict),
+                                  split_portfolio['PF'].map(player_proj_dict),
+                                  split_portfolio['C'].map(player_proj_dict),
+                                  split_portfolio['G'].map(player_proj_dict),
+                                  split_portfolio['F'].map(player_proj_dict),
+                                  split_portfolio['UTIL'].map(player_proj_dict)])
+                        
+                        split_portfolio['Ownership'] = sum([split_portfolio['PG'].map(player_own_dict),
+                                  split_portfolio['SG'].map(player_own_dict),
+                                  split_portfolio['SF'].map(player_own_dict),
+                                  split_portfolio['PF'].map(player_own_dict),
+                                  split_portfolio['C'].map(player_own_dict),
+                                  split_portfolio['G'].map(player_own_dict),
+                                  split_portfolio['F'].map(player_own_dict),
+                                  split_portfolio['UTIL'].map(player_own_dict)])
+                             
+                        st.session_state.display_portfolio = split_portfolio[['PG', 'SG', 'SF', 'PF', 'C', 'G', 'F', 'UTIL', 'Salary', 'Projection', 'Ownership']]
+                        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:8].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'] / len(st.session_state.display_portfolio)
+                        st.session_state.player_freq = st.session_state.player_freq.set_index('Player')
+                        
+                        gc.collect()
                 
     with st.container():
         if 'final_Proj' in st.session_state: