Update app.py

app.py

@@ -21,14 +21,25 @@ SYSTEM_MESSAGE = {
 }
 
 def chat(user_input, messages):
-    # Initialize messages if empty
+    # Initialize clean messages
     if not messages:
         messages = [SYSTEM_MESSAGE]
 
-    messages.append({"role": "user", "content": user_input})
+    # Safety: ensure all contents are strings
+    clean_messages = []
+    for m in messages:
+        clean_messages.append({
+            "role": m["role"],
+            "content": str(m["content"])
+        })
+
+    clean_messages.append({
+        "role": "user",
+        "content": str(user_input)
+    })
 
     prompt = tokenizer.apply_chat_template(
-        messages,
+        clean_messages,
         tokenize=False,
         add_generation_prompt=True
     )
@@ -48,9 +59,12 @@ def chat(user_input, messages):
         skip_special_tokens=True
     )
 
-    messages.append({"role": "assistant", "content": response})
+    clean_messages.append({
+        "role": "assistant",
+        "content": response
+    })
 
-    return messages, ""
+    return clean_messages, ""
 
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 Qwen Chatbot")
@@ -62,13 +76,7 @@ with gr.Blocks() as demo:
     msg.submit(
         chat,
         inputs=[msg, state],
-        outputs=[chatbot, msg],
-    )
-
-    chatbot.change(
-        lambda x: x,
-        chatbot,
-        state
+        outputs=[chatbot, state]
     )
 
 demo.launch()