Update ui/ui_core.py

ui/ui_core.py

@@ -3,6 +3,7 @@ import os
 import pandas as pd
 import pdfplumber
 import gradio as gr
+from tabulate import tabulate
 
 # ✅ Add src to Python path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
@@ -16,9 +17,24 @@ def extract_all_text_from_csv_or_excel(file_path, progress=None, index=0, total=
             df = pd.read_excel(file_path)
         else:
             return f"Unsupported spreadsheet format: {file_path}"
+
         if progress:
             progress((index + 1) / total, desc=f"Processed table: {os.path.basename(file_path)}")
-        return df.to_string(index=False)
+
+        # Group by "Booking Number" or "Form Name" if available
+        if "Booking Number" in df.columns:
+            groups = df.groupby("Booking Number")
+        elif "Form Name" in df.columns:
+            groups = df.groupby("Form Name")
+        else:
+            return tabulate(df, headers="keys", tablefmt="github", showindex=False)
+
+        result = []
+        for group_name, group_df in groups:
+            result.append(f"\n### Group: {group_name}\n")
+            result.append(tabulate(group_df, headers="keys", tablefmt="github", showindex=False))
+        return "\n".join(result)
+
     except Exception as e:
         return f"Error parsing file: {e}"
 
@@ -41,7 +57,7 @@ def extract_all_text_from_pdf(file_path, progress=None, index=0, total=1):
 
 def create_ui(agent: TxAgent):
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("<h1 style='text-align: center;'>\ud83d\udccb CPS: Clinical Patient Support System</h1>")
+        gr.Markdown("<h1 style='text-align: center;'>📋 CPS: Clinical Patient Support System</h1>")
         chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages")
 
         file_upload = gr.File(
@@ -104,4 +120,4 @@ def create_ui(agent: TxAgent):
             ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
-    return demo
+    return demo