Update app.py

//LOG_ 111424_h
Fix
//endlog
/commit

app.py

@@ -27,7 +27,7 @@ class WheelOfNames:
         st.session_state.groups_text = ""
         st.session_state.result_label = ""
 
-    def sort_into_groups(self, remove_used_names):
+    def sort_into_groups(self):
         group_size = st.session_state.group_size_var
         if not group_size.isdigit():
             st.error("Please enter a valid group size.")
@@ -56,8 +56,9 @@ class WheelOfNames:
         groups_text = "\n".join(f"Group {i}: {', '.join(group)}" for i, group in enumerate(groups, 1))
         st.session_state.groups_text = groups_text
 
+        # Optionally remove used names from the original list
+        remove_used_names = st.session_state.remove_used_names
         if remove_used_names:
-            # Remove used names from the original list
             used_names = [name for group in groups for name in group]
             self.names = [name for name in self.names if name not in used_names]
             st.session_state.name_list_text = "\n".join(self.names)
@@ -90,6 +91,8 @@ def main():
         st.session_state.result_label = ""
     if 'group_size_var' not in st.session_state:
         st.session_state.group_size_var = "2"
+    if 'remove_used_names' not in st.session_state:
+        st.session_state.remove_used_names = False
 
     wheel = WheelOfNames(st.session_state.names)
 
@@ -125,10 +128,12 @@ def main():
     group_size_var = st.number_input("Enter group size", min_value=1, value=int(st.session_state.group_size_var))
     st.session_state.group_size_var = str(group_size_var)
 
-    remove_used_names = st.checkbox("Remove Used Names After Sorting")
+    st.subheader("Sorting Options")
+    remove_used_names = st.checkbox("Remove Used Names After Sorting", value=st.session_state.remove_used_names)
+    st.session_state.remove_used_names = remove_used_names
 
     if st.button("Sort into Groups"):
-        wheel.sort_into_groups(remove_used_names)
+        wheel.sort_into_groups()
 
     st.subheader("Groups")
     st.text_area("Groups List", value=st.session_state.groups_text, height=150)