Update app.py

app.py

@@ -213,9 +213,7 @@ def setup_figure(env):
 
     # Total population over time
     for cell_type, counts in env.population_history.items():
-        fig.add_trace(go.Scatter(y=counts, mode='lines', name=cell_type), row=1, col=2)
-
-    # Individual population charts
+        fig.add_trace(go.Scatter(y=counts, mode='lines', name=cell_type), row=1, col=2)# Individual population charts
     for i, cell_type in enumerate(cell_types):
         if cell_type == "modified":
             fig.add_trace(go.Scatter(y=env.population_history[cell_type], mode='lines', 
@@ -302,6 +300,6 @@ while True:
     try:
         fig = q.get(timeout=0.1)
         chart_placeholder.plotly_chart(fig, use_container_width=True)
-except queue.Empty:
+    except queue.Empty:
         time.sleep(0.1)
-    continue
+        continue