Update apps/agents/agents.py

apps/agents/agents.py

@@ -24,15 +24,14 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 
 import gradio as gr
 import openai
-from openai import RateLimitError, APIError
-
+import openai.error
 import tenacity
 
 from apps.agents.text_utils import split_markdown_code
 from camel.agents import TaskSpecifyAgent
 from camel.messages import BaseMessage
 from camel.societies import RolePlaying
-from camel.types import TaskType
+from camel.typing import TaskType
 
 REPO_ROOT = os.path.realpath(
     os.path.join(os.path.dirname(os.path.abspath(__file__)), "../.."))
@@ -197,8 +196,8 @@ def role_playing_start(
             extend_task_specify_meta_dict=meta_dict,
             output_language=language,
         )
-    except (RateLimitError, tenacity.RetryError,
-            RuntimeError, APIError) as ex:
+    except (openai.error.RateLimitError, tenacity.RetryError,
+            RuntimeError) as ex:
         print("OpenAI API exception 0 " + str(ex))
         return (state, str(ex), "", [], gr.update())
 
@@ -247,9 +246,9 @@ def role_playing_chat_init(state) -> \
 
     try:
         init_assistant_msg: BaseMessage
-        init_assistant_msg = session.init_chat()
-    except (RateLimitError, tenacity.RetryError,
-            RuntimeError, APIError) as ex:
+        init_assistant_msg, _ = session.init_chat()
+    except (openai.error.RateLimitError, tenacity.RetryError,
+            RuntimeError) as ex:
         print("OpenAI API exception 1 " + str(ex))
         state.session = None
         return state, state.chat, gr.update()
@@ -290,8 +289,8 @@ def role_playing_chat_cont(state) -> \
     try:
         assistant_response, user_response = session.step(
             state.saved_assistant_msg)
-    except (RateLimitError, tenacity.RetryError,
-            RuntimeError, APIError) as ex:
+    except (openai.error.RateLimitError, tenacity.RetryError,
+            RuntimeError) as ex:
         print("OpenAI API exception 2 " + str(ex))
         state.session = None
         return state, state.chat, gr.update(), gr.update()
@@ -400,16 +399,14 @@ 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,
-                                       allow_custom_value=True)
+                                       interactive=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,
-                                  allow_custom_value=True)
+                                  interactive=True)
             user_ta = gr.TextArea(label="User role (EDIT ME)", lines=1,
                                   interactive=True)
         with gr.Column(scale=2):
@@ -419,7 +416,8 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
                 "Github repo: [https://github.com/lightaime/camel]"
                 "(https://github.com/lightaime/camel)"
                 '<div style="display:flex; justify-content:center;">'
-                '<img src="https://raw.githubusercontent.com/camel-ai/camel/master/misc/primary_logo.png" alt="Logo" style="max-width:50%;">'
+                '<img src="https://raw.githubusercontent.com/lightaime/camel/'
+                'master/misc/logo.png" alt="Logo" style="max-width:50%;">'
                 '</div>')
     with gr.Row():
         with gr.Column(scale=9):
@@ -460,14 +458,7 @@ 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)
-    with gr.Row():
-        chatbot = gr.Chatbot(
-            [],
-            elem_id="chatbot",
-            height=750,
-            show_label=False,
-            type="messages"  # Use new message format
-        )
+    chatbot = gr.Chatbot(label="Chat between autonomous agents")
     empty_state = State.empty()
     session_state: gr.State = gr.State(empty_state)
 
@@ -487,10 +478,10 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
                    chatbot, progress_sl],
                   queue=False) \
             .then(role_playing_chat_init, session_state,
-                  [session_state, chatbot, progress_sl], queue=False) \
-            .then(role_playing_chat_cont, session_state,
-                  [session_state, chatbot, progress_sl, start_bn],
-                  every=1)
+                  [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)
 
     clear_bn.click(stop_session, session_state,
                    [session_state, progress_sl, start_bn])