Update app.py

app.py

@@ -31,10 +31,12 @@ def chat_fn(message, history, model, system_message, max_tokens, temperature, to
     history_list = history or []
     response_generator = respond(message, history_list, model, system_message, max_tokens, temperature, top_p, top_k, repeat_penalty)
     
-    full_response = ""
-    for current_history in response_generator:
-        full_response = current_history[-1][1]  # 获取最新的回复
-        yield current_history, history #  在每次迭代中返回当前历史记录和历史记录
+    for messages in response_generator:
+        # 转换 messages 为 Gradio Chatbot 接受的格式
+        chatbot_messages = []
+        for msg in messages:
+            chatbot_messages.append([msg["content"], msg["role"] == "assistant"]) #  直接使用字典中的值
+        yield chatbot_messages, history  #  yield 正确格式的消息列表
 
 
 def respond(