update

app.py

@@ -115,7 +115,10 @@ def update_displayed_quadrilateral(index, point_combinations, base_image_path):
     # Display the image with the selected quadrilateral
     display_image_with_quadrilateral(base_image, quad_points)
 
-def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper):
+def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper, loc):
+    with loc:
+        btn_back, btn_next = st.columns([2,2])
+
     img = cv2.imread(image_path)
     
     # Check if image is valid
@@ -174,15 +177,17 @@ def process_image(image_path, flag_lower, flag_upper, plant_lower, plant_upper):
         # Show initial quadrilateral
         centroids = update_displayed_quadrilateral(selected_quad_index)
 
-        # Button to go to the previous quadrilateral
-        if st.button('Previous'):
-            selected_quad_index = max(selected_quad_index - 1, 0)
-            centroids = update_displayed_quadrilateral(selected_quad_index)
-
-        # Button to go to the next quadrilateral
-        if st.button('Next'):
-            selected_quad_index = min(selected_quad_index + 1, len(point_combinations) - 1)
-            centroids = update_displayed_quadrilateral(selected_quad_index)
+        with btn_back:
+            # Button to go to the previous quadrilateral
+            if st.button('Previous'):
+                selected_quad_index = max(selected_quad_index - 1, 0)
+                centroids = update_displayed_quadrilateral(selected_quad_index)
+
+        with btn_next:
+            # Button to go to the next quadrilateral
+            if st.button('Next'):
+                selected_quad_index = min(selected_quad_index + 1, len(point_combinations) - 1)
+                centroids = update_displayed_quadrilateral(selected_quad_index)
 
     # If there are exactly 4 largest contours, proceed with existing logic
     elif len(significant_contours) == 4: