Update app.py

//LOG_ 111424_g
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):
+    def sort_into_groups(self, remove_used_names):
         group_size = st.session_state.group_size_var
         if not group_size.isdigit():
             st.error("Please enter a valid group size.")
@@ -56,6 +56,12 @@ 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
 
+        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)
+
     def spin_wheel(self):
         if not self.names:
             st.error("Please add or upload names first")
@@ -119,8 +125,10 @@ 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")
+
     if st.button("Sort into Groups"):
-        wheel.sort_into_groups()
+        wheel.sort_into_groups(remove_used_names)
 
     st.subheader("Groups")
     st.text_area("Groups List", value=st.session_state.groups_text, height=150)