Update app.py

app.py

@@ -44,6 +44,53 @@ user_end = tokenizer.encode_special("<|user_end|>")
 assistant_start = tokenizer.encode_special("<|assistant_start|>")
 assistant_end = tokenizer.encode_special("<|assistant_end|>")
 
+
+
+
+def normalize_history(history):
+    """将 Gradio 传入的 history 统一转换为 [(user, assistant), ...] 格式"""
+    if not history:
+        return []
+    
+    # 处理 messages 格式（列表 of dicts，每个 dict 有 "role" 和 "content"）
+    if isinstance(history[0], dict):
+        pairs = []
+        user_msg = None
+        for msg in history:
+            role = msg.get("role")
+            content = msg.get("content", "")
+            if role == "user":
+                if user_msg is not None:          # 上一条 user 没有 assistant，补 None
+                    pairs.append([user_msg, None])
+                user_msg = content
+            elif role == "assistant":
+                if user_msg is not None:
+                    pairs.append([user_msg, content])
+                    user_msg = None
+                else:                              # 孤立 assistant，补空 user
+                    pairs.append([None, content])
+        if user_msg is not None:                   # 最后一条是 user，无 assistant
+            pairs.append([user_msg, None])
+        return pairs
+    
+    # 处理 tuples 格式（列表 of list/tuple），确保每个元素长度为2
+    normalized = []
+    for turn in history:
+        if isinstance(turn, (list, tuple)):
+            if len(turn) >= 2:
+                normalized.append([turn[0], turn[1]])
+            elif len(turn) == 1:
+                normalized.append([turn[0], None])
+            else:
+                normalized.append([None, None])   # 空轮次，忽略或警告
+        else:
+            # 忽略非列表/元组类型（通常不会出现）
+            continue
+    return normalized
+
+# 在构建 tokens 之前调用
+history = normalize_history(history)
+
 # ---------------------------------------------------------------------------
 # Chat function (streaming)
 # ---------------------------------------------------------------------------