Update app.py

app.py

@@ -23,7 +23,7 @@ CSS = """
 
 def format_message(content):
     if not content:
-        return ""
+        return gr.HTML("")
     
     # 查找 <think> 标签并替换为折叠内容
     think_pattern = re.compile(r"<think>(.*?)</think>", re.DOTALL)
@@ -46,8 +46,10 @@ def format_message(content):
     # 去掉 </details> 后面多余的 <br>
     formatted_content = re.sub(r"(</details>)(<br>)+", r"\1", formatted_content)
     
-    # 返回纯字符串
-    return formatted_content
+    # 返回 gr.HTML 对象以确保正确渲染
+    return gr.HTML(formatted_content)
+
+
 
 def user_input(user_message, history):
     if history is None:
@@ -90,26 +92,31 @@ def predict(message, chat_history, system_msg, temperature, top_p, repetition_pe
     )
     
     buffer = ""
-    history = chat_history.copy()
+    history = chat_history.copy() + [[message, None]]  # 初始化新对话
     
-    # 处理流式响应
+    # 流式处理关键修改点
     for chunk in response.iter_lines():
         if chunk:
             chunk_str = chunk.decode("utf-8").replace("data: ", "")
             try:
                 chunk_data = json.loads(chunk_str)
                 if "choices" in chunk_data:
-                    delta = chunk_data["choices"][0]["delta"]
+                    delta = chunk_data["choices"][0].get("delta", {})
                     if "content" in delta:
+                        # 增量更新缓冲区
                         buffer += delta["content"]
-                        # 更新 AI 的响应
-                        history[-1][1] = format_message(buffer)
-                        yield history  # 每次更新后返回历史记录
+                        
+                        # 只更新当前消息的最新状态
+                        new_history = history.copy()
+                        new_history[-1][1] = format_message(buffer).value  # 提取 HTML 字符串
+                        yield new_history
             except Exception as e:
                 print(f"Error in processing chunk: {e}")
                 continue
     
-    return history
+    # 最终确认更新
+    history[-1][1] = format_message(buffer).value
+    yield history
 
 # JavaScript to ensure the details tag is working properly
 js_script = """
@@ -230,7 +237,7 @@ NPC只有四个，分别是:有狐娘，蜘蛛娘，猫娘，狼女。
             with gr.Accordion("高级设置", open=False):
                 temperature = gr.Slider(0, 2, value=0.7, label="温度")
                 top_p = gr.Slider(0, 1, value=0.7, label="Top-p")
-                repetition_penalty = gr.Slider(1, 2, value=1.17, label="重复惩罚")
+                repetition_penalty = gr.Slider(1, 2, value=1.1, label="重复惩罚")
 
     # Ensure that the user input is submitted correctly
     msg.submit(