update

app.py

@@ -1347,6 +1347,12 @@ def clear_image_gallery():
     validate_dir(st.session_state['dir_uploaded_images'])
     validate_dir(st.session_state['dir_uploaded_images_small'])
 
+def use_test_image():
+    st.session_state.config['leafmachine']['project']['dir_images_local'] = os.path.join(st.session_state.dir_home,'demo','demo_images')
+    st.info(f"Processing images from {os.path.join(st.session_state.dir_home,'demo','demo_images')}")
+    n_images = len([f for f in os.listdir(st.session_state.config['leafmachine']['project']['dir_images_local']) if os.path.isfile(os.path.join(st.session_state.config['leafmachine']['project']['dir_images_local'], f))])
+    st.session_state['processing_add_on'] = f" {n_images} Images"
+                 
 def content_tab_settings():
     col_project_1, col_project_2, col_project_3 = st.columns([2,2,2])
 
@@ -1363,8 +1369,6 @@ def content_tab_settings():
     st.header('OCR Overlay Image')    
     col_ocr_1, col_ocr_2 = st.columns([4,4])  
 
-
-    os.path.join(st.session_state.dir_home, )
     ### Project
     with col_project_1:
         st.subheader('Run name')
@@ -1429,10 +1433,13 @@ def content_tab_settings():
                 st.session_state['input_list_small'].append(file_path_small)
                 print(uploaded_file.name)
 
-            n_images = len([f for f in os.listdir(st.session_state['dir_uploaded_images']) if os.path.isfile(os.path.join(st.session_state['dir_uploaded_images'], f))])
+            # Set the local images to the uploaded images
+            st.session_state.config['leafmachine']['project']['dir_images_local'] = st.session_state['dir_uploaded_images']
+
+            n_images = len([f for f in os.listdir(st.session_state.config['leafmachine']['project']['dir_images_local']) if os.path.isfile(os.path.join(st.session_state.config['leafmachine']['project']['dir_images_local'], f))])
             st.session_state['processing_add_on'] = f" {n_images} Images"
-            print(f"HERE --- {st.session_state['processing_add_on']}")
             uploaded_files = None
+        st.button("Use Test Image",help="This will clear any uploaded images and load the 1 provided test image.",on_click=use_test_image)
 
     with col_local_2:
         if st.session_state['input_list_small']: