Update apps/agents/agents.py

apps/agents/agents.py

@@ -457,7 +457,14 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
         " based on the original (simplistic) idea", lines=1, interactive=False)
     task_prompt_ta = gr.TextArea(label="Planned task prompt", lines=1,
                                  interactive=False, visible=False)
-    chatbot = gr.Chatbot(label="Chat between autonomous agents")
+    with gr.Row():
+        chatbot = gr.Chatbot(
+            [],
+            elem_id="chatbot",
+            height=750,
+            show_label=False,
+            type="messages"  # Use new message format
+        )
     empty_state = State.empty()
     session_state: gr.State = gr.State(empty_state)
 
@@ -479,8 +486,9 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
             .then(role_playing_chat_init, session_state,
                   [session_state, chatbot, progress_sl], queue=False)
 
-    blocks.load(role_playing_chat_cont, session_state,
-                [session_state, chatbot, progress_sl, start_bn], every=0.5)
+    demo = blocks.every(0.5, role_playing_chat_cont, 
+                       inputs=[session_state],
+                       outputs=[session_state, chatbot, progress_sl, start_bn])
 
     clear_bn.click(stop_session, session_state,
                    [session_state, progress_sl, start_bn])