Update src/streamlit_app.py

src/streamlit_app.py

@@ -1214,25 +1214,32 @@ elif len(st.session_state.batch_results) > 0:
                 st.text_input("Branch", key=f"Bank_bank_branch_{selected_hash}")
 
             with tabs[3]:
-                # Build base DF from current edited_data (not raw mapped) so it's always what the user last saved
-                item_rows = form_data.get('Itemized Data', []) or []
-                normalized = []
-                for it in item_rows:
-                    if not isinstance(it, dict):
-                        it = {}
-                    normalized.append({
-                        "Description": it.get("Description", it.get("Item Description", "")),
-                        "Quantity": it.get("Quantity", it.get("Item Quantity", 0)),
-                        "Unit Price": it.get("Unit Price", it.get("Item Unit Price", 0.0)),
-                        "Amount": it.get("Amount", it.get("Item Amount", 0.0)),
-                        "IO Number/Cost Centre": it.get("IO Number/Cost Centre", ""),  # New column
-                        "Tax": it.get("Tax", it.get("Item Tax", 0.0)),
-                        "Line Total": it.get("Line Total", it.get("Item Line Total", 0.0)),
-                    })
-
-                items_df = pd.DataFrame(normalized) if normalized else pd.DataFrame(
-                    columns=["Description", "Quantity", "Unit Price", "Amount", "IO Number/Cost Centre", "Tax", "Line Total"]
-                )
+                # Initialize line items DataFrame in session state ONLY ONCE (preserves edits)
+                items_state_key = f"items_df_{selected_hash}"
+                
+                if items_state_key not in st.session_state:
+                    # Build base DF from current edited_data only on first load
+                    item_rows = form_data.get('Itemized Data', []) or []
+                    normalized = []
+                    for it in item_rows:
+                        if not isinstance(it, dict):
+                            it = {}
+                        normalized.append({
+                            "Description": it.get("Description", it.get("Item Description", "")),
+                            "Quantity": it.get("Quantity", it.get("Item Quantity", 0)),
+                            "Unit Price": it.get("Unit Price", it.get("Item Unit Price", 0.0)),
+                            "Amount": it.get("Amount", it.get("Item Amount", 0.0)),
+                            "IO Number/Cost Centre": it.get("IO Number/Cost Centre", ""),  # New column
+                            "Tax": it.get("Tax", it.get("Item Tax", 0.0)),
+                            "Line Total": it.get("Line Total", it.get("Item Line Total", 0.0)),
+                        })
+
+                    st.session_state[items_state_key] = pd.DataFrame(normalized) if normalized else pd.DataFrame(
+                        columns=["Description", "Quantity", "Unit Price", "Amount", "IO Number/Cost Centre", "Tax", "Line Total"]
+                    )
+                
+                # Use the stored DataFrame
+                items_df = st.session_state[items_state_key]
 
                 # Configure column widths to make Description wider and avoid horizontal scrolling
                 column_config = {
@@ -1279,6 +1286,9 @@ elif len(st.session_state.batch_results) > 0:
                     height=DATA_EDITOR_HEIGHT - 50,  # Reduce height slightly for totals below
                     column_config=column_config,
                 )
+                
+                # Update session state with edited DataFrame (preserves changes across reruns)
+                st.session_state[items_state_key] = edited_df
 
                 # Display non-editable totals row immediately below (looks integrated)
                 if len(edited_df) > 0: