Update app.py

app.py

@@ -241,28 +241,26 @@ def format_number(num):
         return str(num)
 
 def update_chart():
-    global fig  # Assuming fig is a global variable
-    
     # Clear existing traces
-    fig.data = []
+    st.session_state.fig.data = []
     
     # 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]
     y_positions = [cell.y for cell in st.session_state.env.cells]
     
-    fig.add_trace(go.Scatter(x=x_positions, y=y_positions, mode='markers', 
+    st.session_state.fig.add_trace(go.Scatter(x=x_positions, y=y_positions, mode='markers', 
                              marker=dict(color=[colors[ct] for ct in cell_types]), 
                              text=cell_types, hoverinfo='text'), row=1, col=1)
 
     # Population history
     for cell_type, counts in st.session_state.env.population_history.items():
-        fig.add_trace(go.Scatter(y=counts, mode='lines', name=cell_type, 
+        st.session_state.fig.add_trace(go.Scatter(y=counts, mode='lines', name=cell_type, 
                                  line=dict(color=colors[cell_type])), row=1, col=2)
 
     # Population by cell type
     for cell_type, counts in st.session_state.env.population_history.items():
-        fig.add_trace(go.Scatter(y=counts, mode='lines', name=cell_type, 
+        st.session_state.fig.add_trace(go.Scatter(y=counts, mode='lines', name=cell_type, 
                                  line=dict(color=colors[cell_type])), row=2, col=1)
 
     # Organelle distribution
@@ -272,66 +270,23 @@ def update_chart():
         for organelle in cell.organelles:
             organelle_counts[organelle] += 1
     
-    fig.add_trace(go.Bar(x=list(organelle_counts.keys()), y=list(organelle_counts.values()), 
+    st.session_state.fig.add_trace(go.Bar(x=list(organelle_counts.keys()), y=list(organelle_counts.values()), 
                          name="Organelles"), row=2, col=2)
 
     # Update axis labels and layout
-    fig.update_xaxes(title_text="X", row=1, col=1)
-    fig.update_yaxes(title_text="Y", row=1, col=1)
-    fig.update_xaxes(title_text="Time", row=1, col=2)
-    fig.update_yaxes(title_text="Population", row=1, col=2)
-    fig.update_xaxes(title_text="Time", row=2, col=1)
-    fig.update_yaxes(title_text="Population", row=2, col=1)
-    fig.update_xaxes(title_text="Organelle", row=2, col=2)
-    fig.update_yaxes(title_text="Count", row=2, col=2)
+    st.session_state.fig.update_xaxes(title_text="X", row=1, col=1)
+    st.session_state.fig.update_yaxes(title_text="Y", row=1, col=1)
+    st.session_state.fig.update_xaxes(title_text="Time", row=1, col=2)
+    st.session_state.fig.update_yaxes(title_text="Population", row=1, col=2)
+    st.session_state.fig.update_xaxes(title_text="Time", row=2, col=1)
+    st.session_state.fig.update_yaxes(title_text="Population", row=2, col=1)
+    st.session_state.fig.update_xaxes(title_text="Organelle", row=2, col=2)
+    st.session_state.fig.update_yaxes(title_text="Count", row=2, col=2)
 
-    fig.update_layout(height=800, width=1200, title_text="Advanced Cell Evolution Simulation")
+    st.session_state.fig.update_layout(height=800, width=1200, title_text="Advanced Cell Evolution Simulation")
 
     # Update the chart placeholder
-    chart_placeholder.plotly_chart(fig)
-
-# Sidebar for controls and live statistics
-st.sidebar.header("Simulation Controls")
-initial_cells = st.sidebar.slider("Initial number of cells", 10, 500, 200)
-update_interval = st.sidebar.slider("Update interval (seconds)", 0.01, 1.0, 0.05)
-
-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
-}
-
-# Live statistics placeholders
-st.sidebar.header("Live Statistics")
-total_cells_text = st.sidebar.empty()
-cell_type_breakdown = st.sidebar.empty()
-dominant_type_text = st.sidebar.empty()
-avg_energy_text = st.sidebar.empty()
-total_merges_text = st.sidebar.empty()
-
-# Event log
-st.sidebar.header("Event Log")
-event_log = deque(maxlen=10)  # Keep the last 10 events
-event_log_text = st.sidebar.empty()
-
-# Create placeholders for the chart
-chart_placeholder = st.empty()
-
-
-
-if 'running' not in st.session_state:
-    st.session_state.running = False
-if 'total_merges' not in st.session_state:
-    st.session_state.total_merges = 0
-if 'env' not in st.session_state:
-    st.session_state.env = None
-if 'fig' not in st.session_state:
-    st.session_state.fig = None
+    chart_placeholder.plotly_chart(st.session_state.fig)
 
 def start_simulation():
     st.session_state.running = True
@@ -340,7 +295,15 @@ def start_simulation():
         for _ in range(initial_cells):
             cell = Cell(random.uniform(0, st.session_state.env.width), random.uniform(0, st.session_state.env.height))
             st.session_state.env.add_cell(cell)
-        st.session_state.fig = setup_figure(st.session_state.env)
+        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'))
 
 def stop_simulation():
     st.session_state.running = False