update

pages/2_Benchmark_Data.py

@@ -10,43 +10,49 @@ Once verified, you can specify the number of samples you wish to retrieve and do
 """)
 
 
-def check_password():
-    def password_entered():
+def verify_password():
+    """
+    Prompts the user to enter a password and checks it against the environment variable.
+    If the password is correct, sets the session state to indicate successful verification.
+    """
+
+    def on_password_entered():
         if password_input == os.getenv('PASSWORD'):
-            st.session_state['password_correct'] = True
+            st.session_state['password_verified'] = True
         else:
-            st.error("Incorrect Password, please try again.")
+            st.error("Incorrect password, please try again.")
 
     password_input = st.text_input("Enter Password:", type="password")
-    submit_button = st.button("Submit", on_click=password_entered)
+    submit_button = st.button("Submit", on_click=on_password_entered)
 
-    if submit_button and not st.session_state.get('password_correct', False):
+    if submit_button and not st.session_state.get('password_verified', False):
         st.error("Please enter a valid password to access the demo.")
 
 
-
-if not st.session_state.get('password_correct', False):
-    check_password()
+# Check if the password has been verified
+if not st.session_state.get('password_verified', False):
+    verify_password()
 else:
     st.sidebar.success("Password Verified. Proceed with the demo.")
 
     # Allow user to set the number of samples
-    num_samples = st.sidebar.number_input(
+    num_samples = st.number_input(
         "Set number of samples:",
         min_value=1,
         max_value=100,
         value=5
     )
 
-    # Load and display data
-    data = get_data(num_samples)
-    st.dataframe(data['question'])
-
-    # Create a CSV download link
-    csv = data['question'].to_csv(index=False)
-    st.download_button(
-        label="Download data samples as CSV",
-        data=csv,
-        file_name='data_samples.csv',
-        mime='text/csv',
-    )
+    if st.button("Retrieve Data"):
+        # Load and display data
+        data = get_data(num_samples)
+        st.dataframe(data['question'])
+
+        # Create a CSV download link
+        csv = data['question'].to_csv(index=False)
+        st.download_button(
+            label="Download data samples as CSV",
+            data=csv,
+            file_name='data_samples.csv',
+            mime='text/csv',
+        )