file upload gallery

app.py

@@ -18,6 +18,7 @@ from vouchervision.vouchervision_main import voucher_vision, voucher_vision_OCR_
 from vouchervision.general_utils import test_GPU, get_cfg_from_full_path, summarize_expense_report, create_google_ocr_yaml_config, validate_dir
 
 PROMPTS_THAT_NEED_DOMAIN_KNOWLEDGE = ["Version 1","Version 1 PaLM 2"]
+LLM_VERSIONS = ["GPT 4", "GPT 3.5", "Azure GPT 4", "Azure GPT 3.5", "PaLM 2"]
 COLORS_EXPENSE_REPORT = {
         'GPT_4': '#8fff66',    # Bright Green
         'GPT_3_5': '#006400',  # Dark Green
@@ -567,7 +568,8 @@ def display_image_gallery():
         # Loop through each image in the input list
         for image_path in st.session_state['input_list']:
             # Convert the image to base64 (assuming you have this utility function)
-            base64_image = image_to_base64(image_path)
+            img_file = Image.open(image_path)
+            base64_image = image_to_base64(img_file)
 
             # Embed the image with the determined width in the custom div
             img_html = f"""
@@ -1107,7 +1109,8 @@ def content_tab_settings():
     ### Project
     with col_project_1:
         st.subheader('Run name')
-        st.session_state.config['leafmachine']['project']['run_name'] = st.text_input("Run name", st.session_state.config['leafmachine']['project'].get('run_name', ''),label_visibility='collapsed')
+        st.session_state.config['leafmachine']['project']['run_name'] = st.text_input("Run name", st.session_state.config['leafmachine']['project'].get('run_name', ''),
+                                                                                      label_visibility='collapsed')
         # st.session_state.config['leafmachine']['project']['dir_output'] = st.text_input("Output directory", st.session_state.config['leafmachine']['project'].get('dir_output', ''))
         st.write("Run name will be the name of the final zipped folder.")
 
@@ -1116,7 +1119,9 @@ def content_tab_settings():
         st.session_state.config['leafmachine']['project']['dir_images_local'] = st.session_state['dir_uploaded_images'] #st.text_input("Input images directory", st.session_state.config['leafmachine']['project'].get('dir_images_local', ''))
         # st.session_state.config['leafmachine']['project']['continue_run_from_partial_xlsx'] = st.text_input("Continue run from partially completed project XLSX", st.session_state.config['leafmachine']['project'].get('continue_run_from_partial_xlsx', ''), disabled=True)
         st.subheader('LLM Version')
-        st.session_state.config['leafmachine']['LLM_version'] = st.selectbox("LLM version", ["GPT 4", "GPT 3.5", "Azure GPT 4", "Azure GPT 3.5", "PaLM 2"], index=["GPT 4", "GPT 3.5", "Azure GPT 4", "Azure GPT 3.5", "PaLM 2"].index(st.session_state.config['leafmachine'].get('LLM_version', 'Azure GPT 4')))
+        st.session_state.config['leafmachine']['LLM_version'] = st.selectbox("LLM version", LLM_VERSIONS, 
+                                                                             index=LLM_VERSIONS.index(st.session_state.config['leafmachine'].get('LLM_version', 'Azure GPT 4')),
+                                                                             label_visibility='collapsed')
         st.markdown("""***Note:*** GPT-4 is significantly more expensive than GPT-3.5  """)
     
     ### Prompt Version