Update web_ui.py

web_ui.py

@@ -17,13 +17,11 @@ class WebUI:
 
     def __init__(self, agent: Union[Agent, MultiAgentHub, List[Agent]], chatbot_config: Optional[dict] = None):
         """
-        Initialization the chatbot.
+        Initialize the chatbot.
 
         Args:
-            agent: The agent or a list of agents,
-                supports various types of agents such as Assistant, GroupChat, Router, etc.
-            chatbot_config: The chatbot configuration.
-                Set the configuration as {'user.name': '', 'user.avatar': '', 'agent.avatar': '', 'input.placeholder': '', 'prompt.suggestions': []}.
+            agent: The agent or a list of agents, supports various types of agents such as Assistant, GroupChat, Router, etc.
+            chatbot_config: The chatbot configuration. Set the configuration as {'user.name': '', 'user.avatar': '', 'agent.avatar': '', 'input.placeholder': '', 'prompt.suggestions': []}.
         """
         chatbot_config = chatbot_config or {}
 
@@ -59,13 +57,6 @@ class WebUI:
         self.prompt_suggestions = chatbot_config.get('prompt.suggestions', [])
         self.verbose = chatbot_config.get('verbose', False)
 
-    """
-    Run the chatbot.
-
-    Args:
-        messages: The chat history.
-    """
-
     def run(self,
             messages: List[Message] = None,
             share: bool = False,
@@ -78,14 +69,14 @@ class WebUI:
 
         from qwen_agent.gui.gradio import gr, mgr
 
-        customTheme = gr.themes.Default(
+        custom_theme = gr.themes.Default(
             primary_hue=gr.themes.utils.colors.blue,
             radius_size=gr.themes.utils.sizes.radius_none,
         )
 
         with gr.Blocks(
                 css=os.path.join(os.path.dirname(__file__), 'assets/appBot.css'),
-                theme=customTheme,
+                theme=custom_theme,
         ) as demo:
             history = gr.State([])
 
@@ -130,7 +121,7 @@ class WebUI:
                                               'display': True
                                           }])
 
-                    input = mgr.MultimodalInput(placeholder=self.input_placeholder, upload_button_props=dict(file_types=[".pdf", ".docx", ".pptx", ".txt", ".html", ".csv", ".tsv", ".xlsx", ".xls"] + ["." + file_type for file_type in common_programming_language_extensions]))
+                    input_component = mgr.MultimodalInput(placeholder=self.input_placeholder, upload_button_props=dict(file_types=[".pdf", ".docx", ".pptx", ".txt", ".html", ".csv", ".tsv", ".xlsx", ".xls"] + ["." + file_type for file_type in common_programming_language_extensions]))
 
                 with gr.Column(scale=1):
                     if len(self.agent_list) > 1:
@@ -144,27 +135,25 @@ class WebUI:
 
                     agent_info_block = self._create_agent_info_block()
 
-                    # agent_plugins_block = self._create_agent_plugins_block()
-
                     if self.prompt_suggestions:
                         gr.Examples(
                             label='推荐对话',
                             examples=self.prompt_suggestions,
-                            inputs=[input],
+                            inputs=[input_component],
                         )
 
                 if len(self.agent_list) > 1:
                     agent_selector.change(
                         fn=self.change_agent,
                         inputs=[agent_selector],
-                        outputs=[agent_selector, agent_info_block, agent_plugins_block],
+                        outputs=[agent_selector, agent_info_block, agent_info_block],  # Assuming you want to update this block too
                         queue=False,
                     )
 
-                input_promise = input.submit(
+                input_promise = input_component.submit(
                     fn=self.add_text,
-                    inputs=[input, chatbot, history],
-                    outputs=[input, chatbot, history],
+                    inputs=[input_component, chatbot, history],
+                    outputs=[input_component, chatbot, history],
                     queue=False,
                 )
 
@@ -185,7 +174,7 @@ class WebUI:
                         [chatbot, history],
                     )
 
-                input_promise.then(self.flushed, None, [input])
+                input_promise.then(self.flushed, None, [input_component])
 
             demo.load(None)
 
@@ -194,22 +183,19 @@ class WebUI:
                                                                        server_port=server_port)
 
     def change_agent(self, agent_selector):
-        yield agent_selector, self._create_agent_info_block(agent_selector), self._create_agent_plugins_block(
-            agent_selector)
+        yield agent_selector, self._create_agent_info_block(agent_selector), self._create_agent_plugins_block(agent_selector)
 
     def add_text(self, _input, _chatbot, _history):
         from qwen_agent.gui.gradio import gr
         if _input.text == "/clear":
             _chatbot = []
             _history.clear()
-            yield gr.update(interactive=False, value=""), _chatbot, _history
+            yield gr.update(interactive=False, value="")
             return
 
         _history.append({
             ROLE: USER,
-            CONTENT: [{
-                'text': _input.text
-            }],
+            CONTENT: [{'text': _input.text}],
         })
 
         if self.user_config[NAME]:
@@ -250,7 +236,6 @@ class WebUI:
                 yield _chatbot, _history
             return
 
-
         if self.verbose:
             logger.info('agent_run input:\n' + pprint.pformat(_history, indent=2))
 
@@ -262,25 +247,21 @@ class WebUI:
         if self.agent_hub:
             agent_runner = self.agent_hub
         responses = []
-        for responses in agent_runner.run(_history, **self.run_kwargs):
-            # usage = responses.usage
-            # responses = [Message(ASSISTANT, responses.output.choices[0].message.content)]
 
-            if not responses:
+        for response in agent_runner.run(_history, **self.run_kwargs):
+            if not response:
                 continue
-            if responses[-1][CONTENT] == PENDING_USER_INPUT:
+            if response[-1][CONTENT] == PENDING_USER_INPUT:
                 logger.info('Interrupted. Waiting for user input!')
                 break
 
-            display_responses = convert_fncall_to_text(responses)
-            # display_responses[-1][CONTENT] += "\n<summary>" + repr({"usage": usage}) + "</summary>"
+            display_responses = convert_fncall_to_text(response)
             if not display_responses:
                 continue
             if display_responses[-1][CONTENT] is None:
                 continue
 
             while len(display_responses) > num_output_bubbles:
-                # Create a new chat bubble
                 _chatbot.append([None, None])
                 _chatbot[-1][1] = [None for _ in range(len(self.agent_list))]
                 num_output_bubbles += 1
@@ -315,7 +296,6 @@ class WebUI:
 
     def flushed(self):
         from qwen_agent.gui.gradio import gr
-
         return gr.update(interactive=True)
 
     def _get_agent_index_by_name(self, agent_name):
@@ -357,11 +337,10 @@ class WebUI:
                 choices=capabilities,
                 interactive=False,
             )
-
         else:
             return gr.CheckboxGroup(
                 label='插件',
                 value=[],
                 choices=[],
                 interactive=False,
-            )
+            )