Update src/streamlit_app.py

src/streamlit_app.py

@@ -751,7 +751,7 @@ def get_animated_trajectory_viz(trajectory, style='stick', show_unit_cell=True,
 
 
 # Streamlit implementation example
-def display_optimization_trajectory(trajectory, viz_style='stick'):
+def display_optimization_trajectory(trajectory, viz_style='ball-stick'):
     """
     Display optimization trajectory in Streamlit with controls
     """
@@ -767,7 +767,7 @@ def display_optimization_trajectory(trajectory, viz_style='stick'):
     with col1:
         viz_mode = st.selectbox(
             "Visualization Mode",
-            ["Static with paths", "Animation", "Step-by-step"],
+            ["Animation", "Static with paths", "Step-by-step"],
             key="viz_mode"
         )
     
@@ -1300,7 +1300,7 @@ if atoms is not None:
                             try:
                                 # Read trajectory and convert to XYZ
                                 trajectory = read(traj_filename, index=':')  # Read all frames
-                                display_optimization_trajectory(trajectory, "Animation")
+                                display_optimization_trajectory(trajectory, "ball-stick")
                                 
                                 # Create XYZ content
                                 trajectory_xyz = ""