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:
@@ -130,7 +130,7 @@ def init_agent():
         enable_checker=True,
         step_rag_num=8,
         seed=100,
-        additional_default_tools=[]
+        additional_default_tools=[],
     )
     agent.init_model()
     return agent
@@ -141,26 +141,30 @@ 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([])
         download_output = gr.File(label="Download Full Report")
 
         def analyze_potential_oversights(message: str, history: list, conversation: list, files: list):
+            start_time = time.time()
             try:
                 history.append({"role": "user", "content": message})
-                history.append({"role": "assistant", "content": "Analyzing records for potential oversights..."})
+                history.append({"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."})
                 yield history, None
 
                 extracted_data = ""
                 file_hash_value = ""
-                if files:
+                if files and isinstance(files, list):
                     with ThreadPoolExecutor(max_workers=4) as executor:
                         futures = [executor.submit(convert_file_to_json, f.name, f.name.split(".")[-1].lower()) for f in files if hasattr(f, 'name')]
-                        results = [sanitize_utf8(f.result()) for f in as_completed(futures)]
-                        extracted_data = "\n".join(results)
-                        file_hash_value = file_hash(files[0].name)
+                        extracted_data = "\n".join([sanitize_utf8(f.result()) for f in as_completed(futures)])
+                        file_hash_value = file_hash(files[0].name) if files else ""
 
                 analysis_prompt = f"""Review these medical records and identify EXACTLY what might have been missed:
 1. List potential missed diagnoses
@@ -172,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=[],
@@ -183,23 +187,29 @@ Medical Records:\n{extracted_data[:15000]}
                     conversation=conversation
                 ):
                     if isinstance(chunk, str):
-                        response.append(chunk)
+                        response += chunk
                     elif isinstance(chunk, list):
-                        response.extend([c.content for c in chunk if hasattr(c, 'content')])
-                    history[-1] = {"role": "assistant", "content": "".join(response).strip()}
+                        response += "".join([c.content for c in chunk if hasattr(c, 'content')])
+
+                    history[-1]["content"] = response.replace("[TOOL_CALLS]", "").strip()
                     yield history, None
 
-                final_output = "".join(response).strip()
+                final_output = response.replace("[TOOL_CALLS]", "").strip()
                 if not final_output:
                     final_output = "No clear oversights identified. Recommend comprehensive review."
-                history[-1] = {"role": "assistant", "content": final_output}
 
-                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
+                report_path = None
+                if file_hash_value:
+                    possible_report = os.path.join(report_dir, f"{file_hash_value}_report.txt")
+                    if os.path.exists(possible_report):
+                        report_path = possible_report
+
+                history[-1] = {"role": "assistant", "content": final_output}
+                yield history, report_path
 
             except Exception as e:
                 history.append({"role": "assistant", "content": f"❌ Analysis failed: {str(e)}"})
-                return history, None
+                yield history, None
 
         inputs = [msg_input, chatbot, conversation_state, file_upload]
         outputs = [chatbot, download_output]
@@ -218,22 +228,6 @@ if __name__ == "__main__":
     print("Initializing medical analysis agent...")
     agent = init_agent()
 
-    print("Performing warm-up call...")
-    try:
-        warm_up = agent.run_gradio_chat(
-            message="Warm up",
-            history=[],
-            temperature=0.1,
-            max_new_tokens=10,
-            max_token=100,
-            call_agent=False,
-            conversation=[]
-        )
-        for _ in warm_up:
-            pass
-    except Exception as e:
-        print(f"Warm-up error: {str(e)}")
-
     print("Launching interface...")
     demo = create_ui(agent)
     demo.queue().launch(
@@ -241,4 +235,4 @@ if __name__ == "__main__":
         server_port=7860,
         show_error=True,
         share=True
-    )
+    )