update

utils.py

@@ -178,15 +178,6 @@ def append_to_sheet(user_data=None, custom_row_dict=None, custom_sheet_name=None
         # Read headers from the first row of the sheet
         headers = sheet.row_values(1) if sheet.row_count > 0 else []
 
-    # --- Check for new headers in custom_row_dict and add them if needed ---
-    if custom_row_dict is not None:
-        new_headers = [key for key in custom_row_dict.keys() if key not in headers]
-        if new_headers:
-            # Add new headers to the existing header row
-            headers.extend(new_headers)
-            # Update the first row in the sheet with the new headers
-            sheet.update('1:1', [headers])
-
     # --- Build row aligned to headers ---
     if custom_row_dict is not None:
         # Ensure all values are aligned to headers, fill missing with ""
@@ -256,8 +247,7 @@ def format_chat(response, tool_database_labels):
 
             # Clear after rendering
             last_tool_calls = []
-    # Post-process: replace [FinalAnswer] with "\n***Answer:**\n" in the last message's content if present
-    # Insert "**Rasoning:**\n" at the beginning of the first assistant message if [FinalAnswer] is in the last assistant message
+
     if chat_history:
         last_msg = chat_history[-1]
         if isinstance(last_msg.content, str) and "[FinalAnswer]" in last_msg.content:
@@ -270,4 +260,16 @@ def format_chat(response, tool_database_labels):
         last_msg = chat_history[-1]
         if isinstance(last_msg.content, str) and "[FinalAnswer]" in last_msg.content:
             last_msg.content = last_msg.content.replace("[FinalAnswer]", "\n**Answer:**\n")
-    return chat_history
+
+
+    final_answer_messages = [gr.ChatMessage(role="assistant", content=chat_history[-1].content.split("\n**Answer:**\n")[-1].strip())]
+    assistant_count = sum(1 for msg in chat_history if msg.role == "assistant")
+    if assistant_count == 1:
+        # If only one assistant message, show "No reasoning conducted."
+        reasoning_messages = [gr.ChatMessage(role="assistant", content="No reasoning was conducted.")]
+    else:
+        # Include ALL messages in reasoning_messages, including the last one
+        reasoning_messages = chat_history.copy()
+
+
+    return final_answer_messages, reasoning_messages, chat_history