Update apps/agents/agents.py

apps/agents/agents.py

@@ -399,14 +399,16 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
             assistant_dd = gr.Dropdown(default_society['assistant_roles'],
                                        label="Example assistant roles",
                                        value=default_society['assistant_role'],
-                                       interactive=True)
+                                       interactive=True,
+                                       allow_custom_value=True)
             assistant_ta = gr.TextArea(label="Assistant role (EDIT ME)",
                                        lines=1, interactive=True)
         with gr.Column(scale=2):
             user_dd = gr.Dropdown(default_society['user_roles'],
                                   label="Example user roles",
                                   value=default_society['user_role'],
-                                  interactive=True)
+                                  interactive=True,
+                                  allow_custom_value=True)
             user_ta = gr.TextArea(label="User role (EDIT ME)", lines=1,
                                   interactive=True)
         with gr.Column(scale=2):
@@ -486,9 +488,16 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
             .then(role_playing_chat_init, session_state,
                   [session_state, chatbot, progress_sl], queue=False)
 
-    demo = blocks.every(0.5, role_playing_chat_cont, 
-                       inputs=[session_state],
-                       outputs=[session_state, chatbot, progress_sl, start_bn])
+    chat_bot = gr.Button("Continue Chat", visible=False)
+    chat_bot.click(
+        role_playing_chat_cont,
+        inputs=[session_state],
+        outputs=[session_state, chatbot, progress_sl, start_bn]
+    ).then(
+        lambda: gr.update(visible=False, value=None)  # Hide the button
+    ).then(
+        lambda: gr.update(visible=True)  # Show the button again after a delay
+    )
 
     clear_bn.click(stop_session, session_state,
                    [session_state, progress_sl, start_bn])