Update app.py

app.py

@@ -19,7 +19,6 @@ class Cell:
         self.size = 1
         self.color = "lightblue"
         self.division_threshold = 150
-        
         self.update_properties()
 
     def update_properties(self):
@@ -64,11 +63,11 @@ class Cell:
             return new_cell
         return None
 
-    def can_fuse(self, other):
+    def can_merge(self, other):
         return (self.cell_type == other.cell_type and
-                random.random() < 0.005)  # 0.5% chance of fusion
+                random.random() < 0.01)  # 1% chance of merging
 
-    def fuse(self, other):
+    def merge(self, other):
         new_cell_type = self.cell_type
         if self.cell_type == "prokaryote":
             new_cell_type = "early_eukaryote"
@@ -86,7 +85,7 @@ class Cell:
         return new_cell
 
 class Environment:
-    def __init__(self, width, height):
+    def __init__(self, width, height, effects):
         self.width = width
         self.height = height
         self.grid = np.random.rand(height, width) * 10
@@ -97,6 +96,7 @@ class Environment:
             "prokaryote": [], "early_eukaryote": [],
             "advanced_eukaryote": [], "plant_like": [], "complete": []
         }
+        self.effects = effects
 
     def add_cell(self, cell):
         self.cells.append(cell)
@@ -120,15 +120,23 @@ class Environment:
                 if new_cell:
                     new_cells.append(new_cell)
 
-        # Handle cell fusion
+        # Handle cell merging
         for i, cell1 in enumerate(self.cells):
             for cell2 in self.cells[i+1:]:
-                if cell1.can_fuse(cell2):
-                    new_cell = cell1.fuse(cell2)
+                if cell1.can_merge(cell2):
+                    new_cell = cell1.merge(cell2)
                     new_cells.append(new_cell)
                     cells_to_remove.extend([cell1, cell2])
 
-        # Add new cells and remove dead/fused cells
+        # Apply effects
+        if self.effects['radiation']:
+            self.apply_radiation()
+        if self.effects['predation']:
+            self.apply_predation()
+        if self.effects['symbiosis']:
+            self.apply_symbiosis()
+
+        # Add new cells and remove dead/merged cells
         self.cells.extend(new_cells)
         self.cells = [cell for cell in self.cells if cell not in cells_to_remove]
 
@@ -137,6 +145,33 @@ class Environment:
             count = len([cell for cell in self.cells if cell.cell_type == cell_type])
             self.population_history[cell_type].append(count)
 
+    def apply_radiation(self):
+        for cell in self.cells:
+            if random.random() < 0.01:  # 1% chance of mutation
+                cell.energy *= 0.8
+                if random.random() < 0.5:
+                    cell.organelles.add(random.choice(["nucleus", "mitochondria", "chloroplast", "endoplasmic_reticulum", "golgi_apparatus"]))
+                else:
+                    if cell.organelles:
+                        cell.organelles.remove(random.choice(list(cell.organelles)))
+                cell.update_properties()
+
+    def apply_predation(self):
+        for i, predator in enumerate(self.cells):
+            if predator.cell_type in ["advanced_eukaryote", "plant_like", "complete"]:
+                for prey in self.cells[i+1:]:
+                    if prey.cell_type in ["prokaryote", "early_eukaryote"] and random.random() < 0.05:
+                        predator.energy += prey.energy * 0.5
+                        self.cells.remove(prey)
+
+    def apply_symbiosis(self):
+        for i, cell1 in enumerate(self.cells):
+            for cell2 in self.cells[i+1:]:
+                if cell1.cell_type != cell2.cell_type and random.random() < 0.01:
+                    shared_energy = (cell1.energy + cell2.energy) * 0.1
+                    cell1.energy += shared_energy
+                    cell2.energy += shared_energy
+
     def get_visualization_data(self):
         cell_data = {cell_type: {"x": [], "y": [], "size": []} for cell_type in self.population_history.keys()}
         colors = {"prokaryote": "lightblue", "early_eukaryote": "green", "advanced_eukaryote": "red", "plant_like": "darkgreen", "complete": "purple"}
@@ -198,8 +233,20 @@ def setup_figure(env):
 # Streamlit app
 st.title("Advanced Cell Evolution Simulation")
 
-initial_cells = st.slider("Initial number of cells", 10, 200, 100)
-update_interval = st.slider("Update interval (seconds)", 0.1, 5.0, 1.0)
+initial_cells = st.slider("Initial number of cells", 10, 500, 200)
+update_interval = st.slider("Update interval (seconds)", 0.01, 1.0, 0.05)  # Increased speed
+
+# Effect options
+st.sidebar.header("Environmental Effects")
+radiation = st.sidebar.checkbox("Radiation")
+predation = st.sidebar.checkbox("Predation")
+symbiosis = st.sidebar.checkbox("Symbiosis")
+
+effects = {
+    "radiation": radiation,
+    "predation": predation,
+    "symbiosis": symbiosis
+}
 
 # Create placeholders for the chart and stop button
 chart_placeholder = st.empty()
@@ -215,7 +262,7 @@ def toggle_simulation():
 start_stop_button = st.button("Start/Stop Simulation", on_click=toggle_simulation)
 
 if st.session_state.running:
-    env = Environment(100, 100)
+    env = Environment(100, 100, effects)
     for _ in range(initial_cells):
         cell = Cell(random.uniform(0, env.width), random.uniform(0, env.height))
         env.add_cell(cell)
@@ -223,7 +270,8 @@ if st.session_state.running:
     fig = setup_figure(env)
     
     while st.session_state.running:
-        env.update()
+        for _ in range(4):  # Update 4 times per frame to increase simulation speed
+            env.update()
         
         with fig.batch_update():
             cell_data, population_history, colors = env.get_visualization_data()