Update app.py

//LOG_ 111524
Fixing stuff
//endlog
/commut

app.py

@@ -1,6 +1,35 @@
 import streamlit as st
 import random
 
+class WheelOfNames:
+    def __init__(self, names):
+        self.names = names
+
+    def add_name(self, name):
+        if name.strip():
+            self.names.append(name.strip())
+            st.session_state.name_list_text = "\n".join(self.names)
+
+    def upload_file(self):
+        uploaded_file = st.file_uploader("Upload a text file containing names", type=["txt"])
+        if uploaded_file:
+            self.names.extend([line.decode().strip() for line in uploaded_file])
+            st.session_state.name_list_text = "\n".join(self.names)
+
+    def save_file(self):
+        if self.names:
+            return "\n".join(self.names)
+        return None
+
+    def clear_names(self):
+        self.names = []
+        st.session_state.name_list_text = ""
+        st.session_state.groups_text = ""
+        st.session_state.result_label = ""
+
+    import streamlit as st
+import random
+
 class WheelOfNames:
     def __init__(self, names):
         self.names = names
@@ -47,11 +76,15 @@ class WheelOfNames:
         random.shuffle(names_copy)
 
         groups = []
-        while names_copy:
+        while len(names_copy) >= group_size:
             group = names_copy[:group_size]
             groups.append(group)
             names_copy = names_copy[group_size:]
 
+        # If there are remaining names, add them as a group
+        if names_copy:
+            groups.append(names_copy)
+
         # Join groups into a single string
         groups_text = "\n".join(f"Group {i}: {', '.join(group)}" for i, group in enumerate(groups, 1))
         st.session_state.groups_text = groups_text
@@ -78,6 +111,91 @@ class WheelOfNames:
             st.session_state.name_list_text = "\n".join(self.names)
             st.session_state.result_label = ""
 
+def main():
+    st.title("Wheel of Names")
+
+    if 'names' not in st.session_state:
+        st.session_state.names = []
+    if 'name_list_text' not in st.session_state:
+        st.session_state.name_list_text = ""
+    if 'groups_text' not in st.session_state:
+        st.session_state.groups_text = ""
+    if'result_label' not in st.session_state:
+        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)
+
+    with st.sidebar:
+        st.subheader("Names")
+        name_entry = st.text_input("Enter a name")
+        if st.button("Add Name"):
+            wheel.add_name(name_entry)
+            st.session_state.names = wheel.names
+
+        wheel.upload_file()
+
+        if st.button("Save Names to File"):
+            names_data = wheel.save_file()
+            if names_data:
+                st.download_button(
+                    label="Download Names",
+                    data=names_data,
+                    file_name="names.txt",
+                    mime="text/plain"
+                )
+            else:
+                st.error("No names to save.")
+
+        if st.button("Clear Names"):
+            wheel.clear_names()
+            st.session_state.names = wheel.names
+
+        st.subheader("Names List")
+        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))
+    st.session_state.group_size_var = str(group_size_var)
+
+    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()
+
+    st.subheader("Groups")
+    st.text_area("Groups List", value=st.session_state.groups_text, height=150)
+
+    st.subheader("Spin the Wheel")
+    if st.button("Spin"):
+        wheel.spin_wheel()
+
+    st.subheader("Result")
+    st.markdown(f"<h1 style='text-align: center;'>{st.session_state.result_label}</h1>", unsafe_allow_html=True)
+
+if __name__ == "__main__":
+    main()
+
+    def spin_wheel(self):
+        if not self.names:
+            st.error("Please add or upload names first")
+            return
+
+        winning_name = random.choice(self.names)
+        st.session_state.result_label = winning_name
+
+        # Ask user if they want to remove the name
+        remove_name = st.button("Remove Selected Name", key="remove_name_button")
+        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")