Enhance export options in app.py: introduce a selection for 'Simple' or 'Advanced' download types, allowing users to choose between direct portfolio export or customized stack adjustments, improving flexibility and user experience during data export.

app.py

@@ -878,67 +878,71 @@ with tab3:
             st.session_state['portfolio'] = st.session_state['portfolio'].sort_values(by='median', ascending=False)
             with st.expander("Download options"):
                 if stack_dict is not None:
-                    with st.form(key='stack_form'):
-                        st.subheader("Stack Count Adjustments")
-                        st.info("This allows you to fine tune the stacks that you wish to export. If you want to make sure you don't export any of a specific stack you can 0 it out.")
-                        # Create a container for stack value inputs
-                        sort_container = st.container()
-                        with sort_container:
-                            sort_var = st.selectbox("Sort export portfolio by:", options=['median', 'Lineup Edge', 'Own'])
-                        
-                        # Get unique stack values
-                        unique_stacks = sorted(list(set(stack_dict.values())))
-                        
-                        # Create a dictionary to store stack multipliers
-                        if 'stack_multipliers' not in st.session_state:
-                            st.session_state.stack_multipliers = {stack: 0.0 for stack in unique_stacks}
-                        
-                        # Create columns for the stack inputs
-                        num_cols = 6  # Number of columns to display
-                        for i in range(0, len(unique_stacks), num_cols):
-                            cols = st.columns(num_cols)
-                            for j, stack in enumerate(unique_stacks[i:i+num_cols]):
-                                with cols[j]:
-                                    # Create a unique key for each number input
-                                    key = f"stack_count_{stack}"
-                                    # Get the current count of this stack in the portfolio
-                                    current_stack_count = len(st.session_state['portfolio'][st.session_state['portfolio']['Stack'] == stack])
-                                    # Create number input with current value and max value based on actual count
-                                    st.session_state.stack_multipliers[stack] = st.number_input(
-                                        f"{stack} count",
-                                        min_value=0.0,
-                                        max_value=float(current_stack_count),
-                                        value=0.0,
-                                        step=1.0,
-                                        key=key
-                                    )
-                        
-                        portfolio_copy = st.session_state['portfolio'].copy()
-                        
-                        submitted = st.form_submit_button("Submit")
-                        if submitted:
-                            # Create a list to store selected rows
-                            selected_rows = []
+                    download_type = st.selectbox("Simple or Advanced Download?", options=['Simple', 'Advanced'], key='download_choice')
+                    if download_type == 'Simple':
+                        st.session_state['export_file'] = st.session_state['portfolio'].copy()
+                    else:
+                        with st.form(key='stack_form'):
+                            st.subheader("Stack Count Adjustments")
+                            st.info("This allows you to fine tune the stacks that you wish to export. If you want to make sure you don't export any of a specific stack you can 0 it out.")
+                            # Create a container for stack value inputs
+                            sort_container = st.container()
+                            with sort_container:
+                                sort_var = st.selectbox("Sort export portfolio by:", options=['median', 'Lineup Edge', 'Own'])
                             
-                            # For each stack, select the top N rows based on the count value
-                            for stack in unique_stacks:
-                                if stack in st.session_state.stack_multipliers:
-                                    count = int(st.session_state.stack_multipliers[stack])
-                                    # Get rows for this stack
-                                    stack_rows = portfolio_copy[portfolio_copy['Stack'] == stack]
-                                    # Sort by median and take top N rows
-                                    top_rows = stack_rows.nlargest(count, sort_var)
-                                    selected_rows.append(top_rows)
-
-                            # Combine all selected rows
-                            portfolio_concat = pd.concat(selected_rows)
+                            # Get unique stack values
+                            unique_stacks = sorted(list(set(stack_dict.values())))
+                            
+                            # Create a dictionary to store stack multipliers
+                            if 'stack_multipliers' not in st.session_state:
+                                st.session_state.stack_multipliers = {stack: 0.0 for stack in unique_stacks}
+                            
+                            # Create columns for the stack inputs
+                            num_cols = 6  # Number of columns to display
+                            for i in range(0, len(unique_stacks), num_cols):
+                                cols = st.columns(num_cols)
+                                for j, stack in enumerate(unique_stacks[i:i+num_cols]):
+                                    with cols[j]:
+                                        # Create a unique key for each number input
+                                        key = f"stack_count_{stack}"
+                                        # Get the current count of this stack in the portfolio
+                                        current_stack_count = len(st.session_state['portfolio'][st.session_state['portfolio']['Stack'] == stack])
+                                        # Create number input with current value and max value based on actual count
+                                        st.session_state.stack_multipliers[stack] = st.number_input(
+                                            f"{stack} count",
+                                            min_value=0.0,
+                                            max_value=float(current_stack_count),
+                                            value=0.0,
+                                            step=1.0,
+                                            key=key
+                                        )
                             
-                            # Update export_file with filtered data
-                            st.session_state['export_file'] = portfolio_concat.copy()
-                            for col in st.session_state['export_file'].columns:
-                                if col not in excluded_cols:
-                                    st.session_state['export_file'][col] = st.session_state['export_file'][col].map(st.session_state['export_dict'])
-                            st.write('Export portfolio updated!')
+                            portfolio_copy = st.session_state['portfolio'].copy()
+                            
+                            submitted = st.form_submit_button("Submit")
+                            if submitted:
+                                # Create a list to store selected rows
+                                selected_rows = []
+                                
+                                # For each stack, select the top N rows based on the count value
+                                for stack in unique_stacks:
+                                    if stack in st.session_state.stack_multipliers:
+                                        count = int(st.session_state.stack_multipliers[stack])
+                                        # Get rows for this stack
+                                        stack_rows = portfolio_copy[portfolio_copy['Stack'] == stack]
+                                        # Sort by median and take top N rows
+                                        top_rows = stack_rows.nlargest(count, sort_var)
+                                        selected_rows.append(top_rows)
+
+                                # Combine all selected rows
+                                portfolio_concat = pd.concat(selected_rows)
+                                
+                                # Update export_file with filtered data
+                                st.session_state['export_file'] = portfolio_concat.copy()
+                                for col in st.session_state['export_file'].columns:
+                                    if col not in excluded_cols:
+                                        st.session_state['export_file'][col] = st.session_state['export_file'][col].map(st.session_state['export_dict'])
+                                st.write('Export portfolio updated!')
                 else:
                     st.session_state['export_file'] = st.session_state['portfolio'].copy()
             if 'export_file' in st.session_state: