Update app.py

//LOG_ 111424_a
Add save names to file functionality
//endlog
/commit

app.py

@@ -18,13 +18,8 @@ class WheelOfNames:
 
     def save_file(self):
         if self.names:
-            with st.expander("Download Names as Text File"):
-                st.download_button(
-                    label="Download",
-                    data="\n".join(self.names),
-                    file_name="names.txt",
-                    mime="text/plain"
-                )
+            return "\n".join(self.names)
+        return None
 
     def clear_names(self):
         self.names = []
@@ -88,7 +83,19 @@ def main():
             wheel.add_name(name_entry)
 
         wheel.upload_file()
-        wheel.save_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()