Update app.py

app.py

@@ -9,9 +9,9 @@ CSV_FILENAME = "user_selections.csv"
 # Function to assign samples to users
 def assign_samples(csv_path):
     df = pd.read_csv(csv_path)
-    group_1 = df[(df["pool_pos"] == 1) & (~df["question_category"].str.endswith("_B"))].head(10)
-    group_2 = df[(df["pool_pos"] == 2) & (~df["question_category"].str.endswith("_B"))].head(10)
-    group_3 = df[(df["pool_pos"] == 3) & (~df["question_category"].str.endswith("_B"))].head(10)
+    group_1 = df[(df["pool_pos"] == 1) & (~df["question_category"].str.endswith("_B"))].head(5)
+    group_2 = df[(df["pool_pos"] == 2) & (~df["question_category"].str.endswith("_B"))].head(5)
+    group_3 = df[(df["pool_pos"] == 3) & (~df["question_category"].str.endswith("_B"))].head(5)
     return {
         "Bernardo": group_1,
         "Alessandro": group_1,
@@ -47,15 +47,17 @@ if "index" not in st.session_state:
 if "results" not in st.session_state:
     st.session_state.results = []
 
-# User selection form
+# User selectiontion
+def update_name():
+    """Set username and reset index."""
+    st.session_state.username = st.session_state.selected_user
+    st.session_state.index = 0  # Reset progress
+
 if st.session_state.username is None:
     with st.form("user_form"):
         st.write("### Select Your Name")
         selected_user = st.selectbox("Choose your name:", valid_users, key="selected_user")
-        submit_button = st.form_submit_button("Start")
-        if submit_button:
-            st.session_state.username = selected_user
-            st.session_state.index = 0
+        submit_button = st.form_submit_button("Start", on_click=update_name)
     st.stop()
 
 # Retrieve assigned dataset and filter out already labeled samples