Update src/streamlit_app.py

src/streamlit_app.py

@@ -1581,7 +1581,7 @@ if atoms is not None:
                     if "Optimization" in task and "Final Energy" in results: # Check if opt was successful
                         st.markdown("### Optimized Structure")
 
-                        opt_view = get_structure_viz2(opt_atoms_obj, style=viz_style, show_unit_cell=True, width=400, height=400)
+                        opt_view = get_structure_viz2(calc_atoms, style=viz_style, show_unit_cell=True, width=400, height=400)
                         st.components.v1.html(opt_view._make_html(), width=400, height=400)
                         
                         with tempfile.NamedTemporaryFile(delete=False, suffix=".xyz", mode="w+") as tmp_file_opt: