Update app.py

app.py

@@ -156,10 +156,58 @@ if left_column.button('Pause' if not st.session_state['paused'] else 'Play'):
 
 # Placeholder for the Plotly chart in the right column
 
-markdonw_position = right_column.markdown(f"```\n{positions_text}\n```")
+# Implement the pause/play functionality.
+if left_column.button('Play' if st.session_state['paused'] else 'Pause'):
+    st.session_state['paused'] = not st.session_state['paused']
 
+    
 chart_placeholder = right_column.empty()
 
+markdonw_position = right_column.markdown(f"```\n{positions_text}\n```")
+
+
+
+
+
+# Placeholder for displaying additional data below the chart.
+data_placeholder = right_column.empty()
+
+# Function to update and display the plot based on the current frame.
+def update_plot(frame_idx):
+    row = positions_df.iloc[frame_idx]
+    positions = row['positions']
+    data = [{'Agent': agent, 'x': pos[0], 'y': pos[1]} for agent, pos in positions.items()]
+    df = pd.DataFrame(data)
+    fig = px.scatter(df, x='x', y='y', color='Agent', title="Agent Positions", range_x=[-5,5], range_y=[-5,5])
+    fig.update_traces(marker=dict(size=20))
+    chart_placeholder.plotly_chart(fig, use_container_width=True)
+
+    # Update data below the chart.
+    additional_data = f"**Agent Interaction:** {row['Agent']} says, \"{row['Dialog']}\""
+    st.session_state['interaction_log'] = st.session_state.get('interaction_log', '') + additional_data + "\n\n"
+    data_placeholder.markdown(st.session_state['interaction_log'])
+
+# Initially display the plot.
+update_plot(st.session_state['current_frame'])
+
+# Periodic update if not paused.
+while True:
+    if not st.session_state['paused']:
+        st.session_state['current_frame'] = (st.session_state['current_frame'] + 1) % len(positions_df)
+        update_plot(st.session_state['current_frame'])
+        time.sleep(2)  # Adjust the sleep time as needed for your animation speed.
+        st.experimental_rerun()
+
+    time.sleep(0.1)  # This short sleep prevents the loop from locking up the app
+
+
+
+
+
+
+
+"""
+
 # Placeholder for displaying the data below the animation
 data_placeholder = right_column.empty()
 
@@ -198,4 +246,5 @@ while True:
         current_frame += 1
 
     # Pause for a moment before updating the chart with new positions
-    time.sleep(2)
+    time.sleep(2)
+    """