update

app.py

@@ -162,6 +162,7 @@ def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper,
     if len(significant_contours) < 4:
         return None, None, None, None, None, None, None, None, None, None
     elif len(significant_contours) > 4:
+        st.session_state['keep_quad'] = False
         while not st.session_state['keep_quad']:
             with loc:
                 st.warning("Cycle until correct plot bounds are found")
@@ -234,7 +235,7 @@ def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper,
 
         # Sort the centroids
         centroids.sort(key=lambda point: (-math.atan2(point[1] - centroid_y, point[0] - centroid_x)) % (2 * np.pi))
-
+    
     # Create a polygon mask using the sorted centroids
     poly_mask = np.zeros_like(flag_mask)
     cv2.fillPoly(poly_mask, [np.array(centroids)], 255)
@@ -555,6 +556,8 @@ if 'dir_uploaded_images_small' not in st.session_state:
     st.session_state['dir_uploaded_images_small'] = os.path.join(st.session_state['dir_home'],'uploads_small')
     validate_dir(os.path.join(st.session_state['dir_home'],'uploads_small'))
 
+if 'keep_quad' not in st.session_state:
+    st.session_state['keep_quad'] = False
 
 st.title("GreenSight")
 st.write("Simple color segmentation app to estimate the vegetation coverage in a plot. Corners of the plot need to be marked with solid, uniforly colored flags.")