Update app.py

app.py

@@ -183,20 +183,44 @@ def run_simulation(num_steps, initial_cells):
         if step % 10 == 0:  # Visualize every 10 steps
             yield env.visualize()
 
-# Streamlit app
 st.title("Cell Evolution Simulation")
 
 num_steps = st.slider("Number of simulation steps", 100, 1000, 500)
 initial_cells = st.slider("Initial number of cells", 10, 100, 50)
+update_interval = st.slider("Update interval (milliseconds)", 100, 1000, 200)
 
 if st.button("Run Simulation"):
-    simulation = run_simulation(num_steps, initial_cells)
+    env = Environment(100, 100)
     
-    # Create a placeholder for the chart
-    chart_placeholder = st.empty()
+    # Add initial cells
+    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)
+    chart = st.plotly_chart(fig, use_container_width=True)
     
-    # Update the chart for each step
-    for chart in simulation:
-        chart_placeholder.plotly_chart(chart, use_container_width=True)
+    # Run simulation
+    for step in range(num_steps):
+        env.update()
+        
+        # Update the figure data
+        with fig.batch_update():
+            cell_data, population_history = 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"]
+            
+            for i, (cell_type, counts) in enumerate(population_history.items()):
+                fig.data[i+4].y = counts  # +4 because we have 4 cell types in the first subplot
+        
+        fig.layout.title.text = f"Cell Evolution Simulation (Time: {env.time})"
+        
+        # Update the chart
+        chart.plotly_chart(fig, use_container_width=True)
+        
+        time.sleep(update_interval / 1000)  # Convert milliseconds to seconds
 
     st.write("Simulation complete!")