port game of life to streamlit

Author: ankitk50

examples/conways_game_of_life/app.py

@@ -19,7 +19,6 @@
 import random
 
 model = st.title('Boltzman Wealth Model')
-# num_agents= st.slider('Choose how many agents to include in the model',min_value=1,max_value=100,value=50)
 num_ticks = st.slider('Select number of Simulation Runs',min_value=1,max_value=100,value=50)
 height = st.slider('Select Grid Height',min_value=10,max_value=100,step=10,value=15)
 width = st.slider('Select Grid Width',min_value=10,max_value=100,step=10,value=20)
@@ -37,7 +36,6 @@
     step =0
     # init grid
     df_grid= pd.DataFrame()
-    # df_gini = pd.DataFrame({'step':[0],'gini':[-1]})
     agent_counts = np.zeros((model.grid.width, model.grid.height))
     for x in range(width):
         for y in range(height):
@@ -53,34 +51,27 @@
     my_bar = st.progress(0, text='Simulation Progress') # progress
     placeholder = st.empty()
     st.subheader('Agent Grid')
-    chart = st.altair_chart(heatmap)
+    chart = st.altair_chart(heatmap,use_container_width=True)
     color_scale = alt.Scale(domain=[0,1],
-                        range=['red', 'green'])
+                        range=['red', 'yellow'])
     for i in range(num_ticks):
         model.step()
         my_bar.progress((i/num_ticks), text='Simulation progress')
         placeholder.text('Step = %d' % i)
-        for cell in model.grid.coord_iter():
-            cell_content, x, y = cell
+        for contents, x,y in model.grid.coord_iter():
+            # print('x:',x,'y:',y, 'state:',contents)
             selected_row = df_grid[(df_grid['x'] == x) & (df_grid['y'] == y)]
-            df_grid.loc[selected_row.index, 'state'] = cell.state #random.choice([1,2])
-        
-        # df_gini = pd.concat([df_gini , pd.DataFrame({'step':[i],'gini':[model.datacollector.model_vars['Gini'][i]]})])
-        # st.table(df_grid)
+            df_grid.loc[selected_row.index, 'state'] = contents.state #random.choice([1,2])
+
         heatmap = alt.Chart(df_grid).mark_circle(size=100).encode(
             x='x',
             y ='y',
             color = alt.Color('state',scale=color_scale)).interactive().properties(width=800, height=600)
         chart.altair_chart(heatmap)
 
-        time.sleep(0.01)
+        time.sleep(0.1)
 
 
     tock = time.time()
     st.success('Simulation completed in {:.2f} secs'.format(tock-tick))
-    
-    # st.subheader('Agent Grid')
-    # fig = px.imshow(agent_counts,labels={'color':'Agent Count'})
-    # st.plotly_chart(fig)
-    # st.subheader('Gini value over sim ticks (Plotly)')
-    # chart = st.line_chart(model.datacollector.model_vars['Gini'])
+