Update app.py

app.py

@@ -71,7 +71,6 @@ def user_input(user_message, history):
     return "", history
 
 def predict(message, chat_history, system_msg, temperature, top_p, repetition_penalty):
-    """处理AI回复"""
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {API_KEY}"
@@ -79,13 +78,10 @@ def predict(message, chat_history, system_msg, temperature, top_p, repetition_pe
     
     if chat_history is None:
         chat_history = []
-        
-    # 不需要再添加用户消息,因为已经在user_input中添加了
-    history = chat_history.copy()
     
     # 构造对话历史
     messages = [{"role": "system", "content": system_msg}]
-    for msg in history:
+    for msg in chat_history:
         messages.append({"role": "user", "content": msg[0]})
         if msg[1]:
             messages.append({"role": "assistant", "content": msg[1]})
@@ -107,21 +103,22 @@ def predict(message, chat_history, system_msg, temperature, top_p, repetition_pe
     )
     
     buffer = ""
-    first_chunk = True
     
-    # 流式处理
+    # 流式处理关键修改点
     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].get("delta", {})
                     if "content" in delta:
+                        # 增量更新缓冲区
                         buffer += delta["content"]
                         
-                        # 更新最后一条消息的回复
+                        # 只更新当前消息的最新状态
                         new_history = history.copy()
-                        new_history[-1][1] = format_message(buffer)
+                        new_history[-1][1] = format_message(buffer)  # 直接使用字符串
                         yield new_history
             except Exception as e:
                 print(f"Error in processing chunk: {e}")
@@ -136,6 +133,7 @@ def predict(message, chat_history, system_msg, temperature, top_p, repetition_pe
 
 
 
+
 # JavaScript to ensure the details tag is working properly
 js_script = """
 <script>