Update src/streamlit_app.py

src/streamlit_app.py

@@ -180,7 +180,7 @@ def perform_ocr(image, bbox):
     except Exception as e:
         return f"OCR Error: {str(e)}"
 
-def scale_image_to_fixed_size(image, max_width=800, max_height=1000):
+def scale_image_to_fixed_size(image, max_width=900, max_height=1100):
     """Scale image to fit within max dimensions while maintaining aspect ratio - NO PADDING"""
     # Convert to RGB with proper handling
     if image.mode not in ('RGB', 'RGBA'):
@@ -681,10 +681,8 @@ elif st.session_state.page == 'viewer':
                                 st.session_state.edited_data[selected_file]['gt_parse'] = gt_parse
                                 st.session_state.modified_indices.add(selected_file)
                                 
-                                # After successful OCR, deactivate OCR target to prevent repeated triggers/loops
-                                st.session_state.ocr_active_section = None
-                                st.session_state.ocr_active_field = None
-                                st.session_state.ocr_line_item_row = None
+                                # Keep the OCR field active so user can draw multiple rectangles for the same field
+                                # Field will only change when user clicks a different OCR button
                                 
                                 # Clear canvas for next OCR by bumping canvas_key then rerun
                                 st.session_state.canvas_key += 1