Update app.py

app.py

@@ -362,8 +362,15 @@ def update_chart():
         
         chart_placeholder.plotly_chart(st.session_state.fig, use_container_width=True)
 
+# Main loop placeholder
+main_placeholder = st.empty()
+
 # Main loop
 while True:
-    run_simulation_step()
-    time.sleep(update_interval)
-    st.experimental_rerun()
+    with main_placeholder.container():
+        run_simulation_step()
+        time.sleep(update_interval)
+    
+    # Break the loop if the simulation is stopped
+    if not st.session_state.running:
+        break