update

pages/1_Demo_1.py

@@ -10,6 +10,26 @@ import os
 st.set_page_config(page_title="Gender Bias Analysis", page_icon="🔍", layout="wide")
 st.title('Gender Bias Analysis in Text Generation')
 
+# Initialize session state variables
+if 'password_correct' not in st.session_state:
+    st.session_state['password_correct'] = False
+if 'data_size' not in st.session_state:
+    st.session_state['data_size'] = 10
+if 'bold' not in st.session_state:
+    st.session_state['bold'] = None
+if 'female_bold' not in st.session_state:
+    st.session_state['female_bold'] = []
+if 'male_bold' not in st.session_state:
+    st.session_state['male_bold'] = []
+if 'male_prompts' not in st.session_state:
+    st.session_state['male_prompts'] = []
+if 'female_prompts' not in st.session_state:
+    st.session_state['female_prompts'] = []
+if 'male_continuations' not in st.session_state:
+    st.session_state['male_continuations'] = []
+if 'female_continuations' not in st.session_state:
+    st.session_state['female_continuations'] = []
+
 # Password protection function
 def check_password():
     def password_entered():
@@ -18,16 +38,12 @@ def check_password():
         else:
             st.error("Incorrect Password, please try again.")
 
-    if 'password_correct' not in st.session_state:
-        st.session_state['password_correct'] = False
-
-    if not st.session_state['password_correct']:
-        password_input = st.text_input("Enter Password:", type="password")
-        st.button("Submit", on_click=password_entered)
+    password_input = st.text_input("Enter Password:", type="password")
+    st.button("Submit", on_click=password_entered)
 
 # Data loading function
 def load_data():
-    if 'bold' not in st.session_state:
+    if st.session_state['bold'] is None:
         st.session_state['bold'] = load_dataset("AlexaAI/bold", split="train")
 
 # Sampling function
@@ -95,7 +111,7 @@ def evaluate_regard():
         st.json(regard_results_avg)
 
 # Main app logic
-if not st.session_state.get('password_correct', False):
+if not st.session_state['password_correct']:
     check_password()
 else:
     st.sidebar.success("Password Verified. Proceed with the demo.")
@@ -103,7 +119,7 @@ else:
 
     st.subheader('Step 1: Set Data Size')
     data_size = st.slider('Select number of samples per category:', min_value=1, max_value=50,
-                          value=st.session_state.get('data_size', 10))
+                          value=st.session_state['data_size'])
     st.session_state['data_size'] = data_size
 
     if st.button('Show Data'):
@@ -111,12 +127,12 @@ else:
         st.write(f'Sampled {data_size} female and male American actors.')
         display_samples()
 
-    if st.session_state.get('female_bold') and st.session_state.get('male_bold'):
+    if st.session_state['female_bold'] and st.session_state['male_bold']:
         st.subheader('Step 2: Generate Text')
         if st.button('Generate Text'):
             generate_text()
 
-    if st.session_state.get('male_continuations') and st.session_state.get('female_continuations'):
+    if st.session_state['male_continuations'] and st.session_state['female_continuations']:
         st.subheader('Step 3: Evaluate')
         display_samples()
         if st.button('Evaluate'):