Update src/streamlit_app.py

src/streamlit_app.py

@@ -1280,13 +1280,17 @@ if atoms is not None:
                         
                         with open(tmp_filepath_opt, 'r') as file_opt:
                             xyz_content_opt = file_opt.read()
-                        
-                        st.download_button(
-                            label="Download Optimized Structure (XYZ)",
-                            data=xyz_content_opt,
-                            file_name="optimized_structure.xyz",
-                            mime="chemical/x-xyz"
-                        )
+                        @st.fragment
+                        def show_optimized_structure_download_button():
+                            # st.button("Release the balloons", help="Fragment rerun")
+                            # st.balloons()
+                            st.download_button(
+                                label="Download Optimized Structure (XYZ)",
+                                data=xyz_content_opt,
+                                file_name="optimized_structure.xyz",
+                                mime="chemical/x-xyz"
+                            )
+                        show_optimized_structure_download_button()
                         os.unlink(tmp_filepath_opt)
 
                         # Convert trajectory to XYZ for download