Update app.py

app.py

@@ -7,6 +7,17 @@ import time
 from collections import deque  # Add this line
 import threading
 
+
+# Initialize session state variables
+if 'running' not in st.session_state:
+    st.session_state.running = 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
+
 class Organelle:
     def __init__(self, type):
         self.type = type
@@ -244,6 +255,10 @@ def update_chart():
     # Clear existing traces
     st.session_state.fig.data = []
     
+    if st.session_state.running and st.session_state.env is not None:
+    st.session_state.env.update()
+    update_chart()
+    
     # Cell positions
     cell_types = [cell.cell_type for cell in st.session_state.env.cells]
     x_positions = [cell.x for cell in st.session_state.env.cells]
@@ -285,25 +300,31 @@ def update_chart():
 
     st.session_state.fig.update_layout(height=800, width=1200, title_text="Advanced Cell Evolution Simulation")
 
-    # Update the chart placeholder
+st.sidebar.header("Simulation Settings")
+st.session_state.radiation = st.sidebar.checkbox("Enable Radiation", value=False)
+st.session_state.predation = st.sidebar.checkbox("Enable Predation", value=False)
+st.session_state.symbiosis = st.sidebar.checkbox("Enable Symbiosis", value=False)
+    
+# Update the chart placeholder
     chart_placeholder.plotly_chart(st.session_state.fig)
 
 def start_simulation():
     st.session_state.running = True
     if st.session_state.env is None:
+        # Initialize your environment here
+        effects = {
+            'radiation': st.session_state.radiation,
+            'predation': st.session_state.predation,
+            'symbiosis': st.session_state.symbiosis
+        }
         st.session_state.env = Environment(100, 100, effects)
-        for _ in range(initial_cells):
-            cell = Cell(random.uniform(0, st.session_state.env.width), random.uniform(0, st.session_state.env.height))
+        # Add initial cells to the environment
+        for _ in range(50):
+            cell = Cell(random.uniform(0, 100), random.uniform(0, 100))
             st.session_state.env.add_cell(cell)
-        st.session_state.fig = go.Figure(layout=go.Layout(
-            height=800, width=1200,
-            title='Advanced Cell Evolution Simulation',
-            showlegend=True,
-        ))
-        # Set up subplots
-        st.session_state.fig = make_subplots(rows=2, cols=2, 
-                                             subplot_titles=('Cell Positions', 'Population History', 
-                                                             'Population by Cell Type', 'Organelle Distribution'))
+    
+    if st.session_state.fig is None:
+        st.session_state.fig = setup_figure(st.session_state.env)
 
 def stop_simulation():
     st.session_state.running = False