Update ui/ui_core.py

ui/ui_core.py

@@ -3,7 +3,6 @@ import os
 import pandas as pd
 import pdfplumber
 import gradio as gr
-import re
 from typing import List
 
 # ✅ Fix: Add src to Python path
@@ -34,7 +33,7 @@ def extract_all_text_from_csv_or_excel(file_path: str, progress=None, index=0, t
             line = " | ".join(str(cell) for cell in row if pd.notna(cell))
             if line:
                 lines.append(line)
-        return f"📄 {os.path.basename(file_path)}\n\n" + "\n".join(lines)
+        return f"\U0001F4C4 {os.path.basename(file_path)}\n\n" + "\n".join(lines)
 
     except Exception as e:
         return f"[Error reading {os.path.basename(file_path)}]: {str(e)}"
@@ -55,7 +54,7 @@ def extract_all_text_from_pdf(file_path: str, progress=None, index=0, total=1) -
                         progress((index + (i / num_pages)) / total, desc=f"Reading PDF: {os.path.basename(file_path)} ({i+1}/{num_pages})")
                 except Exception as e:
                     extracted.append(f"[Error reading page {i+1}]: {str(e)}")
-        return f"📄 {os.path.basename(file_path)}\n\n" + "\n\n".join(extracted)
+        return f"\U0001F4C4 {os.path.basename(file_path)}\n\n" + "\n\n".join(extracted)
 
     except Exception as e:
         return f"[Error reading PDF {os.path.basename(file_path)}]: {str(e)}"
@@ -79,7 +78,7 @@ def chunk_text(text: str, max_tokens: int = 8192) -> List[str]:
 
 def create_ui(agent: TxAgent):
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown("<h1 style='text-align: center;'>📋 CPS: Clinical Patient Support System</h1>")
+        gr.Markdown("<h1 style='text-align: center;'>\U0001F4CB CPS: Clinical Patient Support System</h1>")
 
         chatbot = gr.Chatbot(label="CPS Assistant", height=600, type="messages", show_copy_button=True)
         file_upload = gr.File(
@@ -91,12 +90,6 @@ def create_ui(agent: TxAgent):
         send_button = gr.Button("Send", variant="primary")
         conversation_state = gr.State([])
 
-        # Centered status box styled
-        status_box = gr.Markdown(
-            "<div style='text-align:center; font-size: 18px;'>⏳ Processing in progress... Please wait...</div>",
-            visible=False
-        )
-
         def handle_chat(message: str, history: list, conversation: list, uploaded_files: list, progress=gr.Progress()):
             context = (
                 "You are an expert clinical AI assistant reviewing medical form or interview data. "
@@ -108,10 +101,9 @@ def create_ui(agent: TxAgent):
             )
 
             try:
-                # Show loading centered
-                yield [{"role": "assistant", "content": "⏳ Processing... Please wait while I analyze the files."}]
+                # Show centered loading message
+                yield history + [{"role": "assistant", "content": "<div style='text-align:center'>⏳ Processing... Please wait while I analyze the files.</div>"}]
 
-                # Step 1: Extract files
                 extracted_text = ""
                 if uploaded_files and isinstance(uploaded_files, list):
                     total_files = len(uploaded_files)
@@ -129,9 +121,8 @@ def create_ui(agent: TxAgent):
                         except Exception as file_error:
                             extracted_text += f"[Error processing {os.path.basename(path)}]: {str(file_error)}\n"
 
-                # Step 2: Chunk and sanitize
                 sanitized = sanitize_utf8(extracted_text.strip())
-                chunks = chunk_text(sanitized, max_tokens=8192)
+                chunks = chunk_text(sanitized)
 
                 all_responses = ""
                 for i, chunk in enumerate(chunks):
@@ -154,25 +145,23 @@ def create_ui(agent: TxAgent):
                             all_responses += update
 
                 all_responses = sanitize_utf8(all_responses.strip())
-
-                # Final messages
                 final_history = history + [
                     {"role": "user", "content": message},
                     {"role": "assistant", "content": all_responses}
                 ]
                 yield final_history
 
-            except Exception as e:
-                print(f"[TxAgent Error] {e}")
+            except Exception as chat_error:
+                print(f"Chat error: {chat_error}")
                 final_history = history + [
                     {"role": "user", "content": message},
-                    {"role": "assistant", "content": "⚠️ An error occurred while processing the files."}
+                    {"role": "assistant", "content": "❌ An error occurred while processing your request."}
                 ]
                 yield final_history
 
         inputs = [message_input, chatbot, conversation_state, file_upload]
-        send_button.click(fn=handle_chat, inputs=inputs, outputs=[status_box, chatbot])
-        message_input.submit(fn=handle_chat, inputs=inputs, outputs=[status_box, chatbot])
+        send_button.click(fn=handle_chat, inputs=inputs, outputs=chatbot)
+        message_input.submit(fn=handle_chat, inputs=inputs, outputs=chatbot)
 
         gr.Examples([
             ["Upload your medical form and ask what the doctor might've missed."],
@@ -180,5 +169,4 @@ def create_ui(agent: TxAgent):
             ["Is there anything abnormal in the attached blood work report?"]
         ], inputs=message_input)
 
-    return demo
-
+    return demo