Update app.py

app.py

@@ -38,7 +38,6 @@ os.environ.update({
 
 from txagent.txagent import TxAgent
 
-# Medical keywords for priority detection
 MEDICAL_KEYWORDS = {
     'diagnosis', 'assessment', 'plan', 'results', 'medications',
     'allergies', 'summary', 'impression', 'findings', 'recommendations'
@@ -59,7 +58,7 @@ def extract_priority_pages(file_path: str, max_pages: int = 20) -> str:
                 text_chunks.append(f"=== Page {i+1} ===\n{(page.extract_text() or '').strip()}")
             for i, page in enumerate(pdf.pages[3:max_pages], start=4):
                 page_text = page.extract_text() or ""
-                if any(re.search(rf'\b{kw}\b', page_text.lower()) for kw in MEDICAL_KEYWORDS):
+                if any(re.search(rf'\\b{kw}\\b', page_text.lower()) for kw in MEDICAL_KEYWORDS):
                     text_chunks.append(f"=== Page {i} ===\n{page_text.strip()}")
         return "\n\n".join(text_chunks)
     except Exception as e:
@@ -69,7 +68,6 @@ def convert_file_to_json(file_path: str, file_type: str) -> str:
     try:
         h = file_hash(file_path)
         cache_path = os.path.join(file_cache_dir, f"{h}.json")
-
         if os.path.exists(cache_path):
             return open(cache_path, "r", encoding="utf-8").read()
 
@@ -151,11 +149,7 @@ def create_ui(agent: TxAgent):
         gr.Markdown("<h3 style='text-align: center;'>Identify potential oversights in patient care</h3>")
 
         chatbot = gr.Chatbot(label="Analysis", height=600, type="messages")
-        file_upload = gr.File(
-            label="Upload Medical Records",
-            file_types=[".pdf", ".csv", ".xls", ".xlsx"],
-            file_count="multiple"
-        )
+        file_upload = gr.File(label="Upload Medical Records", file_types=[".pdf", ".csv", ".xls", ".xlsx"], file_count="multiple")
         msg_input = gr.Textbox(placeholder="Ask about potential oversights...", show_label=False)
         send_btn = gr.Button("Analyze", variant="primary")
         conversation_state = gr.State([])
@@ -163,7 +157,8 @@ def create_ui(agent: TxAgent):
         def analyze_potential_oversights(message: str, history: list, conversation: list, files: list):
             start_time = time.time()
             try:
-                history.append((message, "Analyzing records for potential oversights..."))
+                history.append({"role": "user", "content": message})
+                history.append({"role": "assistant", "content": "Analyzing records for potential oversights..."})
                 yield history
 
                 extracted_data = ""
@@ -172,21 +167,21 @@ def create_ui(agent: TxAgent):
                         futures = [executor.submit(convert_file_to_json, f.name, f.name.split(".")[-1].lower()) for f in files if hasattr(f, 'name')]
                         extracted_data = "\n".join([sanitize_utf8(f.result()) for f in as_completed(futures)])
 
-                analysis_prompt = """Review these medical records and identify EXACTLY what might have been missed:
+                analysis_prompt = f"""Review these medical records and identify EXACTLY what might have been missed:
 1. List potential missed diagnoses
 2. Flag any medication conflicts
 3. Note incomplete assessments
 4. Highlight abnormal results needing follow-up
 
 Medical Records:
-{records}
+{extracted_data[:15000]}
 
 Provide ONLY the potential oversights in this format:
 
 ### Potential Oversights:
 1. [Missed diagnosis] - [Evidence from records]
 2. [Medication issue] - [Supporting data]
-3. [Assessment gap] - [Relevant findings]""".format(records=extracted_data[:15000])
+3. [Assessment gap] - [Relevant findings]"""
 
                 response = []
                 for chunk in agent.run_gradio_chat(
@@ -202,9 +197,8 @@ Provide ONLY the potential oversights in this format:
                         response.append(chunk)
                     elif isinstance(chunk, list):
                         response.extend([c.content for c in chunk if hasattr(c, 'content')])
-
                     if len(response) % 3 == 0:
-                        history[-1] = (message, "".join(response).strip())
+                        history[-1] = {"role": "assistant", "content": "".join(response).strip()}
                         yield history
 
                 final_output = "".join(response).strip()
@@ -214,12 +208,12 @@ Provide ONLY the potential oversights in this format:
                 if not final_output.startswith(("1.", "-", "*", "#")):
                     final_output = "• " + final_output.replace("\n", "\n• ")
 
-                history[-1] = (message, f"### Potential Clinical Oversights:\n{final_output}")
+                history[-1] = {"role": "assistant", "content": f"### Potential Clinical Oversights:\n{final_output}"}
                 print(f"Analysis completed in {time.time()-start_time:.2f}s")
                 yield history
 
             except Exception as e:
-                history.append((message, f"❌ Analysis failed: {str(e)}"))
+                history.append({"role": "assistant", "content": f"❌ Analysis failed: {str(e)}"})
                 yield history
 
         inputs = [msg_input, chatbot, conversation_state, file_upload]