Update src/streamlit_app.py

src/streamlit_app.py

@@ -1317,13 +1317,18 @@ if atoms is not None:
                                 
                                 st.markdown("### Optimization Trajectory")
                                 st.write(f"Captured {len(trajectory)} optimization steps")
+                                @st.fragment
+                                def show_trajectory_download_button():
+                                    # st.button("Release the balloons", help="Fragment rerun")
+                                    # st.balloons()
+                                    st.download_button(
+                                        label="Download Optimization Trajectory (XYZ)",
+                                        data=trajectory_xyz,
+                                        file_name="optimization_trajectory.xyz",
+                                        mime="chemical/x-xyz"
+                                    )
+                                show_trajectory_download_button()
                                 
-                                st.download_button(
-                                    label="Download Optimization Trajectory (XYZ)",
-                                    data=trajectory_xyz,
-                                    file_name="optimization_trajectory.xyz",
-                                    mime="chemical/x-xyz"
-                                )
                                 
                             except Exception as e:
                                 st.warning(f"Could not process trajectory: {e}")