Update app.py

app.py

@@ -275,6 +275,20 @@ 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
@@ -306,78 +320,34 @@ with col1:
 with col2:
     stop_button = st.button("Stop Simulation", on_click=stop_simulation)
 
-def update_chart():
-    if st.session_state.env is not None and st.session_state.fig is not None:
-        cell_data, population_history, colors = st.session_state.env.get_visualization_data()
-        
-        # Update cell distribution
-        for i, (cell_type, data) in enumerate(cell_data.items()):
-            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"]
-        
-        # Update total population
-        total_population = [sum(counts) for counts in zip(*population_history.values())]
-        st.session_state.fig.data[5].y = total_population
-        
-        # Update population by cell type
-        for i, (cell_type, counts) in enumerate(population_history.items()):
-            st.session_state.fig.data[6+i].y = counts
-        
-        # Update organelle distribution
-        organelle_counts = {"nucleus": 0, "mitochondria": 0, "chloroplast": 0, "endoplasmic_reticulum": 0, "golgi_apparatus": 0}
-        for cell in st.session_state.env.cells:
-            for organelle in cell.organelles:
-                organelle_counts[organelle] += 1
-        st.session_state.fig.data[11].y = list(organelle_counts.values())
-        
-        chart_placeholder.plotly_chart(st.session_state.fig, use_container_width=True)
-
-def update_statistics():
-    if st.session_state.env is not None:
-        total_cells = len(st.session_state.env.cells)
-        total_cells_text.text(f"Total Cells: {format_number(total_cells)}")
-        
-        cell_type_counts = {cell_type: len([cell for cell in st.session_state.env.cells if cell.cell_type == cell_type]) for cell_type in st.session_state.env.population_history.keys()}
-        breakdown = "\n".join([f"{cell_type}: {format_number(count)}" for cell_type, count in cell_type_counts.items()])
-        cell_type_breakdown.text(f"Cell Type Breakdown:\n{breakdown}")
-        
-        dominant_type = max(cell_type_counts, key=cell_type_counts.get)
-        dominant_type_text.text(f"Dominant Type: {dominant_type}")
-        
-        avg_energy = sum(cell.energy for cell in st.session_state.env.cells) / total_cells if total_cells > 0 else 0
-        avg_energy_text.text(f"Average Energy: {avg_energy:.2f}")
-        
-        total_merges_text.text(f"Total Merges: {st.session_state.total_merges}")
-        
-        event_log_text.text("\n".join(event_log))
-
 # Main simulation loop
 simulation_container = st.empty()
 
-while True:
-    with simulation_container.container():
-        if st.session_state.running and 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)
+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")
                 
-                # 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()
             
-            update_chart()
-            update_statistics()
+            time.sleep(update_interval)
         
-        time.sleep(update_interval)
-    
-    simulation_container.empty()
+        st.experimental_rerun()
 
-    # Break the loop if the simulation is not running
-    if not st.session_state.running:
-        break
+if st.session_state.running:
+    simulation_thread = threading.Thread(target=run_simulation)
+    simulation_thread.start()
 
-st.write("Simulation stopped. Click 'Start Simulation' to run again.")
+if not st.session_state.running:
+    st.write("Simulation stopped. Click 'Start Simulation' to run again.")