Update rag_search.py

rag_search.py

@@ -415,6 +415,9 @@ if 'api_key_valid' not in st.session_state:
 if 'previous_api_key' not in st.session_state:
     st.session_state['previous_api_key'] = None
 
+OPENAI_KEY = st.secrets["OPENAI_KEY"]
+GEMINI_KEY = st.secrets["GEMINI_KEY"]
+COHERE_KEY = st.secrets["COHERE_KEY"]
 
 def load_data(file_path, key='df'):
     return pd.read_hdf(file_path, key=key)
@@ -436,7 +439,7 @@ if datasett is not None:
         label='Select columns to analyze',
         options=st.session_state['parsed_responses'].columns)
     with col2:
-        COHERE_KEY = st.text_input('Cohere APIs Key', '', type='password', help="Enter your Cohere API key")
+        COHERE_KEY = st.text_input('Cohere APIs Key', COHERE_KEY, type='password', help="Enter your Cohere API key")
     
     question = st.text_input("Ask a question")