update

app.py

@@ -221,8 +221,8 @@ def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper,
             return None, None, None, None, None, None, None, None, None, None
         
         # Initial quadrilateral index
-        selected_quad_index = 0
-        centroids = get_points_from_contours(valid_quadrilaterals[selected_quad_index])
+        st.session_state['selected_quad_index'] = 0
+        centroids = get_points_from_contours(valid_quadrilaterals[st.session_state['selected_quad_index']])
         # Function to update displayed quadrilateral based on selected index
         def update_displayed_quadrilateral(index):
             # Extract the four points of the current quadrilateral
@@ -230,16 +230,16 @@ def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper,
             return centroids
 
         # Show initial quadrilateral
-        centroids = update_displayed_quadrilateral(selected_quad_index)
+        centroids = update_displayed_quadrilateral(st.session_state['selected_quad_index'])
 
         with btn_back:
             if st.button('Previous'):
-                selected_quad_index = max(selected_quad_index - 1, 0)
-                update_displayed_quadrilateral(selected_quad_index)
+                st.session_state['selected_quad_index'] = max(st.session_state['selected_quad_index'] - 1, 0)
+                update_displayed_quadrilateral(st.session_state['selected_quad_index'])
         with btn_next:
             if st.button('Next'):
-                selected_quad_index = min(selected_quad_index + 1, len(valid_quadrilaterals) - 1)
-                update_displayed_quadrilateral(selected_quad_index)
+                st.session_state['selected_quad_index'] = min(st.session_state['selected_quad_index'] + 1, len(valid_quadrilaterals) - 1)
+                update_displayed_quadrilateral(st.session_state['selected_quad_index'])
 
     poly_mask = create_polygon_mask(centroids, flag_mask.shape)
 
@@ -559,6 +559,9 @@ 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 'selected_quad_index' not in st.session_state:
+    st.session_state['selected_quad_index'] = 0
+
 
 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.")