Update app.py

app.py

@@ -322,7 +322,25 @@ with tab2:
                 player_swap = st.selectbox('Select player to swap to:', options = st.session_state.final_Proj['Player'].unique(), key='dk_swap')
                 if st.button('Make swaps'):
                     with hold_container:
-                        st.session_state.display_portfolio.replace(player_check, player_swap, inplace=True)
+                        if pos_var_list == "PG":
+                            st.session_state.display_portfolio['PG'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['G'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['UTIL'].replace(player_check, player_swap, inplace=True)
+                        elif pos_var_list == "SG":
+                            st.session_state.display_portfolio['SG'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['G'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['UTIL'].replace(player_check, player_swap, inplace=True)
+                        elif pos_var_list == "SF":
+                            st.session_state.display_portfolio['SF'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['F'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['UTIL'].replace(player_check, player_swap, inplace=True)
+                        elif pos_var_list == "PF":
+                            st.session_state.display_portfolio['PF'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['F'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['UTIL'].replace(player_check, player_swap, inplace=True)
+                        elif pos_var_list == "C":
+                            st.session_state.display_portfolio['C'].replace(player_check, player_swap, inplace=True)
+                            st.session_state.display_portfolio['UTIL'].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),