Update app.py

app.py

@@ -275,21 +275,8 @@ event_log_text = st.sidebar.empty()
 # Create placeholders for the chart
 chart_placeholder = st.empty()
 
-import streamlit as st
-import time
-from collections import deque
-import threading
-
-# ... (keep all the imports and class definitions)
-
-# ... (keep all the function definitions)
 
-# Streamlit app
-st.title("Continuous Cell Evolution Simulation")
-
-# ... (keep all the sidebar controls and placeholders)
 
-# Initialize session state
 if 'running' not in st.session_state:
     st.session_state.running = False
 if 'total_merges' not in st.session_state:
@@ -323,31 +310,24 @@ with col2:
 # Main simulation loop
 simulation_container = st.empty()
 
-def run_simulation():
-    while st.session_state.running:
-        with simulation_container.container():
-            if st.session_state.env is not None:
-                for _ in range(4):  # Update 4 times per frame to increase simulation speed
-                    initial_cell_count = len(st.session_state.env.cells)
-                    st.session_state.env.update()
-                    final_cell_count = len(st.session_state.env.cells)
-                    
-                    # Check for merges
-                    if final_cell_count < initial_cell_count:
-                        merges = initial_cell_count - final_cell_count
-                        st.session_state.total_merges += merges
-                        event_log.appendleft(f"Time {st.session_state.env.time}: {merges} cell merges occurred")
-                
-                update_chart()
-                update_statistics()
+if st.session_state.running and st.session_state.env is not None:
+    with simulation_container.container():
+        for _ in range(4):  # Update 4 times per frame to increase simulation speed
+            initial_cell_count = len(st.session_state.env.cells)
+            st.session_state.env.update()
+            final_cell_count = len(st.session_state.env.cells)
             
-            time.sleep(update_interval)
+            # Check for merges
+            if final_cell_count < initial_cell_count:
+                merges = initial_cell_count - final_cell_count
+                st.session_state.total_merges += merges
+                event_log.appendleft(f"Time {st.session_state.env.time}: {merges} cell merges occurred")
         
-        st.experimental_rerun()
-
-if st.session_state.running:
-    simulation_thread = threading.Thread(target=run_simulation)
-    simulation_thread.start()
+        update_chart()
+        update_statistics()
+    
+    time.sleep(update_interval)
+    st.experimental_rerun()
 
 if not st.session_state.running:
     st.write("Simulation stopped. Click 'Start Simulation' to run again.")