Update app.py

app.py

@@ -234,6 +234,7 @@ def main():
                 chatbot = gr.Chatbot(
                     label="HunyuanOCR Chat",
                     height=600,
+                    type="messages",   # ✅ explicitly say we are using messages format
                 )
                 user_input = gr.Textbox(
                     label="Your question",
@@ -278,13 +279,13 @@ def main():
             # If nothing to do, return unchanged
             if (not text or not text.strip()) and not img_path:
                 return chat_value, history_msgs, ""
-
+        
             # 1) Build messages with new user turn
             messages = build_hunyuan_messages(history_msgs, text.strip(), img_path)
-
+        
             # 2) Call model
             answer = call_local_model(messages)
-
+        
             # 3) Append assistant turn to history
             messages.append(
                 {
@@ -292,12 +293,10 @@ def main():
                     "content": [{"type": "text", "text": answer}],
                 }
             )
-
-            # 4) Rebuild visible chat (list of (user, assistant) tuples)
-            chat_display = rebuild_chat_display(messages)
-
-            # 5) Return updated chat, messages, and clear textbox
-            return chat_display, messages, ""
+        
+            # 4) Return updated messages both to Chatbot and history_state
+            #    Chatbot (type="messages") expects this format directly
+            return messages, messages, ""
 
         send_btn.click(
             on_send,