persistent decryption-key input

app.py

@@ -28,7 +28,6 @@ def initialize():
     st.session_state.setdefault('db', None)
     st.session_state.setdefault('results', pd.DataFrame())
     st.session_state.setdefault('decrypt_key', None)
-    st.session_state.setdefault('valid_decrypt_key', False)
 
     with open('config.yaml', 'r') as stream:
         st.session_state['config'] = yaml.safe_load(stream)
@@ -75,15 +74,14 @@ def show_demo(placeholder):
 
             with st.form("submission_form"):
 
-                if not st.session_state['valid_decrypt_key']:
-                    with st.expander(label="Authentication", expanded=True, icon="🔑"):
-                        st.text_input(
-                            label="Encryption key", 
-                            value="", 
-                            max_chars=None, 
-                            key='decrypt_key',
-                            placeholder="A URL-safe base64-encoded 32-byte key"
-                        )
+                with st.expander(label="Authentication", expanded=True, icon="🔑"):
+                    st.text_input(
+                        label="Encryption key", 
+                        value="", 
+                        max_chars=None, 
+                        key='decrypt_key',
+                        placeholder="A URL-safe base64-encoded 32-byte key"
+                    )
 
                 with st.expander(label="Model", expanded=False, icon="🧠"):
 

modeling.py

@@ -45,7 +45,6 @@ def load_model():
                 cipher = Fernet(st.session_state['decrypt_key'])
                 decrypted_df = cipher.decrypt(encrypted_data)
                 st.session_state['db'] = pickle.loads(decrypted_df)
-                st.session_state['valid_decrypt_key'] = True
             except Exception as e:
                 st.error(body="Error: No valid encryption key!", icon="🔑")
                 logging.error(e)