Update ui/ui_core.py

ui/ui_core.py

@@ -6,16 +6,15 @@ import gradio as gr
 from tabulate import tabulate
 from typing import List, Optional
 
-# ✅ Add src to Python path
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src"))
+# ✅ Fix: Add src to Python path with correct parentheses
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
+
 from txagent.txagent import TxAgent
 
 def safe_extract_table_data(table: List[List[str]]) -> List[str]:
-    """Safely extract data from a PDF table row by row."""
     extracted_rows = []
     if not table or not isinstance(table, list):
         return extracted_rows
-    
     for row in table:
         if not row or not isinstance(row, list):
             continue
@@ -29,7 +28,6 @@ def safe_extract_table_data(table: List[List[str]]) -> List[str]:
     return extracted_rows
 
 def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, total=1) -> str:
-    """Safely extract text from CSV or Excel files with error handling."""
     try:
         if not os.path.exists(file_path):
             return f"File not found: {file_path}"
@@ -44,7 +42,6 @@ def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, t
         if progress:
             progress((index + 1) / total, desc=f"Processed table: {os.path.basename(file_path)}")
 
-        # Safely check for grouping columns
         group_column = None
         for col in ["Booking Number", "Form Name"]:
             if col in df.columns:
@@ -71,7 +68,6 @@ def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, t
         return f"Error parsing file {os.path.basename(file_path)}: {str(e)}"
 
 def extract_all_text_from_pdf(file_path: str, progress=None, index=0, total=1) -> str:
-    """Safely extract text from PDF files with comprehensive error handling."""
     extracted = []
     try:
         if not os.path.exists(file_path):
@@ -111,7 +107,6 @@ def create_ui(agent: TxAgent):
         conversation_state = gr.State([])
 
         def handle_chat(message: str, history: list, conversation: list, uploaded_files: list, progress=gr.Progress()):
-            """Handle chat with comprehensive error handling for file processing."""
             context = (
                 "You are an expert clinical AI assistant reviewing medical form or interview data. "
                 "Your job is to analyze this data and reason about any information or red flags that a human doctor might have overlooked. "
@@ -162,10 +157,15 @@ def create_ui(agent: TxAgent):
                 for update in generator:
                     try:
                         if isinstance(update, list):
-                            cleaned = [msg for msg in update if hasattr(msg, 'role') and 
-                                     not (msg.role == "assistant" and 
-                                     hasattr(msg, 'content') and 
-                                     msg.content.strip().startswith("🧰"))]
+                            cleaned = [
+                                msg for msg in update
+                                if hasattr(msg, 'role')
+                                and not (
+                                    msg.role == "assistant"
+                                    and hasattr(msg, 'content')
+                                    and msg.content.strip().startswith("🧰")
+                                )
+                            ]
                             if cleaned:
                                 final_response = cleaned
                                 yield cleaned
@@ -190,4 +190,4 @@ def create_ui(agent: TxAgent):
             ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
-    return demo
+    return demo