Update src/streamlit_app.py

src/streamlit_app.py

@@ -505,19 +505,20 @@ def show_motion_comparison():
     # Single plot and table section (no placeholders needed)
     st.markdown("---")
     
-    # Create a cache key based on all parameters to avoid unnecessary redraws
-    cache_key = f"{speed1}_{angle1}_{height1}_{speed2}_{angle2}_{height2}_{speed3}_{angle3}_{height3}"
-    
-    # Use containers to prevent flickering
-    plot_container = st.container()
-    table_container = st.container()
-    
-    with plot_container:
-        # Create comparison plot
+    # Create comparison plot using st.cache_data to prevent re-rendering
+    @st.cache_data
+    def create_comparison_plot(speed1, angle1, height1, speed2, angle2, height2, speed3, angle3, height3):
+        # Recreate trajectories inside cached function
+        trajectories_cached = [
+            ("Trajectory 1", Motion2D(launch_speed=speed1, launch_angle=angle1, launch_height=height1), 'blue'),
+            ("Trajectory 2", Motion2D(launch_speed=speed2, launch_angle=angle2, launch_height=height2), 'red'),
+            ("Trajectory 3", Motion2D(launch_speed=speed3, launch_angle=angle3, launch_height=height3), 'green')
+        ]
+        
         fig, ax = plt.subplots(figsize=(12, 8))
         
         max_range = 0
-        for name, motion, color in trajectories:
+        for name, motion, color in trajectories_cached:
             data = motion.trajectory_data(motion.calculate_flight_time())
             ax.plot(data['x'], data['y'], linewidth=3, label=name, color=color)
             
@@ -533,17 +534,21 @@ def show_motion_comparison():
         ax.grid(True, alpha=0.3)
         ax.legend(fontsize=12)
         ax.set_ylim(bottom=0)
-        ax.set_xlim(0, max_range * 1.1)
+        ax.set_xlim(0, max_range * 1.1 if max_range > 0 else 10)
         
-        st.pyplot(fig, clear_figure=True)
-        plt.close(fig)  # Explicitly close the figure
+        return fig
     
-    with table_container:
-        # Comparison table
-        st.markdown("### 📋 Trajectory Comparison Table")
+    # Create cached comparison data
+    @st.cache_data
+    def create_comparison_data(speed1, angle1, height1, speed2, angle2, height2, speed3, angle3, height3):
+        trajectories_data = [
+            ("Trajectory 1", Motion2D(launch_speed=speed1, launch_angle=angle1, launch_height=height1)),
+            ("Trajectory 2", Motion2D(launch_speed=speed2, launch_angle=angle2, launch_height=height2)),
+            ("Trajectory 3", Motion2D(launch_speed=speed3, launch_angle=angle3, launch_height=height3))
+        ]
         
         comparison_data = []
-        for name, motion, _ in trajectories:
+        for name, motion in trajectories_data:
             info = motion.get_launch_info()
             comparison_data.append({
                 "Trajectory": name,
@@ -555,7 +560,17 @@ def show_motion_comparison():
                 "Max Height (m)": f"{info['max_height']:.1f}"
             })
         
-        st.table(comparison_data)
+        return comparison_data
+    
+    # Display cached plot
+    fig = create_comparison_plot(speed1, angle1, height1, speed2, angle2, height2, speed3, angle3, height3)
+    st.pyplot(fig, clear_figure=True)
+    plt.close(fig)
+    
+    # Display cached comparison table
+    st.markdown("### 📋 Trajectory Comparison Table")
+    comparison_data = create_comparison_data(speed1, angle1, height1, speed2, angle2, height2, speed3, angle3, height3)
+    st.table(comparison_data)
 
 if __name__ == "__main__":
     main()