Enhance data loading functionality by introducing a flag to track portfolio loading status. This allows for better handling of session state when loading from Paydirt DB, ensuring the portfolio is correctly managed in the session.

app.py

@@ -560,11 +560,13 @@ if selected_tab == 'Data Load':
         st.info("Go ahead and upload a portfolio file here. Only include player columns.")
 
         upload_toggle = st.selectbox("What source are you uploading from?", options=['Paydirt DB', 'SaberSim (Just IDs)', 'Draftkings/Fanduel (Names + IDs)', 'Other (Just Names)'])
+        portfolio_loaded = False
         if upload_toggle == 'Paydirt DB':
             if st.button("Load from Database after inserting site CSV"):
                 if site_var == 'Draftkings':
                     portfolio_load = init_DK_NFL_lineups(type_var, 'Main', 'proj', 50, dk_nfl_showdown_db_translation, 1000, [])
                     portfolio_file = pd.DataFrame(portfolio_load)
+                    portfolio_loaded = True
                     if 'portfolio' in st.session_state:
                         del st.session_state['portfolio']
                     if 'export_portfolio' in st.session_state:
@@ -572,6 +574,7 @@ if selected_tab == 'Data Load':
                 else:
                     portfolio_load = init_FD_NFL_lineups(type_var, 'Main', 'proj', 50, fd_nfl_showdown_db_translation, 1000, [])
                     portfolio_file = pd.DataFrame(portfolio_load)
+                    portfolio_loaded = True
                     if 'portfolio' in st.session_state:
                         del st.session_state['portfolio']
                     if 'export_portfolio' in st.session_state:
@@ -592,7 +595,7 @@ if selected_tab == 'Data Load':
                 del st.session_state['export_portfolio']
 
         if 'portfolio' not in st.session_state:
-            if portfolio_file:
+            if portfolio_file or portfolio_loaded:
                 if upload_toggle == 'Paydirt DB':
                     st.session_state['export_portfolio'] = portfolio_file
                     st.session_state['portfolio'] = portfolio_file
@@ -612,7 +615,7 @@ if selected_tab == 'Data Load':
 
                     st.session_state['portfolio'] = st.session_state['portfolio'].dropna(how='all')
                     st.session_state['portfolio'] = st.session_state['portfolio'].reset_index(drop=True)
-                    
+
                 else:
                     st.session_state['export_portfolio'], st.session_state['portfolio'] = load_file(portfolio_file, site_var, type_var, sport_var, 'portfolio')
                     st.session_state['export_portfolio'] = st.session_state['export_portfolio'].dropna(how='all')