Update app.py

app.py

@@ -236,6 +236,16 @@ def setup_figure(env):
 
     return fig
 
+import random
+import numpy as np
+import streamlit as st
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
+import time
+import threading
+
+# Your classes and functions...
+
 # Streamlit app
 st.title("Advanced Cell Evolution Simulation")
 
@@ -246,50 +256,52 @@ update_interval = st.slider("Update interval (seconds)", 0.1, 5.0, 1.0)
 chart_placeholder = st.empty()
 
 # Create a button to start/stop the simulation
-start_button = st.button("Start Simulation")
-
-if start_button:
-    env = Environment(100, 100)
-    
-    # Add initial cells
+if 'run_simulation' not in st.session_state:
+    st.session_state.run_simulation = False
+if 'env' not in st.session_state:
+    st.session_state.env = None
+if 'fig' not in st.session_state:
+    st.session_state.fig = None
+
+def start_simulation():
+    st.session_state.run_simulation = True
+    st.session_state.env = Environment(100, 100)
     for _ in range(initial_cells):
-        cell = Cell(random.uniform(0, env.width), random.uniform(0, env.height))
-        env.add_cell(cell)
-    
-    # Set up the figure
-    fig = setup_figure(env)
-    
-    # Run simulation
-    while True:
-        env.update()
+        cell = Cell(random.uniform(0, st.session_state.env.width), random.uniform(0, st.session_state.env.height))
+        st.session_state.env.add_cell(cell)
+    st.session_state.fig = setup_figure(st.session_state.env)
+    thread = threading.Thread(target=run_simulation)
+    thread.start()
+
+def stop_simulation():
+    st.session_state.run_simulation = False
+
+start_button = st.button("Start Simulation", on_click=start_simulation)
+stop_button = st.button("Stop Simulation", on_click=stop_simulation)
+
+def run_simulation():
+    while st.session_state.run_simulation:
+        st.session_state.env.update()
         
-        with fig.batch_update():
-            cell_data, population_history = env.get_visualization_data()
+        with st.session_state.fig.batch_update():
+            cell_data, population_history = st.session_state.env.get_visualization_data()
             for i, (cell_type, data) in enumerate(cell_data.items()):
-                fig.data[i].x = data["x"]
-                fig.data[i].y = data["y"]
-                fig.data[i].marker.size = data["size"]
+                st.session_state.fig.data[i].x = data["x"]
+                st.session_state.fig.data[i].y = data["y"]
+                st.session_state.fig.data[i].marker.size = data["size"]
             
             for i, (cell_type, counts) in enumerate(population_history.items()):
-                fig.data[i+5].y = counts  # +5 because we have 5 cell types in the first subplot
+                st.session_state.fig.data[i+5].y = counts  # +5 because we have 5 cell types in the first subplot
                 if cell_type != "modified" and cell_type != "plant_like":
-                    fig.data[i+10].y = counts  # Update individual population charts
+                    st.session_state.fig.data[i+10].y = counts  # Update individual population charts
                 else:
-                    fig.data[13].y = population_history["plant_like"]
-                    fig.data[14].y = population_history["modified"]
+                    st.session_state.fig.data[13].y = population_history["plant_like"]
+                    st.session_state.fig.data[14].y = population_history["modified"]
         
-        fig.layout.title.text = f"Advanced Cell Evolution Simulation (Time: {env.time})"
+        st.session_state.fig.layout.title.text = f"Advanced Cell Evolution Simulation (Time: {st.session_state.env.time})"
         
         # Update the chart
-        chart_placeholder.plotly_chart(fig, use_container_width=True)
+        chart_placeholder.plotly_chart(st.session_state.fig, use_container_width=True)
         
         # Wait for the specified interval
-        time.sleep(update_interval)
-
-        # Check if the Streamlit script should stop
-        if not st.session_state.get('run_simulation', True):
-            break
-
-# Add a button to stop the simulation
-if st.button("Stop Simulation"):
-    st.session_state.run_simulation = False
+        time.sleep(update_interval)