file upload gallery

app.py

@@ -10,6 +10,7 @@ import pandas as pd
 from typing import Union
 from google.oauth2 import service_account
 from streamlit_extras.let_it_rain import rain
+from streamlit_image_select import image_select
 from vouchervision.LeafMachine2_Config_Builder import write_config_file
 from vouchervision.VoucherVision_Config_Builder import build_VV_config, run_demo_tests_GPT, run_demo_tests_Palm , TestOptionsGPT, TestOptionsPalm, check_if_usable, run_api_tests
 from vouchervision.vouchervision_main import voucher_vision, voucher_vision_OCR_test
@@ -287,7 +288,7 @@ def save_uploaded_file(directory, img_file):
 def delete_directory(dir_path):
     try:
         shutil.rmtree(dir_path)
-        st.success(f"Deleted folder: {dir_path}")
+        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}")
 
@@ -1046,7 +1047,7 @@ def content_tab_settings():
 
     st.write("---")
     st.header('Input Images')
-    col_local_1, col_local_2 = st.columns([4,2])              
+    col_local_1, col_local_2, col_local_3 = st.columns([4,2,4])              
 
     # st.write("---")
     # st.header('Modules')
@@ -1089,12 +1090,21 @@ def content_tab_settings():
         st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
         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'])
+            st.session_state['input_list'] = []
             st.session_state['dir_uploaded_images'] = os.path.join(st.session_state.dir_home,'uploads')
             validate_dir(st.session_state['dir_uploaded_images'])
+
+            # Process the new iamges
             for uploaded_file in uploaded_files:
                 print(uploaded_file)
+                st.session_state['input_list'].appened(os.path.join(st.session_state.dir_home,'uploads',uploaded_file))
                 save_uploaded_file(st.session_state['dir_uploaded_images'], uploaded_file)
+
+    with col_local_3:
+        selected_img = image_select("Uploaded Images Ready for Transcription", st.session_state['input_list'], use_container_width=False)
+        st.button("Clear Staged Images",on_click=delete_directory, args=[st.session_state['dir_uploaded_images']])
         
     with col_cropped_1:
         default_crops = st.session_state.config['leafmachine']['cropped_components'].get('save_cropped_annotations', ['leaf_whole'])

requirements.txt

@@ -32,4 +32,5 @@ dask
 psutil
 py-cpuinfo
 azureml-sdk
-azure-identity
+azure-identity
+streamlit-image-select