Update app.py

app.py

@@ -59,6 +59,16 @@ print('*****************************************')
 mode = int(mode)
 
 
+# Check if 'key' already exists in session_state
+# If not, then initialize it
+if 'visibility' not in st.session_state:
+    st.session_state['visibility'] = 'visible'
+
+# Session State also supports the attribute based syntax
+if 'visibility' not in st.session_state:
+    st.session_state.key = 'visible'
+
+    
 if "visibility" not in st.session_state:
     st.session_state.visibility = "visible"
     st.session_state.disabled = False