file upload gallery

app.py

@@ -279,20 +279,26 @@ def create_space_saver():
         st.session_state.config['leafmachine']['project']['delete_temps_keep_VVE'] = st.checkbox("Delete Temporary Files (KEEP files required for VoucherVisionEditor)", st.session_state.config['leafmachine']['project'].get('delete_temps_keep_VVE', False))
         st.session_state.config['leafmachine']['project']['delete_all_temps'] = st.checkbox("Keep only the final transcription file", st.session_state.config['leafmachine']['project'].get('delete_all_temps', False),help="*WARNING:* This limits your ability to do quality assurance. This will delete all folders created by VoucherVision, leaving only the `transcription.xlsx` file.")
 
-def save_uploaded_file(directory, img_file):
+def save_uploaded_file(directory, img_file, image=None):
     if not os.path.exists(directory):
         os.makedirs(directory)
     # Assuming the uploaded file is an image
-    with Image.open(img_file) as image:
+    try:
+        with Image.open(img_file) as image:
+            full_path = os.path.join(directory, img_file.name)
+            image.save(full_path, "JPEG")
+        # Return the full path of the saved image
+        return full_path
+    except:
         full_path = os.path.join(directory, img_file.name)
-        image.save(full_path, "JPEG")
-    # Return the full path of the saved image
-    return full_path
+        image.save(img_file, "JPEG")
+        return directory
 
 def delete_directory(dir_path):
     try:
         shutil.rmtree(dir_path)
         st.session_state['input_list'] = []
+        st.session_state['input_list_small'] = []
         st.success(f"Deleted previously uploaded images, making room for new images: {dir_path}")
     except OSError as e:
         st.error(f"Error: {dir_path} : {e.strerror}")
@@ -1192,6 +1198,7 @@ def content_tab_settings():
     ### Input Images Local
     with col_local_1:
         st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
+        st.session_state['dir_uploaded_images_small'] = os.path.join(st.session_state.dir_home,'uploads_small')
         uploaded_files = st.file_uploader("Upload Images", type=['jpg', 'jpeg'], accept_multiple_files=True)
         if uploaded_files:
             # Clear input image gallery and input list
@@ -1201,15 +1208,20 @@ def content_tab_settings():
             # Process the new iamges
             for uploaded_file in uploaded_files:
                 file_path = save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file)
+                img = Image.open(file_path)
+                img.thumbnail((120, 120), Image.Resampling.LANCZOS)  
+                file_path_small = save_uploaded_file(st.session_state['dir_uploaded_images_small'], uploaded_file.name, img)
+
                 st.session_state['input_list'].append(file_path)
+                st.session_state['input_list_small'].append(file_path_small)
                 print(uploaded_file.name)
 
     with col_local_2:
-        if st.session_state['input_list']:
+        if st.session_state['input_list_small']:
             st.subheader('Image Gallery')
-            # selected_img = image_select("Uploaded Images Ready for Transcription", st.session_state['input_list'], use_container_width=False)
+            selected_img = image_select("Uploaded Images Ready for Transcription", st.session_state['input_list_small'], use_container_width=False)
             # st.image(st.session_state['input_list'])
-            display_image_gallery()
+            # display_image_gallery()
             # st.button("Clear Staged Images",on_click=delete_directory, args=[st.session_state['dir_uploaded_images']])
         
     with col_cropped_1:
@@ -1499,6 +1511,8 @@ if 'zip_filepath' not in st.session_state:
     st.session_state['zip_filepath'] = None
 if 'input_list' not in st.session_state:
     st.session_state['input_list'] = []
+if 'input_list_small' not in st.session_state:
+    st.session_state['input_list_small'] = []
     
 
 # if not st.session_state.private_file: