Fixing this

server.py

@@ -54,7 +54,6 @@ def _read_tabular_from_b64(filename: str, b64_content: str) -> pd.DataFrame:
     bio = io.BytesIO(raw)
     if ext.endswith('.xlsx'):
         return pd.read_excel(bio)
-    # default csv
     return pd.read_csv(io.StringIO(raw.decode('utf-8', errors='ignore')))
 
 # =========================
@@ -70,7 +69,7 @@ class GmailApiService:
             from googleapiclient.discovery import build
             base64_creds = os.getenv('GDRIVE_SA_KEY_BASE64')
             if not base64_creds:
-                print("[Server Log] WARNING: GDRIVE_SA_KEY_BASE4 not found for Gmail.")
+                print("[Server Log] WARNING: GDRIVE_SA_KEY_BASE64 not found for Gmail.")
                 return
             creds_json = base64.b64decode(base64_creds).decode('utf-8')
             creds_dict = json.loads(creds_json)
@@ -349,7 +348,6 @@ def handle_initialize_and_process_files(payload):
             if k not in payload:
                 raise ValueError(f"Missing required field: {k}")
 
-        # Read files
         df_app = _read_tabular_from_b64(payload['app_data_filename'], payload['app_data_content'])
         df_quantum = _read_tabular_from_b64(payload['quantum_data_filename'], payload['quantum_data_content'])
 
@@ -358,7 +356,6 @@ def handle_initialize_and_process_files(payload):
         if 'Name' not in df_quantum.columns:
             raise ValueError("Quantum Data must contain a 'Name' column.")
 
-        # Build PRN lookup from App Data
         df_app_filtered = df_app.dropna(subset=['PRN'])
         df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
         prn_lookup = {
@@ -366,16 +363,13 @@ def handle_initialize_and_process_files(payload):
             for _, row in df_app_filtered.iterrows()
         }
 
-        # Assign PRN to Quantum by exact Name match (expected to be cleaned already)
         df_quantum['PRN'] = df_quantum['Name'].apply(lambda name: prn_lookup.get(name, ""))
 
-        # Prepare master with Status
         master_df = df_quantum.copy()
         if 'PRN' not in master_df.columns:
             master_df['PRN'] = ''
         master_df['Status'] = ''
 
-        # Save to session
         session_data[session_id] = {
             'emails': payload.get('emails', []),
             'filename': (payload.get('filename') or '').strip(),
@@ -386,7 +380,6 @@ def handle_initialize_and_process_files(payload):
             'patient_data': master_df.to_dict('records')
         }
 
-        # Signal UI to open login modal
         emit('data_processed')
         print(f"[Server Log] Data prepared. Total records: {len(master_df)}")
     except Exception as e:

worker.py

@@ -12,6 +12,7 @@ from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException
 
+
 class QuantumBot:
     def __init__(self, socketio, app):
         self.socketio = socketio
@@ -31,8 +32,8 @@ class QuantumBot:
 
     def initialize_driver(self):
         """
-        Headless on Linux/HF (HEADLESS=1 default), desktop Chrome when HEADLESS=0.
-        Optionally define CHROME_BINARY and CHROMEDRIVER for custom paths.
+        Headless on Linux/CI (HEADLESS=1 default), desktop Chrome when HEADLESS=0.
+        Optional env: CHROME_BINARY, CHROMEDRIVER, USER_AGENT
         """
         try:
             self.micro_status("Initializing browser...")
@@ -40,7 +41,6 @@ class QuantumBot:
             headless = os.getenv("HEADLESS", "1") == "1"
 
             if headless:
-                # Headless / CI / HF runner
                 chrome_binary = os.getenv("CHROME_BINARY", "/usr/bin/chromium")
                 chromedriver_path = os.getenv("CHROMEDRIVER", "/usr/bin/chromedriver")
                 options.binary_location = chrome_binary
@@ -53,12 +53,10 @@ class QuantumBot:
                 options.add_argument(f"--user-agent={ua}")
                 service = ChromeService(executable_path=chromedriver_path)
             else:
-                # Local desktop with installed Chrome
                 from webdriver_manager.chrome import ChromeDriverManager
                 options.add_argument("--start-maximized")
                 service = ChromeService(ChromeDriverManager().install())
 
-            # Anti-automation flags
             options.add_argument('--disable-blink-features=AutomationControlled')
             options.add_experimental_option("excludeSwitches", ["enable-automation"])
             options.add_experimental_option('useAutomationExtension', False)
@@ -148,7 +146,9 @@ class QuantumBot:
 
     def _set_date_range_on_page(self, start_date_str, end_date_str):
         self.micro_status("Opening calendar...")
-        date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]")))
+        date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+            EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]"))
+        )
         time.sleep(2)
         date_button.click()
         self._wait_for_page_load(timeout=10)
@@ -234,7 +234,6 @@ class QuantumBot:
             self.micro_status(f"Navigating to Void page for '{patient_name}'")
             self.driver.get("https://gateway.quantumepay.com/credit-card/void")
 
-            # Search with retries for stability
             for attempt in range(15):
                 try:
                     self.micro_status(f"Searching... (Attempt {attempt + 1})")
@@ -252,11 +251,17 @@ class QuantumBot:
                 EC.element_to_be_clickable((By.XPATH, f"//tr[contains(., \"{patient_name}\")]//button[@data-v-b6b33fa0]"))
             ).click()
 
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.LINK_TEXT, "Transaction Detail"))).click()
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.LINK_TEXT, "Transaction Detail"))
+            ).click()
 
             self.micro_status("Adding to Vault...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Add to Vault']"))).click()
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//div[@class='modal-footer']//button/span[normalize-space()='Confirm']"))).click()
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Add to Vault']"))
+            ).click()
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.XPATH, "//div[@class='modal-footer']//button/span[normalize-space()='Confirm']"))
+            ).click()
 
             try:
                 self.micro_status("Verifying success and saving...")
@@ -264,18 +269,23 @@ class QuantumBot:
                 company_input.clear(); company_input.send_keys(patient_name)
 
                 contact_input = WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.NAME, "company_contact")))
-                # Clear reliably then fill PRN
                 contact_input.click(); contact_input.send_keys(Keys.CONTROL + "a"); contact_input.send_keys(Keys.BACK_SPACE)
                 contact_input.send_keys(str(patient_prn))
 
-                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Save Changes']"))).click()
-                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Confirm']]"))).click()
+                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                    EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Save Changes']"))
+                ).click()
+                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                    EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Confirm']]"))
+                ).click()
                 time.sleep(2)
                 return 'Done'
             except TimeoutException:
                 self.micro_status(f"'{patient_name}' is in a bad state, cancelling.")
                 try:
-                    WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]"))).click()
+                    WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                        EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]"))
+                    ).click()
                 except Exception:
                     pass
                 return 'Bad'
@@ -309,16 +319,22 @@ class QuantumBot:
             time.sleep(0.6); details_button.click()
             self._wait_for_page_load(timeout=5)
 
-            transaction_link = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.LINK_TEXT, "Transaction Detail")))
+            transaction_link = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.LINK_TEXT, "Transaction Detail"))
+            )
             time.sleep(0.6); transaction_link.click()
             self._wait_for_page_load()
 
             self.micro_status("Adding to Vault...")
-            vault_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Add to Vault']")))
+            vault_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Add to Vault']"))
+            )
             time.sleep(0.6); vault_button.click()
             self._wait_for_page_load()
 
-            confirm_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//div[@class='modal-footer']//button/span[normalize-space()='Confirm']")))
+            confirm_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.XPATH, "//div[@class='modal-footer']//button/span[normalize-space()='Confirm']"))
+            )
             time.sleep(0.6); confirm_button.click()
             self._wait_for_page_load()
 
@@ -335,11 +351,15 @@ class QuantumBot:
                     contact_input.click(); contact_input.send_keys(Keys.CONTROL + "a"); contact_input.send_keys(Keys.BACK_SPACE)
                     contact_input.send_keys(str(patient_prn))
 
-                save_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Save Changes']")))
+                save_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                    EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Save Changes']"))
+                )
                 time.sleep(0.6); save_button.click()
                 self._wait_for_page_load()
 
-                final_confirm = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Confirm']]")))
+                final_confirm = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                    EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Confirm']]"))
+                )
                 time.sleep(0.6); final_confirm.click()
                 time.sleep(1.2)
                 self._wait_for_page_load()
@@ -347,7 +367,9 @@ class QuantumBot:
             except TimeoutException:
                 self.micro_status(f"'{patient_name}' is in a bad state, cancelling.")
                 try:
-                    cancel_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]")))
+                    cancel_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                        EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]"))
+                    )
                     time.sleep(0.5); cancel_button.click()
                 except Exception:
                     pass