update

app.py

@@ -178,13 +178,7 @@ def warp_and_display_images(img, centroids, base_name, flag_mask_rgb, plant_mask
     return plant_rgb, plot_rgb, plant_rgb_warp, plant_mask_warp
 
 def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper, loc):
-    # Ensure session_state variables are initialized
-    if 'valid_quadrilaterals' not in st.session_state:
-        st.session_state.valid_quadrilaterals = [perm for perm in itertools.permutations(significant_contours, 4) if is_valid_quadrilateral(get_points_from_contours(perm))]
-    if 'selected_quad_index' not in st.session_state:
-        st.session_state.selected_quad_index = 0
-    if 'centroids' not in st.session_state:
-        st.session_state.centroids = get_points_from_contours(st.session_state.valid_quadrilaterals[0])
+    
 
     with loc:
         btn_back, btn_next = st.columns([2,2])
@@ -227,6 +221,14 @@ def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper,
     # Filter out noise based on a predefined area threshold
     significant_contours = [cnt for cnt in sorted_contours if cv2.contourArea(cnt) > MIN_AREA]
 
+    # Ensure session_state variables are initialized
+    if 'valid_quadrilaterals' not in st.session_state:
+        st.session_state.valid_quadrilaterals = [perm for perm in itertools.permutations(significant_contours, 4) if is_valid_quadrilateral(get_points_from_contours(perm))]
+    if 'selected_quad_index' not in st.session_state:
+        st.session_state.selected_quad_index = 0
+    if 'centroids' not in st.session_state:
+        st.session_state.centroids = get_points_from_contours(st.session_state.valid_quadrilaterals[0])
+
     # Logic to handle cases where there are more than 4 significant contours
     if len(significant_contours) < 4:
         with loc: