Update src/streamlit_app.py

src/streamlit_app.py

@@ -268,22 +268,22 @@ STRICT POLICY RULES (apply exactly, do not deviate):
    - In this JSON, output all values as strings. If a field is not present or cannot be determined with high confidence, output "" (empty string). Do not use null, 0, or placeholders.
 2) Currency selection (multi-currency invoices)
    - If multiple currencies are shown, ALWAYS choose the recipient/customer currency for all monetary fields in items and summary.
-   - Do NOT perform FX conversion. Select the column/figures that are explicitly in the recipient’s currency.
+   - Do NOT perform FX conversion. Select the column/figures that are explicitly in the recipient's currency.
    - For "summary.currency", prefer the printed 3-letter code (e.g., USD, EUR, INR). If only an unambiguous symbol is present, map it (₹→INR, €→EUR, $→USD when clearly USD). If ambiguous, leave "".
-3) Tax handling (no rounding of rates; don’t recompute given totals)
+3) Tax handling (no rounding of rates; don't recompute given totals)
    - Do NOT round tax percentages. Use the original precision for any calculations; keep the printed formatting for "summary.tax_rate".
-   - If a TOTAL tax amount is explicitly printed on the invoice (e.g., “Tax”, “VAT”, “IGST”, “Total Tax”), TREAT IT AS AUTHORITATIVE. Do NOT recompute a new total.
+   - If a TOTAL tax amount is explicitly printed on the invoice (e.g., "Tax", "VAT", "IGST", "Total Tax"), TREAT IT AS AUTHORITATIVE. Do NOT recompute a new total.
      a) If per-line tax amounts are printed, copy them directly.
-     b) If per-line tax amounts are not printed, allocate the printed TOTAL tax proportionally across line items by each line’s net amount (quantity * unit_price − discount). Use precise arithmetic; ensure the sum of allocated per-line taxes equals the printed TOTAL tax (adjust the last cent minimally if required).
+     b) If per-line tax amounts are not printed, allocate the printed TOTAL tax proportionally across line items by each line's net amount (quantity * unit_price − discount). Use precise arithmetic; ensure the sum of allocated per-line taxes equals the printed TOTAL tax (adjust the last cent minimally if required).
    - If NO total tax amount is printed but a tax rate is printed, compute per-line tax as: tax = (quantity * unit_price − discount) × (exact, unrounded tax rate). Then set "summary.tax_amount" = sum of per-line taxes.
    - "items[].amount" is the pre-tax line amount AFTER discount. "items[].Line_total" = amount + tax.
 4) Discounts
    - If discounts are present (per-line or overall), compute tax on the discounted base: (quantity * unit_price − discount). Never compute tax on the undiscounted amount.
 5) Due date calculation from payment terms
-   - Preserve the invoice’s original date format for both "invoice_date" and "due_date".
+   - Preserve the invoice's original date format for both "invoice_date" and "due_date".
    - If explicit due date is printed, use it as "due_date".
    - If payment terms specify Net X (e.g., Net 30), set due_date = invoice_date + X days (same format as invoice_date).
-   - If terms say “upon receipt”, “upon publication”, or equivalent, due_date = invoice_date.
+   - If terms say "upon receipt", "upon publication", or equivalent, due_date = invoice_date.
    - If both a printed due date and terms exist and they conflict, prefer the printed due date.
 6) Items array
    - Include every visible line item. Preserve multi-line descriptions using literal "\\n" where line breaks exist.
@@ -1125,17 +1125,7 @@ elif len(st.session_state.batch_results) > 0:
     with frame_right:
         st.subheader(f"Editable Invoice: {current['file_name']}")
 
-        # Quick swap outside the form (one clean rerun)
-        swap_cols = st.columns([1,1,2])
-        with swap_cols[0]:
-            if st.button("⇄ Swap Sender ↔ Recipient", key=f"swap_{selected_hash}"):
-                sn = f"Sender Name_{selected_hash}"
-                rn = f"Recipient Name_{selected_hash}"
-                sa = f"Sender Address_{selected_hash}"
-                ra = f"Recipient Address_{selected_hash}"
-                st.session_state[sn], st.session_state[rn] = st.session_state[rn], st.session_state[sn]
-                st.session_state[sa], st.session_state[ra] = st.session_state[ra], st.session_state[sa]
-                st.rerun()
+        # SWAP BUTTON REMOVED COMPLETELY
 
         # ----------------- FORM START -----------------
         with st.form(key=f"edit_form_{selected_hash}", clear_on_submit=False):
@@ -1300,9 +1290,9 @@ elif len(st.session_state.batch_results) > 0:
 
             st.session_state.batch_results[selected_hash]["edited_data"] = updated
             st.success(f"✅ Saved: {current['file_name']} | Updated totals: Subtotal=${calculated_subtotal:,.2f}, Tax=${calculated_total_tax:,.2f}, Total=${calculated_total:,.2f}")
-            st.rerun()  # Force rerun to show updated totals in the form fields
+            # REMOVED st.rerun() - This was causing the download button to disappear
 
-        # Per-file CSV download (uses the current editor contents even if not saved)
+        # Per-file CSV download (MOVED OUTSIDE THE FORM - Always visible)
         d_currency = st.session_state.get(f"Currency_{selected_hash}", 'USD')
         if d_currency == 'Other':
             d_currency = st.session_state.get(f"Currency_Custom_{selected_hash}", '')
@@ -1372,4 +1362,4 @@ else:
     with frame_left:
         st.caption("Ready when you are.")
     with frame_right:
-        st.caption("Preview & editor will appear here after extraction.")
+        st.caption("Preview & editor will appear here after extraction.")