Update src/streamlit_app.py

src/streamlit_app.py

@@ -319,7 +319,6 @@ with col2:
     st.markdown(f"**Current:** <span style='color:{field_color}'>●</span> {display_field_name}", unsafe_allow_html=True)
 
     # ========== ZOOM CONTROLS (NOW ON RHS) ==========
-    st.markdown("---")
     st.markdown("#### 🔍 Zoom")
     
     current_zoom = st.session_state.zoom_values[selected_name]
@@ -371,13 +370,21 @@ with col2:
     st.caption(f"Original: {pil_image.width}×{pil_image.height}")
 
     # ========== OCR & VALUE SECTION ==========
-    st.markdown("---")
     st.markdown("#### ✏️ OCR & Value")
 
     current_rect_orig = st.session_state.field_rects_orig[selected_name].get(storage_field_name)
     current_val = st.session_state.field_values[selected_name].get(storage_field_name, "")
+    value_state_key = f"value_{selected_name}_{storage_field_name}"
+    if value_state_key not in st.session_state:
+        st.session_state[value_state_key] = current_val
     
-    new_val = st.text_area("Value (auto-filled by OCR)", value=current_val, height=80, label_visibility="collapsed", placeholder="Value (auto-filled by OCR)")
+    new_val = st.text_area(
+        "Value (auto-filled by OCR)",
+        key=value_state_key,
+        height=80,
+        label_visibility="collapsed",
+        placeholder="Value (auto-filled by OCR)",
+    )
     
     col_btn1, col_btn2, col_btn3 = st.columns(3)
     
@@ -398,6 +405,7 @@ with col2:
                     text = pytesseract.image_to_string(crop, config="--psm 6").strip()
                     if text:
                         st.session_state.field_values[selected_name][storage_field_name] = text
+                        st.session_state[value_state_key] = text
                         st.success(f"OCR: {text}")
                     else:
                         st.warning("Empty result")
@@ -428,7 +436,6 @@ with col2:
                     st.write(f"  {lif}: {v}")
 
     # ========== EXPORT SECTION ==========
-    st.markdown("---")
     st.markdown("#### 📤 JSONL Export")
 
     # Export ALL labeled remittances
@@ -540,6 +547,8 @@ with col1:
             new_rect_orig = display_to_orig(new_rect_display, scale)
             new_rect_orig["stroke"] = field_color
             st.session_state.field_rects_orig[selected_name][storage_field_name] = new_rect_orig
+            # bump rect version so canvas resets to only newest rect for this field
+            st.session_state.rect_version[selected_name] += 1
             
             # Auto-run OCR
             x1 = max(0, int(new_rect_orig["left"]))
@@ -553,8 +562,14 @@ with col1:
                     text = pytesseract.image_to_string(crop, config="--psm 6").strip()
                     if text:
                         st.session_state.field_values[selected_name][storage_field_name] = text
+                        value_state_key = f"value_{selected_name}_{storage_field_name}"
+                        st.session_state[value_state_key] = text
                         st.toast(f"✅ OCR: {text[:50]}{'...' if len(text) > 50 else ''}")
                     else:
-                        st.toast(f"✅ Rectangle saved (no text detected)")
+                        st.toast("✅ Rectangle saved (no text detected)")
                 except Exception:
-                    st.toast(f"✅ Rectangle saved")
+                    st.toast("✅ Rectangle saved")
+            else:
+                st.toast("✅ Rectangle saved")
+            # force rerun so text area and canvas refresh with latest data
+            st.experimental_rerun()