update set api keys

app.py

@@ -293,7 +293,7 @@ def create_private_file():
         'AZURE_API_TYPE': os.getenv('AZURE_API_TYPE'),
         'AZURE_DEPLOYMENT_NAME': os.getenv('AZURE_DEPLOYMENT_NAME'),
         'GOOGLE_APPLICATION_CREDENTIALS': os.getenv('GOOGLE_APPLICATION_CREDENTIALS'),
-        'GOOGLE_PALM_API': os.getenv('GOOGLE_PALM_API')
+        'PALM_API_KEY': os.getenv('PALM_API_KEY')
     }
 
     # Check if all environment variables are set
@@ -338,6 +338,13 @@ def create_private_file():
                     st.secrets["db_username"]
                 with c_button_ocr:
                     st.empty()
+                
+            with st.container():
+                with c_button_ocr:
+                    st.write("##")
+                    st.button("Test OCR", on_click=test_API, args=['google_vision',c_in_ocr,openai_api_key,azure_openai_api_version,azure_openai_api_key,
+                                                                        azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
+
 
         if os.getenv('OPENAI_API_KEY') is None:
             st.write("---")
@@ -352,6 +359,12 @@ def create_private_file():
                                                     type='password')
                 with c_button_openai:
                     st.empty()
+            with st.container():
+                with c_button_openai:
+                    st.write("##")
+                    st.button("Test OpenAI", on_click=test_API, args=['openai',c_in_openai,openai_api_key,azure_openai_api_version,azure_openai_api_key,
+                                                                        azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
+            
 
 
         if os.getenv('AZURE_API_KEY') is None:
@@ -382,43 +395,30 @@ def create_private_file():
                 c_in_azure, c_button_azure = st.columns([10,2])
                 with c_button_azure:
                     st.empty()
+            with st.container():
+                with c_button_azure:
+                    st.write("##")
+                    st.button("Test Azure OpenAI", on_click=test_API, args=['azure_openai',c_in_azure,openai_api_key,azure_openai_api_version,azure_openai_api_key,
+                                                                        azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
+                
 
 
-        if os.getenv('GOOGLE_PALM_API') is None:
+        if os.getenv('PALM_API_KEY') is None:
             st.write("---")
             st.subheader("Google PaLM 2")
             st.markdown('Follow these [instructions](https://developers.generativeai.google/tutorials/setup) to generate an API key for PaLM 2. You may need to also activate an account with [MakerSuite](https://makersuite.google.com/app/apikey) and enable "early access."')
             with st.container():
                 c_in_palm, c_button_palm = st.columns([10,2])
                 with c_in_palm:
-                    google_palm = st.text_input("Google PaLM 2 API Key", os.environ.get('GOOGLE_PALM_API', ''),
+                    google_palm = st.text_input("Google PaLM 2 API Key", os.environ.get('PALM_API_KEY', ''),
                                                     help='The MakerSuite API key e.g. a 32-character string',
                                                     placeholder='e.g. SATgthsykuE64FgrrrrEervr3S4455t_geyDeGq',
                                                     type='password')
-
-        with st.container():
-            with c_button_ocr:
-                st.write("##")
-                st.button("Test OCR", on_click=test_API, args=['google_vision',c_in_ocr,openai_api_key,azure_openai_api_version,azure_openai_api_key,
-                                                                    azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
-
-        with st.container():
-            with c_button_openai:
-                st.write("##")
-                st.button("Test OpenAI", on_click=test_API, args=['openai',c_in_openai,openai_api_key,azure_openai_api_version,azure_openai_api_key,
-                                                                    azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
-                
-        with st.container():
-            with c_button_azure:
-                st.write("##")
-                st.button("Test Azure OpenAI", on_click=test_API, args=['azure_openai',c_in_azure,openai_api_key,azure_openai_api_version,azure_openai_api_key,
-                                                                    azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
-                
-        with st.container():
-            with c_button_palm:
-                st.write("##")
-                st.button("Test PaLM 2", on_click=test_API, args=['palm',c_in_palm,openai_api_key,azure_openai_api_version,azure_openai_api_key,
-                                                                    azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
+            with st.container():
+                with c_button_palm:
+                    st.write("##")
+                    st.button("Test PaLM 2", on_click=test_API, args=['palm',c_in_palm,openai_api_key,azure_openai_api_version,azure_openai_api_key,
+                                                                        azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])
 
         st.button("Set API Keys",type='primary', on_click=set_API_keys, args=[openai_api_key,azure_openai_api_version,azure_openai_api_key,
                                                                     azure_openai_api_base,azure_openai_organization,azure_openai_api_type,google_vision,google_palm])