Update app.py

app.py

@@ -7,6 +7,8 @@ import time
 import threading
 import queue
 
+
+
 # Your classes and functions...
 
 class Organelle:
@@ -252,21 +254,15 @@ update_interval = st.slider("Update interval (seconds)", 0.1, 5.0, 1.0)
 # Create a placeholder for the chart
 chart_placeholder = st.empty()
 
-# Create a function to run the simulation
-def run_simulation():
+if st.button("Start Simulation"):
     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)
+    
     fig = setup_figure(env)
     
-    while True:
-        try:
-            q.get(timeout=0.1)
-            break
-        except queue.Empty:
-            pass
-        
+    for _ in range(100):  # Run for 100 steps
         env.update()
         
         with fig.batch_update():
@@ -277,42 +273,12 @@ def run_simulation():
                 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
+                fig.data[i+5].y = counts
                 if cell_type != "modified" and cell_type != "plant_like":
-                    fig.data[i+10].y = counts  # Update individual population charts
+                    fig.data[i+10].y = counts
                 else:
                     fig.data[13].y = population_history["plant_like"]
                     fig.data[14].y = population_history["modified"]
         
-        if q.empty():
-            q.put(fig)
-
-# Create a function to start the simulation
-def start_simulation():
-    thread = threading.Thread(target=run_simulation)
-    thread.start()
-
-# Create a function to stop the simulation
-stop_button_pressed = False
-
-def stop_simulation():
-    global stop_button_pressed
-    stop_button_pressed = True
-    q.put(None)
-
-# Create buttons to start and stop the simulation
-start_button = st.button("Start Simulation", on_click=start_simulation)
-stop_button = st.button("Stop Simulation", on_click=stop_simulation)
-
-while True:
-    try:
-        fig = q.get(timeout=0.1)
-        if fig is not None:
-            chart_placeholder.plotly_chart(fig, use_container_width=True)
-        else:
-            break
-    except queue.Empty:
-        time.sleep(0.1)
-        if stop_button_pressed:
-            break
-        continue
+        chart_placeholder.plotly_chart(fig, use_container_width=True)
+        time.sleep(update_interval)