Update src/streamlit_app.py

src/streamlit_app.py

@@ -1343,8 +1343,12 @@ elif len(st.session_state.batch_results) > 0:
             except (AttributeError, ValueError):
                 due_date_str = ""
 
+        # Get the current line items from session state (not the form-scoped edited_df)
+        items_state_key = f"items_df_{selected_hash}"
+        current_items_df = st.session_state.get(items_state_key, pd.DataFrame())
+        
         # Calculate totals from line items (always, for both save and download)
-        line_items_list = edited_df.to_dict('records')
+        line_items_list = current_items_df.to_dict('records')
         calculated_subtotal = sum(clean_float(item.get('Amount', 0)) for item in line_items_list)
         calculated_total_tax = sum(clean_float(item.get('Tax', 0)) for item in line_items_list)
         calculated_total = sum(clean_float(item.get('Line Total', 0)) for item in line_items_list)
@@ -1388,8 +1392,16 @@ elif len(st.session_state.batch_results) > 0:
             # Save to batch_results (this persists the data)
             st.session_state.batch_results[selected_hash]["edited_data"] = updated
             
+            # CRITICAL: Clear items_df from session state so it rebuilds from saved data on next rerun
+            items_state_key = f"items_df_{selected_hash}"
+            if items_state_key in st.session_state:
+                del st.session_state[items_state_key]
+            
             # Show success message
             st.success("✅ Saved")
+            
+            # Rerun to reload the form with saved data
+            st.rerun()
 
         # Per-file CSV download (ALWAYS visible, uses current edited values)
         download_data = {