Update app.py

app.py

@@ -188,9 +188,9 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
     update_session_title_if_needed(sid, user_text)
 
     api_messages = build_api_messages(sid, system_message)
-    display_msgs = get_messages(sid)
+    display_msgs = chatbot_msgs[:]  # Use existing chat content
 
-    # Add initial empty assistant message for typing indicator
+    # Add initial assistant message for typing
     display_msgs.append({"role": "assistant", "content": "…"})
     yield (display_msgs, "", selected_label)
 
@@ -211,11 +211,11 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
                 yield (display_msgs, "", selected_label)
         add_message(sid, "assistant", partial)
     except Exception as e:
-        err = f"⚠️ Error: {str(e)}"
-        display_msgs[-1]["content"] = err
+        display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
         yield (display_msgs, "", selected_label)
 
 
+
 def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice):
     sid = label_to_id(selected_label)
     if sid is None: