Fix special str and value type for web-ui (#5632)

* fix special str display for chatbot

* fix value type for gr.Slider

Author: slin000111

swift/ui/llm_infer/llm_infer.py

@@ -344,6 +344,18 @@ def agent_type(cls, response):
             return 'toolbench'
         return None
 
+    @classmethod
+    def parse_text(cls, messages):
+        prepared_msgs = []
+        for message in messages:
+            if isinstance(message, tuple):
+                query = message[0].replace('<', '&lt;').replace('>', '&gt;').replace('*', '&ast;')
+                response = message[1].replace('<', '&lt;').replace('>', '&gt;').replace('*', '&ast;')
+                prepared_msgs.append((query, response))
+            else:
+                prepared_msgs.append(message)
+        return prepared_msgs
+
     @classmethod
     def send_message(cls, running_task, template_type, prompt: str, image, video, audio, infer_request: InferRequest,
                      infer_model_type, system, max_new_tokens, temperature, top_k, top_p, repetition_penalty):
@@ -367,8 +379,10 @@ def send_message(cls, running_task, template_type, prompt: str, image, video, au
                 infer_request.messages[-1]['medias'].append(media)
 
         if not prompt:
-            yield '', cls._replace_tag_with_media(infer_request), gr.update(value=None), gr.update(
-                value=None), gr.update(value=None), infer_request
+            chatbot_content = cls._replace_tag_with_media(infer_request)
+            chatbot_content = cls.parse_text(chatbot_content)
+            yield '', chatbot_content, gr.update(value=None), gr.update(value=None), gr.update(
+                value=None), infer_request
             return
         else:
             infer_request.messages[-1]['content'] = infer_request.messages[-1]['content'] + prompt
@@ -407,5 +421,7 @@ def send_message(cls, running_task, template_type, prompt: str, image, video, au
                 continue
             stream_resp_with_history += chunk.choices[0].delta.content if chat else chunk.choices[0].text
             infer_request.messages[-1]['content'] = stream_resp_with_history
-            yield '', cls._replace_tag_with_media(infer_request), gr.update(value=None), gr.update(
-                value=None), gr.update(value=None), infer_request
+            chatbot_content = cls._replace_tag_with_media(infer_request)
+            chatbot_content = cls.parse_text(chatbot_content)
+            yield '', chatbot_content, gr.update(value=None), gr.update(value=None), gr.update(
+                value=None), infer_request

swift/ui/llm_infer/runtime.py

@@ -1,6 +1,7 @@
 # Copyright (c) Alibaba, Inc. and its affiliates.
 import collections
 import os.path
+import re
 import subprocess
 import sys
 import time
@@ -274,6 +275,10 @@ def task_changed(cls, task, base_tab):
             if e.elem_id in all_args:
                 if isinstance(e, gr.Dropdown) and e.multiselect:
                     arg = all_args[e.elem_id].split(' ')
+                elif isinstance(e, gr.Slider) and re.fullmatch(cls.int_regex, all_args[e.elem_id]):
+                    arg = int(all_args[e.elem_id])
+                elif isinstance(e, gr.Slider) and re.fullmatch(cls.float_regex, all_args[e.elem_id]):
+                    arg = float(all_args[e.elem_id])
                 else:
                     if e.elem_id == 'model':
                         if is_custom_path: