Update app.py

app.py

@@ -12,14 +12,17 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 def chat_fn(message, history):
-    # 只保留最近 3 轮历史
-    recent_history = history[-6:]  # 3轮对话，每轮2条消息
+    recent_history = history[-6:]  # 只保留最近 3 轮
     full_prompt = ""
     for msg in recent_history:
+        content = msg['content']
+        if isinstance(content, list):
+            content = " ".join([str(c) for c in content])
         if msg["role"] == "user":
-            full_prompt += f"<|user|>{msg['content']}<|assistant|>"
+            full_prompt += f"<|user|>{content}<|assistant|>"
         elif msg["role"] == "assistant":
-            full_prompt += msg['content']
+            full_prompt += content
+
     full_prompt += f"<|user|>{message}<|assistant|>"
 
     inputs = tokenizer(full_prompt, return_tensors="pt").to(model.device)
@@ -35,6 +38,7 @@ def chat_fn(message, history):
         output_text = output_text.split("<|assistant|>")[-1]
     return output_text.strip()
 
+
 with gr.Blocks(title="caobin LLM Chatbot") as demo:
     gr.Markdown("# 🤖 caobin's AI assistant")
     chatbot = gr.Chatbot(height=450)