file upload gallery

app.py

@@ -1218,6 +1218,17 @@ def show_header_welcome():
     st.session_state.logo = Image.open(st.session_state.logo_path)
     st.image(st.session_state.logo, width=250)
 
+def determine_n_images():
+    try:
+        # Check if 'dir_uploaded_images' key exists and it is not empty
+        if 'dir_uploaded_images' in st and st['dir_uploaded_images']:
+            dir_path = st['dir_uploaded_images']  # This would be the path to the directory
+            return len([f for f in os.listdir(dir_path) if os.path.isfile(os.path.join(dir_path, f))])
+        else:
+            return None
+    except:
+        return None
+
 def content_header():
     col_run_1, col_run_2, col_run_3, col_run_4 = st.columns([2,2,2,2])
 
@@ -1244,8 +1255,15 @@ def content_header():
         show_header_welcome()
         st.subheader('Run VoucherVision')
         N_STEPS = 6
+        
+        if determine_n_images():
+            st.session_state['processing_add_on'] = f" {determine_n_images()} Images"
+        else:
+            st.session_state['processing_add_on'] = ''
+
+        
         if check_if_usable():
-            if st.button("Start Processing", type='primary'):
+            if st.button(f"Start Processing{st.session_state['processing_add_on']}", type='primary'):
             
                 # First, write the config file.
                 write_config_file(st.session_state.config, st.session_state.dir_home, filename="VoucherVision.yaml")
@@ -1399,8 +1417,9 @@ def content_tab_settings():
         if uploaded_files:
             # Clear input image gallery and input list
             delete_directory(st.session_state['dir_uploaded_images'])
-            st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
+            delete_directory(st.session_state['dir_uploaded_images_small'])
             validate_dir(st.session_state['dir_uploaded_images'])
+            validate_dir(st.session_state['dir_uploaded_images_small'])
             # Process the new iamges
             for uploaded_file in uploaded_files:
                 file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file)
@@ -1744,6 +1763,8 @@ if 'show_prompt_name_w' not in st.session_state:
     st.session_state['show_prompt_name_w'] = None
 if 'user_clicked_load_prompt_yaml' not in st.session_state:
     st.session_state['user_clicked_load_prompt_yaml'] = None
+if 'processing_add_on' not in st.session_state:
+    st.session_state['processing_add_on'] = ''
 
 if 'LLM_VERSIONS' not in st.session_state:
     st.session_state['LLM_VERSIONS'] = {