api_key stored in session state

app.py

@@ -106,6 +106,11 @@ def main():
     <div align="right">
     Creado por Jorge Henao 🇨🇴 <a href="https://twitter.com/jhenaotw" target='_blank'>Twitter</a> <a href="https://www.linkedin.com/in/henaojorge" target='_blank'>LinkedIn</a> <a href="https://linktr.ee/jorgehenao" target='_blank'>Linktree</a>
     </div>""", unsafe_allow_html=True)    
+    
+    session_state = st.session_state
+    if "api_key" not in session_state:
+        session_state.api_key = ""
+    
     st.sidebar.title("Configuración de búsqueda")
     with st.sidebar.expander("Parámetros de recuperación", expanded= True):        
         index = st.selectbox("Selecciona el documento que deseas explorar", index_titles)
@@ -114,11 +119,15 @@ def main():
         
     with st.sidebar.expander("Configuración OpenAI"):        
         openai_api_key = st.text_input("API Key", type="password", placeholder="Copia aquí tu OpenAI API key (no será guardada)",
-                          help="puedes obtener tu api key de OpenAI en https://platform.openai.com/account/api-keys.")
+                          help="puedes obtener tu api key de OpenAI en https://platform.openai.com/account/api-keys.",
+                          value= session_state.api_key)
         openai_api_model = st.text_input("Modelo", value= "text-davinci-003")
         openai_api_temp = st.slider("Temperatura", 0.1, 1.0, 0.5, step=0.1)
-        openai_api_max_tokens = st.slider("Max tokens", 10, 100, 30, step=10)
+        openai_api_max_tokens = st.slider("Max tokens", 10, 100, 60, step=10)
     
+    if openai_api_key:
+        session_state.password = openai_api_key
+        
     sample_questions = get_samples_for_index(index).splitlines()
     query = st.text_area("",placeholder="Escribe aquí tu pregunta, cuanto más contexto le des, mejor serán las respuestas")
     with st.expander("Algunas preguntas de ejemplo", expanded= False):