Update ui/ui_core.py

ui/ui_core.py

@@ -15,17 +15,22 @@ def sanitize_utf8(text: str) -> str:
 
 def clean_final_response(text: str) -> str:
     cleaned = text.replace("[TOOL_CALLS]", "").strip()
-    sections = cleaned.split("[Final Analysis]")
-    if len(sections) > 1:
-        final = sections[1].strip()
-        formatted_final = final.replace("\n", "</li><li style='margin-bottom:0.75em;'>")
-        return (
-            "<div style='padding:1em;border:2px solid #4B4CED;background:#242C3B;border-radius:12px;color:#fff;'>"
-            "<h3 style='margin-top:0;color:#37B6E9;'>🧠 Final Analysis</h3>"
-            f"<ul><li style='margin-bottom:0.75em;'>{formatted_final}</li></ul>"
-            "</div>"
+    responses = cleaned.split("[Final Analysis]")
+
+    if len(responses) <= 1:
+        return f"<div style='padding:1em;border:1px solid #ccc;border-radius:12px;color:#fff;background:#353F54;'><p>{cleaned}</p></div>"
+
+    # Support multiple [Final Analysis] sections
+    panels = []
+    for i, section in enumerate(responses[1:], 1):
+        final = section.strip()
+        panels.append(
+            f"<details style='background:#242C3B;color:#fff;border-radius:12px;margin-bottom:1em;border:1px solid #4B4CED;'>"
+            f"<summary style='font-size:1.1em;font-weight:bold;padding:0.75em;background:#4B4CED;color:#fff;border-radius:12px 12px 0 0;'>🧠 Final Analysis #{i}</summary>"
+            f"<div style='padding:1em;line-height:1.6;'>{final.replace(chr(10), '<br>')}</div>"
+            f"</details>"
         )
-    return f"<div style='padding:1em;border:1px solid #ccc;border-radius:12px;color:#fff;background:#353F54;'><p>{cleaned}</p></div>"
+    return "".join(panels)
 
 def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, total=1) -> str:
     try: