Update ui/ui_core.py

ui/ui_core.py

@@ -14,7 +14,16 @@ def sanitize_utf8(text: str) -> str:
     return text.encode("utf-8", "ignore").decode("utf-8")
 
 def clean_final_response(text: str) -> str:
-    return text.replace("[TOOL_CALLS]", "").strip()
+    cleaned = text.replace("[TOOL_CALLS]", "").strip()
+    sections = cleaned.split("[Final Analysis]")
+    if len(sections) > 1:
+        return (
+            f"<div style='padding:1em;border:1px solid #ccc;border-radius:12px;margin-bottom:1em;'>"
+            f"<h3 style='margin-top:0;'>Summary</h3><p>{sections[0].strip()}</p></div>"
+            f"<div style='padding:1em;border:2px solid #4B4CED;background:#f9f9ff;border-radius:12px;'>"
+            f"<h3 style='margin-top:0;'>🧠 Final Analysis</h3><p>{sections[1].strip()}</p></div>"
+        )
+    return f"<div style='padding:1em;border:1px solid #ccc;border-radius:12px;'><p>{cleaned}</p></div>"
 
 def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, total=1) -> str:
     try: