update

app.py

@@ -275,6 +275,13 @@ def clear_image_gallery():
     validate_dir(st.session_state['dir_uploaded_images'])
     validate_dir(st.session_state['dir_uploaded_images_small'])
 
+def reset_demo_images():
+    st.session_state['dir_input'] = os.path.join(st.session_state['dir_home'],"demo")
+    st.session_state['input_list'] = [os.path.join(st.session_state['dir_input'], fname) for fname in os.listdir(st.session_state['dir_input']) if fname.endswith(('.jpg', '.jpeg', '.png'))]
+    n_images = len([f for f in os.listdir(st.session_state['dir_input']) if os.path.isfile(os.path.join(st.session_state['dir_input'], f))])
+    st.session_state['processing_add_on'] = f" {n_images} Images"
+    st.session_state['uploader_idk'] += 1
+
 def main():
     _, R_coverage, R_plot_area_cm2, R_save = st.columns([5,2,2,2])
     img_gallery, img_main, img_seg, img_green, img_warp = st.columns([1,4,2,2,2])
@@ -309,8 +316,7 @@ def main():
         st.info(f"Processing **{n_images}** images from {st.session_state['dir_input']}")
         
     if st.session_state['dir_input'] is None:
-        st.session_state['dir_input'] = os.path.join(st.session_state['dir_home'],"demo")
-        st.session_state['input_list'] = [os.path.join(st.session_state['dir_input'], fname) for fname in os.listdir(st.session_state['dir_input']) if fname.endswith(('.jpg', '.jpeg', '.png'))]
+        reset_demo_images()
     
     # dir_input = st.text_input("Input directory for images:", value=os.path.join(st.session_state['dir_home'],"demo"))
     dir_output = os.path.join(st.session_state['dir_home'],"demo_out") # st.text_input("Output directory:", value=os.path.join(st.session_state['dir_home'],"demo_out"))
@@ -322,15 +328,18 @@ def main():
     file_name = os.path.join(directory_manager.data, f"{run_name}.csv")
     headers = ['image',"plant_coverage_uncorrected_percen", "plant_coverage_corrected_percent", "plant_area_corrected_cm2"]
     file_exists = os.path.isfile(file_name)
-
+    st.button("Reset Demo Images", on_click=reset_demo_images)
 
         
     if len(st.session_state['input_list']) == 0 or st.session_state['input_list'] is None:
         st.balloons()
+        create_download_button(dir_output, run_name)
+
     else:
         with img_gallery:
             selected_img = image_select("Select an image", st.session_state['input_list'], use_container_width=False)
             base_name = os.path.basename(selected_img)
+            create_download_button(dir_output, run_name)
         
         if selected_img:
 
@@ -405,6 +414,8 @@ def main():
                         # Remove processed image from the list
                         st.session_state['input_list'].remove(selected_img)
                         st.rerun()
+                    create_download_button(dir_output, run_name)
+
 
 st.set_page_config(layout="wide", page_title='GreenSight')
 
@@ -436,4 +447,6 @@ if 'dir_uploaded_images_small' not in st.session_state:
 
 
 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.")
+st.write("If you exit the session before completing the segmentation of all images, all progress will be lost!")
 main()