Update app.py

app.py

@@ -313,18 +313,44 @@ def generate_reply_stream(ids: list[int], max_new_tokens=150, temperature=0.8, t
         states = out.past_key_values
         logits = out.logits[0, -1]
 
+def extract_text_content(content) -> str:
+    """Safely extract plain text from string, list, or dict content structures returned by Gradio."""
+    if isinstance(content, str):
+        return content
+    if isinstance(content, list):
+        parts = []
+        for item in content:
+            if isinstance(item, str):
+                parts.append(item)
+            elif isinstance(item, dict):
+                if "text" in item:
+                    parts.append(str(item["text"]))
+                elif "content" in item:
+                    parts.append(extract_text_content(item["content"]))
+            else:
+                parts.append(str(item))
+        return " ".join(parts)
+    if isinstance(content, dict):
+        if "text" in content:
+            return str(content["text"])
+        return str(content)
+    return str(content) if content is not None else ""
+
 def chat_function(message, history):
     """Gradio ChatInterface streaming handler formatted with speed stats (Live, Avg, Min, Max)."""
     messages = []
     for turn in history:
         if isinstance(turn, (list, tuple)):
             user_msg, asst_msg = turn
-            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "user", "content": extract_text_content(user_msg)})
             if asst_msg:
-                messages.append({"role": "assistant", "content": asst_msg})
+                messages.append({"role": "assistant", "content": extract_text_content(asst_msg)})
         elif isinstance(turn, dict):
-            messages.append(turn)
-    messages.append({"role": "user", "content": message})
+            messages.append({
+                "role": turn.get("role", "user"),
+                "content": extract_text_content(turn.get("content", ""))
+            })
+    messages.append({"role": "user", "content": extract_text_content(message)})
 
     # Encode token sequence according to model chat template
     user_id = tokenizer.convert_tokens_to_ids(USER_TOKEN)
@@ -334,6 +360,8 @@ def chat_function(message, history):
     for turn in messages:
         role = turn["role"]
         content = turn["content"]
+        if not content.strip():
+            continue
         content_ids = tokenizer.encode(" " + content)
         if role == "user":
             ids += [user_id] + content_ids