file upload gallery

app.py

@@ -1354,17 +1354,11 @@ def content_tab_settings():
         #                                                                      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
 
-        # 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
+        # Determine the available versions based on the API keys present
         available_versions = []
         for api_name, versions in st.session_state['LLM_VERSIONS'].items():
-            key_state = f'has_key_{api_name.lower().replace(" ", "_")}'
+            key_state = st.session_state['api_name_to_key_state'][api_name]
             if st.session_state.get(key_state, False):
                 available_versions.extend(versions)
 
@@ -1739,29 +1733,24 @@ if 'show_prompt_name_w' not in st.session_state:
 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'] = {
         'OpenAI API': ["GPT 4", "GPT 3.5"],
         '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'
-        }
-    
-    
+if 'api_name_to_key_state ' not in st.session_state:
+    st.session_state['api_name_to_key_state'] = {
+        'OpenAI API': 'has_key_openai',
+        'Google OCR API': 'has_key_google_OCR',
+        'Palm API': 'has_key_palm2',
+        'Azure API': 'has_key_azure'
+    }
+        
+ # Initialize API key states if not already in session_state
+for api_name, key_state in st.session_state['api_name_to_key_state'].items():
+    if key_state not in st.session_state:
+        st.session_state[key_state] = False   
 
 # if not st.session_state.private_file:
 #     create_private_file()