update

app.py

@@ -804,7 +804,7 @@ def create_download_button(zip_filepath):
     with open(zip_filepath, 'rb') as f:
         bytes_io = BytesIO(f.read())
     st.download_button(
-        label=f"Download Results for{st.session_state['processing_add_on']}",
+        label=f"Download Results for{st.session_state['processing_add_on']}",type='primary',
         data=bytes_io,
         file_name=os.path.basename(zip_filepath),
         mime='application/zip'
@@ -839,7 +839,7 @@ def btn_load_prompt(selected_yaml_file, dir_prompt):
         }
 
 def refresh():
-    st.write('')
+    st.rerun()
 
 def upload_local_prompt_to_server(dir_prompt):
     uploaded_file = st.file_uploader("Upload a custom prompt file", type=['yaml'])
@@ -1260,7 +1260,7 @@ def content_header():
             st.session_state['processing_add_on'] = f" {determine_n_images()} Images"
         else:
             st.session_state['processing_add_on'] = ''
-
+        print(f"HERE --- {st.session_state['processing_add_on']}")
         
         if check_if_usable():
             if st.button(f"Start Processing{st.session_state['processing_add_on']}", type='primary'):
@@ -1347,6 +1347,12 @@ def content_header():
         #         for message in info:
         #             st.error(message)
 
+def clear_image_gallery():
+    delete_directory(st.session_state['dir_uploaded_images'])
+    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'])
+
 def content_tab_settings():
     col_project_1, col_project_2, col_project_3 = st.columns([2,2,2])
 
@@ -1416,10 +1422,8 @@ def content_tab_settings():
         uploaded_files = st.file_uploader("Upload Images", type=['jpg', 'jpeg'], accept_multiple_files=True)
         if uploaded_files:
             # Clear input image gallery and input list
-            delete_directory(st.session_state['dir_uploaded_images'])
-            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'])
+            clear_image_gallery()
+
             # Process the new iamges
             for uploaded_file in uploaded_files:
                 file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file)
@@ -1737,10 +1741,10 @@ if 'proceed_to_private' not in st.session_state:
 if 'dir_uploaded_images' not in st.session_state:
     st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
     validate_dir(os.path.join(st.session_state.dir_home,'uploads'))
-# if 'private_file' not in st.session_state:
-#     st.session_state.private_file = does_private_file_exist()
-#     if st.session_state.private_file:
-#         st.session_state.proceed_to_main = True
+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'))
+
 
 # Initialize session_state variables if they don't exist
 if 'prompt_info' not in st.session_state: