file upload gallery

app.py

@@ -1355,14 +1355,20 @@ def content_tab_settings():
         #                                                                      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
+
+        # Initialize API key states if not already in session_state
+        for key_state, env_var in st.session_state['api_keys'].items():
+            if key_state not in st.session_state:
+                st.session_state[key_state] = os.getenv(env_var) is not None
+
+        # Now check which LLM versions are available based on the available API keys
         available_versions = []
         for api_name, versions in st.session_state['LLM_VERSIONS'].items():
-            print(f'has_key_{api_name.lower().replace(" ", "_")}')
-            if st.session_state.get(f'has_key_{api_name.lower().replace(" ", "_")}', False):
+            key_state = f'has_key_{api_name.lower().replace(" ", "_")}'
+            if st.session_state.get(key_state, False):
                 available_versions.extend(versions)
-                print(available_versions)
 
-        # Show available LLM versions in a select box
+        # Show available LLM versions in a select box if there are any
         st.subheader('LLM Version')
         if available_versions:
             # Get current selection from session_state, defaulting to the first available version
@@ -1747,6 +1753,13 @@ if 'LLM_VERSIONS' not in st.session_state:
         'Azure API': ["Azure GPT 4", "Azure GPT 3.5"],
         'Palm API': ["PaLM 2"]
     }
+if 'api_keys' not in st.session_state:
+    st.session_state['api_keys'] = {
+            'has_key_openai': 'OPENAI_API_KEY',
+            'has_key_google_OCR': 'GOOGLE_APPLICATION_CREDENTIALS',
+            'has_key_palm2': 'PALM_API_KEY',
+            'has_key_azure': 'AZURE_API_KEY'
+        }