file upload gallery

app.py

@@ -20,7 +20,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"]
+# 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
@@ -609,23 +609,30 @@ def image_to_base64(img):
 #         st.markdown(css, unsafe_allow_html=True)  
 
 def show_available_APIs():
-    has_key_openai = bool(os.getenv('OPENAI_API_KEY')) is not None
-    has_key_google_OCR = bool(os.getenv('GOOGLE_APPLICATION_CREDENTIALS')) is not None
-    has_key_palm2 = bool(os.getenv('PALM_API_KEY')) is not None
-    has_key_azure = bool(os.getenv('AZURE_API_KEY')) is not None
+    st.session_state['has_key_openai'] = os.getenv('OPENAI_API_KEY') is not None
+    st.session_state['has_key_google_OCR'] = os.getenv('GOOGLE_APPLICATION_CREDENTIALS') is not None
+    st.session_state['has_key_palm2'] = os.getenv('PALM_API_KEY') is not None
+    st.session_state['has_key_azure'] = os.getenv('AZURE_API_KEY') is not None
 
     emoji_good = ":heavy_check_mark:"
     emoji_bad = ":x:"
 
     table = {
-        'OpenAI API': emoji_good if has_key_openai else emoji_bad,
-        'Google OCR API': emoji_good if has_key_google_OCR else emoji_bad,
-        'Palm API': emoji_good if has_key_palm2 else emoji_bad,
-        'Azure API': emoji_good if has_key_azure else emoji_bad,
+        'Google OCR API': emoji_good if st.session_state['has_key_google_OCR'] else emoji_bad,
+        'OpenAI API': emoji_good if st.session_state['has_key_openai'] else emoji_bad,
+        'Palm API': emoji_good if st.session_state['has_key_palm2'] else emoji_bad,
+        'Azure API': emoji_good if st.session_state['has_key_azure'] else emoji_bad,
     }
     for api_name, status in table.items():
         st.markdown(f"* {status} {api_name}")
 
+    # Define the available versions based on the API keys
+    st.session_state['LLM_VERSIONS'] = {
+        'OpenAI API': ["GPT 4", "GPT 3.5"],
+        'Azure API': ["Azure GPT 4", "Azure GPT 3.5"],
+        'Palm API': ["PaLM 2"]
+    }
+
 def display_image_gallery():
     # Initialize the container
     con_image = st.empty()
@@ -1346,13 +1353,34 @@ def content_tab_settings():
 
     ### LLM Version
     with col_project_2:
-        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)
+        # LLM_VERSIONS_available =
+        # 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", 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  """)
+        # Now filter the LLM_VERSIONS based on the available API keys
+        available_versions = []
+        for api_name, versions in st.session_state['LLM_VERSIONS'].items():
+            if st.session_state.get(f'has_key_{api_name.lower().replace(" ", "_")}', False):
+                available_versions.extend(versions)
+
+        # Show available LLM versions in a select box
         st.subheader('LLM Version')
-        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  """)
+        if available_versions:
+            # Get current selection from session_state, defaulting to the first available version
+            current_selection = st.session_state.config['leafmachine'].get('LLM_version', available_versions[0])
+            # Update the selection with a selectbox
+            st.session_state.config['leafmachine']['LLM_version'] = st.selectbox(
+                "LLM version", available_versions, 
+                index=available_versions.index(current_selection),
+                label_visibility='collapsed'
+            )
+            st.markdown("""***Note:*** GPT-4 is significantly more expensive than GPT-3.5""")
+        else:
+            st.error("No LLM versions are available due to missing API keys.")
     
     ### Prompt Version
     with col_project_3:
@@ -1709,6 +1737,18 @@ 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 'has_key_openai' not in st.session_state:
+    st.session_state['has_key_openai'] = False
+if 'has_key_google_OCR' not in st.session_state:
+    st.session_state['has_key_google_OCR'] = False
+if 'has_key_palm2' not in st.session_state:
+    st.session_state['has_key_palm2'] = False
+if 'has_key_azure' not in st.session_state:
+    st.session_state['has_key_azure'] = False
+if 'LLM_VERSIONS' not in st.session_state:
+    st.session_state['LLM_VERSIONS'] = None
+    
     
 
 # if not st.session_state.private_file: