Almost Done

worker.py

@@ -11,7 +11,7 @@ from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
-from selenium.common.exceptions import TimeoutException
+from selenium.common.exceptions import TimeoutException, NoSuchElementException
 
 class QuantumBot:
     def __init__(self, socketio, app, logger):
@@ -24,7 +24,6 @@ class QuantumBot:
             self.logger.info("Attempting to kill any lingering chromium processes...")
             subprocess.run(['pkill', '-f', 'chromium'], check=True, timeout=5)
             time.sleep(1)
-            self.logger.info("Lingering processes terminated.")
         except Exception as e:
             self.logger.warning(f"Could not kill chrome processes (this is often normal): {e}")
 
@@ -141,32 +140,20 @@ class QuantumBot:
         try:
             self.micro_status(f"Searching for '{patient_name}'...")
             search_box = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']")))
-            
-            # --- Definitive Search Fix ---
-            # 1. Get a reference to the current first row of the table.
-            try:
-                old_first_row = self.driver.find_element(By.XPATH, "//tbody/tr[1]")
-            except NoSuchElementException:
-                old_first_row = None # Handle case where table is initially empty
-
-            # 2. Type the search text.
             search_box.click(); time.sleep(0.5); search_box.clear(); time.sleep(0.5)
             search_box.send_keys(patient_name)
+            self._wait_for_page_load()
             
-            # 3. Wait for the old row to disappear (become stale). This is the key.
-            if old_first_row:
-                self.micro_status("Waiting for search results to update...")
-                WebDriverWait(self.driver, 15).until(EC.staleness_of(old_first_row))
-            else:
-                time.sleep(2) # If no old row, just pause briefly for results to appear
+            try:
+                self.micro_status("Verifying search result...")
+                row_xpath = f"//tr[contains(., \"{patient_name}\")]"
+                WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.XPATH, row_xpath)))
+                self.micro_status("Patient found. Opening transaction details...")
+                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, f"{row_xpath}//button[@data-v-b6b33fa0]"))).click()
+            except TimeoutException:
+                self.micro_status(f"Patient '{patient_name}' not found in search results.")
+                return 'Not Found'
 
-            # 4. Now, confidently wait for the new row to appear.
-            self.micro_status("Waiting for new search result row to appear...")
-            new_row_xpath = f"//tr[contains(., \"{patient_name}\")]"
-            WebDriverWait(self.driver, 15).until(EC.presence_of_element_located((By.XPATH, new_row_xpath)))
-            
-            self.micro_status("Opening transaction details...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, f"{new_row_xpath}//button[@data-v-b6b33fa0]"))).click()
             WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.LINK_TEXT, "Transaction Detail"))).click()
             self._wait_for_page_load()
             self.micro_status("Adding to Vault...")