Update ui/ui_core.py

ui/ui_core.py

@@ -20,14 +20,15 @@ def clean_final_response(text: str) -> str:
     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:#1F2633;color:#fff;border-radius:12px;margin-bottom:1em;border:1px solid #37B6E9;'>"
-            f"<summary style='font-size:1.1em;font-weight:bold;padding:0.75em;background:#37B6E9;color:#fff;border-radius:12px 12px 0 0;'>🧠 Final Analysis #{i}</summary>"
+            f"<div style='background:#2B2B2B;color:#E0E0E0;border-radius:12px;margin-bottom:1em;border:1px solid #888;'>"
+            f"<div style='font-size:1.1em;font-weight:bold;padding:0.75em;background:#3A3A3A;color:#fff;border-radius:12px 12px 0 0;'>🧠 Final Analysis #{i}</div>"
             f"<div style='padding:1em;line-height:1.6;'>{final.replace(chr(10), '<br>')}</div>"
-            f"</details>"
+            f"</div>"
         )
     return "".join(panels)
 
@@ -41,21 +42,21 @@ def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, t
 
         df = None
         if file_path.endswith(".csv"):
-            df = pd.read_csv(file_path, encoding_errors="replace")
+            df = pd.read_csv(file_path, encoding_errors="replace", dtype=str, header=None)
         elif file_path.endswith((".xls", ".xlsx")):
-            df = pd.read_excel(file_path, engine=None)
-        else:
-            return f"Unsupported spreadsheet format: {file_path}"
+            try:
+                df = pd.read_excel(file_path, engine="openpyxl", dtype=str, header=None)
+            except:
+                df = pd.read_excel(file_path, engine="xlrd", dtype=str, header=None)
 
         if df is None or df.empty:
-            return f"⚠️ No data found in: {os.path.basename(file_path)}"
+            return f"[Warning] No data extracted from: {file_path}"
 
         lines = []
         for _, row in df.iterrows():
             line = " | ".join(str(cell) for cell in row if pd.notna(cell))
             if line:
                 lines.append(line)
-
         return f"\U0001F4C4 {os.path.basename(file_path)}\n\n" + "\n".join(lines)
 
     except Exception as e:
@@ -99,6 +100,8 @@ def chunk_text(text: str, max_tokens: int = 8192) -> List[str]:
         chunks.append(" ".join(chunk))
     return chunks
 
+# ... rest of the UI code remains unchanged
+
 def create_ui(agent: TxAgent):
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("<h1 style='text-align: center;'>\U0001F4CB CPS: Clinical Patient Support System</h1>")