Update app.py

app.py

@@ -56,7 +56,7 @@ def extract_priority_pages(file_path: str, max_pages: int = 20) -> str:
                 text_chunks.append(f"=== Page {i+1} ===\n{text.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:
@@ -140,8 +140,9 @@ def create_ui(agent):
         send_btn = gr.Button("Analyze", variant="primary")
         download_output = gr.File(label="Download Full Report")
 
-        def analyze(message: str, history: list, files: list):
-            history = history + [{"role": "user", "content": message}]
+        def analyze(message: str, history: List[dict], files: List):
+            # Append user message
+            history.append({"role": "user", "content": message})
             history.append({"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."})
             yield history, None
 
@@ -152,7 +153,7 @@ def create_ui(agent):
                     futures = [executor.submit(convert_file_to_json, f.name, f.name.split(".")[-1].lower()) for f in files]
                     results = [sanitize_utf8(f.result()) for f in as_completed(futures)]
                     extracted = "\n".join(results)
-                    file_hash_value = file_hash(files[0].name)
+                    file_hash_value = file_hash(files[0].name) if files else ""
 
             prompt = f"""Review these medical records and identify EXACTLY what might have been missed:
 1. List potential missed diagnoses
@@ -165,6 +166,11 @@ Medical Records:
 """
 
             try:
+                # Remove the temporary "Analyzing..." message
+                if history and history[-1]["content"].startswith("⏳"):
+                    history.pop()
+
+                # Process agent response
                 for chunk in agent.run_gradio_chat(
                     message=prompt,
                     history=[],
@@ -176,14 +182,21 @@ Medical Records:
                 ):
                     if chunk is None:
                         continue
+
                     if isinstance(chunk, list):
                         for m in chunk:
-                            history.append({"role": m.role, "content": m.content})
-                            yield history, None
-                    elif isinstance(chunk, str):
-                        history[-1]["content"] += chunk
+                            if hasattr(m, 'content') and m.content:
+                                history.append({"role": m.role, "content": sanitize_utf8(m.content)})
+                                yield history, None
+                    elif isinstance(chunk, str) and chunk.strip():
+                        # Append new assistant message or update the last one
+                        if history and history[-1]["role"] == "assistant":
+                            history[-1]["content"] += sanitize_utf8(chunk)
+                        else:
+                            history.append({"role": "assistant", "content": sanitize_utf8(chunk)})
                         yield history, None
 
+                # Provide report path if available
                 report_path = os.path.join(report_dir, f"{file_hash_value}_report.txt") if file_hash_value else None
                 yield history, report_path if report_path and os.path.exists(report_path) else None
 
@@ -206,4 +219,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=[report_dir],
         share=False
-    )
+    )