Update app.py

app.py

@@ -74,7 +74,7 @@ button.pulse:hover {
 with gr.Blocks(css=css) as demo:
 
     gr.Markdown("<h1 style='text-align:center;color:#e0e0e0;'>Lambdai-v1-1B Chat</h1>")
-    chatbot = gr.Chatbot(elem_id="chatbot", height=480, render_markdown=True)
+    chatbot = gr.Chatbot(elem_id="chatbot", height=480, render_markdown=True, type='messages')
 
     with gr.Row():
         system_message = gr.Textbox(value="You are a helpful assistant.", label="System message", lines=1)
@@ -102,7 +102,7 @@ with gr.Blocks(css=css) as demo:
             f"Thinking… generating reasoning path…"
             f"</div>"
         )
-        yield chat_history + [[message, thinking_html]]
+        yield chat_history + [{"role": "user", "content": message}, {"role": "assistant", "content": thinking_html}]
 
         # Prepara payload para API
         messages = [{"role": "system", "content": system_message}]
@@ -122,7 +122,7 @@ with gr.Blocks(css=css) as demo:
         ):
             delta = chunk.choices[0].delta.content or ""
             response += delta
-            yield chat_history + [[message, response]]
+            yield chat_history + [{"role": "user", "content": message}, {"role": "assistant", "content": response}]
 
     send_button.click(
         fn=respond,