Update app.py

//LOG_ 111424_d
Fix spinning/sorting issues
//endlog
/commit

app.py

@@ -44,13 +44,13 @@ class WheelOfNames:
 
         random.shuffle(self.names)
         groups = []
-        while self.names:
-            group = self.names[:group_size]
+        names_copy = self.names.copy()
+        while names_copy:
+            group = names_copy[:group_size]
             groups.append(group)
-            self.names = self.names[group_size:]
+            names_copy = names_copy[group_size:]
 
         st.session_state.groups_text = "\n".join(f"Group {i}: {', '.join(group)}" for i, group in enumerate(groups, 1))
-        st.session_state.name_list_text = "\n".join(self.names)
 
     def spin_wheel(self):
         if not self.names:
@@ -59,8 +59,13 @@ class WheelOfNames:
 
         winning_name = random.choice(self.names)
         st.session_state.result_label = winning_name
-        self.names.remove(winning_name)
-        st.session_state.name_list_text = "\n".join(self.names)
+
+        # Ask user if they want to remove the name
+        remove_name = st.button("Remove Selected Name")
+        if remove_name:
+            self.names.remove(winning_name)
+            st.session_state.name_list_text = "\n".join(self.names)
+            st.session_state.result_label = ""
 
 def main():
     st.title("Wheel of Names")
@@ -104,7 +109,7 @@ def main():
             st.session_state.names = wheel.names
 
         st.subheader("Names List")
-        st.text_area("Names List", value=st.session_state.name_list_text, height=150)
+        st.text_area("Names List", value=st.session_state.name_list_text, height=110)
 
     st.subheader("Group Size")
     group_size_var = st.number_input("Enter group size", min_value=1, value=int(st.session_state.group_size_var))
@@ -112,7 +117,6 @@ def main():
 
     if st.button("Sort into Groups"):
         wheel.sort_into_groups()
-        st.session_state.names = wheel.names
 
     st.subheader("Groups")
     st.text_area("Groups List", value=st.session_state.groups_text, height=150)
@@ -120,7 +124,6 @@ def main():
     st.subheader("Spin the Wheel")
     if st.button("Spin"):
         wheel.spin_wheel()
-        st.session_state.names = wheel.names
 
     st.subheader("Result")
     st.markdown(f"<h1 style='text-align: center;'>{st.session_state.result_label}</h1>", unsafe_allow_html=True)