Update app.py

app.py

@@ -60,7 +60,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:
@@ -155,7 +155,6 @@ def create_ui(agent: TxAgent):
             start_time = time.time()
             try:
                 history.append({"role": "user", "content": message})
-                history.append({"role": "assistant", "content": "Analyzing records for potential oversights..."})
                 yield history, None
 
                 extracted_data = ""
@@ -177,7 +176,7 @@ Medical Records:\n{extracted_data[:15000]}
 
 ### Potential Oversights:\n"""
 
-                response = []
+                response = ""
                 for chunk in agent.run_gradio_chat(
                     message=analysis_prompt,
                     history=[],
@@ -187,21 +186,14 @@ Medical Records:\n{extracted_data[:15000]}
                     call_agent=False,
                     conversation=conversation
                 ):
+                    partial = ""
                     if isinstance(chunk, str):
-                        response.append(chunk)
+                        partial = chunk
                     elif isinstance(chunk, list):
-                        response.extend([c.content for c in chunk if hasattr(c, 'content')])
-
-                    if len(response) % 3 == 0:
-                        history[-1] = {"role": "assistant", "content": "".join(response).strip()}
-                        yield history, None
-
-                final_output = "".join(response).strip()
-                if not final_output:
-                    final_output = "No clear oversights identified. Recommend comprehensive review."
-                if not final_output.startswith(("1.", "-", "*", "#")):
-                    final_output = "• " + final_output.replace("\n", "\n• ")
-                history[-1] = {"role": "assistant", "content": final_output}
+                        partial = "".join([c.content for c in chunk if hasattr(c, 'content')])
+                    response += partial
+                    history[-1] = {"role": "assistant", "content": response.strip()}
+                    yield history, None
 
                 report_path = os.path.join(report_dir, f"{file_hash_value}_report.txt")
                 return history, report_path if os.path.exists(report_path) else None