Fixing this

server.py

@@ -1,23 +1,10 @@
-import time
-import threading
-import subprocess
-import pandas as pd
-from selenium import webdriver
-from selenium.webdriver.chrome.service import Service as ChromeService
-from selenium.webdriver.chrome.options import Options as ChromeOptions
-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 datetime import datetime
-import os
-
-# Socket / Flask stack
 import eventlet
 eventlet.monkey_patch()
 
+import pandas as pd
 import io
+import threading
+import os
 import base64
 import json
 import re
@@ -25,6 +12,7 @@ from datetime import datetime
 from flask import Flask, Response, request, jsonify
 from flask_socketio import SocketIO, emit
 from flask_cors import CORS
+from worker import QuantumBot
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 from email.mime.base import MIMEBase
@@ -33,216 +21,6 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-
-class QuantumBot:
-    def __init__(self, socketio, app):
-        self.socketio = socketio; self.app = app; self.driver = None
-        self.DEFAULT_TIMEOUT = 30; self.termination_event = threading.Event()
-
-    def initialize_driver(self):
-        try:
-            self.micro_status("Initializing headless browser with stable configuration...")
-            options = ChromeOptions()
-            options.binary_location = "/usr/bin/chromium"
-            options.add_argument("--headless=new")
-            options.add_argument("--no-sandbox")
-            options.add_argument("--disable-dev-shm-usage")
-            options.add_argument("--window-size=1920,1080")
-            service = ChromeService(executable_path="/usr/bin/chromedriver")
-            self.driver = webdriver.Chrome(service=service, options=options)
-            self.micro_status("Browser initialized successfully.")
-            return True, None
-        except Exception as e:
-            error_message = f"Message: {str(e)}"; print(f"CRITICAL ERROR in WebDriver Initialization: {error_message}")
-            return False, error_message
-
-    def micro_status(self, message):
-        print(f"[Bot Log] {message}")
-        with self.app.app_context():
-            self.socketio.emit('micro_status_update', {'message': message})
-
-    def stop(self):
-        self.micro_status("Termination signal received..."); self.termination_event.set()
-
-    def login(self, username, password):
-        try:
-            self.micro_status("Navigating to login page..."); self.driver.get("https://gateway.quantumepay.com/account/login")
-            time.sleep(2); self.micro_status("Entering credentials...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "Username"))).send_keys(username)
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "Password"))).send_keys(password)
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login"))).click()
-            self.micro_status("Waiting for OTP screen...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "code1")))
-            return True, None
-        except Exception as e:
-            error_message = f"Error during login: {str(e)}"; print(f"[Bot Log] ERROR during login: {error_message}")
-            return False, error_message
-
-    def submit_otp(self, otp):
-        try:
-            self.micro_status(f"Submitting OTP..."); otp_digits = list(otp)
-            for i in range(6): self.driver.find_element(By.ID, f"code{i+1}").send_keys(otp_digits[i])
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login"))).click()
-            self.micro_status("Verifying login success...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//span[text()='Payments']")))
-            return True, None
-        except Exception as e:
-            error_message = f"Error during OTP submission: {str(e)}"; print(f"[Bot Log] ERROR during OTP submission: {error_message}")
-            return False, error_message
-
-    def start_processing(self, process_type, patient_data, **kwargs):
-        if process_type == 'void':
-            return self.process_void_list(patient_data)
-        elif process_type == 'refund':
-            return self.process_refund_list(patient_data, kwargs.get('start_date'), kwargs.get('end_date'))
-        else: return []
-
-    def process_void_list(self, patient_data):
-        results = []
-        for index, record in enumerate(patient_data):
-            if self.termination_event.is_set(): print("[Bot Log] Termination detected."); break
-            patient_name = record['Name']; patient_prn = record.get('PRN', '')
-            status = 'Skipped - No PRN'
-            if patient_prn and str(patient_prn).strip():
-                self.micro_status(f"Processing VOID for '{patient_name}' ({index + 1}/{len(patient_data)})...")
-                status = self._process_single_void(patient_name, patient_prn)
-            else:
-                self.micro_status(f"Skipping '{patient_name}' (No PRN)."); time.sleep(0.5)
-            results.append({'Name': patient_name, 'PRN': patient_prn, 'Status': status})
-            with self.app.app_context():
-                self.socketio.emit('log_update', {'name': patient_name, 'prn': patient_prn, 'status': status})
-                self.socketio.emit('stats_update', {'processed': index + 1, 'remaining': len(patient_data) - (index + 1), 'status': status})
-        return results
-
-    def process_refund_list(self, patient_data, start_date_str, end_date_str):
-        results = []
-        try:
-            self.micro_status("Navigating to Refund page to set date range...")
-            self.driver.get("https://gateway.quantumepay.com/credit-card/refund")
-            if not self._set_date_range_on_page(start_date_str, end_date_str):
-                raise Exception("Failed to set date range.")
-        except Exception as e:
-            self.micro_status(f"Critical error setting date range: {e}. Aborting refund process.")
-            for record in patient_data:
-                results.append({'Name': record['Name'], 'PRN': record.get('PRN', ''), 'Status': 'Error - Date Setup Failed'})
-            return results
-        for index, record in enumerate(patient_data):
-            if self.termination_event.is_set(): print("[Bot Log] Termination detected."); break
-            patient_name = record['Name']; patient_prn = record.get('PRN', '')
-            self.micro_status(f"Processing REFUND for '{patient_name}' ({index + 1}/{len(patient_data)})...")
-            status = self._process_single_refund(patient_name, patient_prn)
-            results.append({'Name': patient_name, 'PRN': patient_prn, 'Status': status})
-            with self.app.app_context():
-                self.socketio.emit('log_update', {'name': patient_name, 'prn': patient_prn, 'status': status})
-                self.socketio.emit('stats_update', {'processed': index + 1, 'remaining': len(patient_data) - (index + 1), 'status': status})
-        return results
-
-    def _process_single_void(self, patient_name, patient_prn):
-        try:
-            self.micro_status(f"Navigating to Void page for '{patient_name}'")
-            self.driver.get("https://gateway.quantumepay.com/credit-card/void")
-            search_successful = False
-            for attempt in range(15):
-                try:
-                    self.micro_status(f"Searching... (Attempt {attempt + 1})")
-                    WebDriverWait(self.driver, 2).until(EC.presence_of_element_located((By.XPATH, "//div[contains(@class, 'table-wrapper')]")))
-                    search_box = WebDriverWait(self.driver, 2).until(EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']")))
-                    search_box.click(); time.sleep(0.5); search_box.clear(); time.sleep(0.5)
-                    search_box.send_keys(patient_name); search_successful = True; break
-                except Exception: time.sleep(1)
-            if not search_successful: raise Exception("Failed to search for patient.")
-            time.sleep(3)
-            self.micro_status("Opening transaction details...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(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()
-            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()
-            try:
-                self.micro_status("Verifying success and saving...")
-                company_input = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.NAME, "company_name")))
-                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")))
-                self.micro_status("Clearing Contact Name field...")
-                contact_input.click(); contact_input.send_keys(Keys.CONTROL + "a"); contact_input.send_keys(Keys.BACK_SPACE)
-                self.micro_status(f"Entering PRN: {patient_prn}...")
-                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()
-                time.sleep(5); return 'Done'
-            except TimeoutException:
-                self.micro_status(f"'{patient_name}' is in a bad state, cancelling.")
-                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]"))).click()
-                return 'Bad'
-        except Exception as e:
-            print(f"An error occurred during VOID for {patient_name}: {e}"); return 'Error'
-
-    def _get_calendar_months(self):
-        try:
-            titles = self.driver.find_elements(By.XPATH, "//div[contains(@class, 'vc-title')]")
-            return [datetime.strptime(title.text.strip(), "%B %Y") for title in titles]
-        except Exception: return []
-
-    def _select_date_in_calendar(self, target_date):
-        target_month_str = target_date.strftime("%B %Y")
-        self.micro_status(f"Navigating calendar to {target_month_str}...")
-        for _ in range(24):
-            visible_months = self._get_calendar_months()
-            if any(d.strftime("%B %Y") == target_month_str for d in visible_months):
-                self.micro_status(f"Found month. Selecting day {target_date.day}.")
-                day_format = "%#d" if os.name == 'nt' else "%-d"
-                aria_label = target_date.strftime(f"%A, %B {day_format}, %Y")
-                day_xpath = f"//span[@aria-label='{aria_label}']"
-                day_element = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.XPATH, day_xpath)))
-                self.driver.execute_script("arguments.click();", day_element)
-                return
-            arrow_button_xpath = "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-left')]" if target_date < visible_months else "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-right')]"
-            self.driver.find_element(By.XPATH, arrow_button_xpath).click()
-            time.sleep(0.5)
-        raise Exception(f"Could not navigate to date {target_date.strftime('%Y-%m-%d')}")
-
-    def _set_date_range_on_page(self, start_date_str, end_date_str):
-        try:
-            self.micro_status("Opening calendar to set date range...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]"))).click()
-            time.sleep(1)
-            start_date = datetime.strptime(start_date_str, "%Y-%m-%d")
-            end_date = datetime.strptime(end_date_str, "%Y-%m-%d")
-            self._select_date_in_calendar(start_date)
-            time.sleep(1)
-            self._select_date_in_calendar(end_date)
-            self.micro_status("Date range set. Waiting for data to filter...")
-            time.sleep(5)
-            return True
-        except Exception as e:
-            self.micro_status(f"Error setting date range: {e}")
-            return False
-
-    def _process_single_refund(self, patient_name, patient_prn):
-        try:
-            search_successful = False
-            for attempt in range(15):
-                try:
-                    self.micro_status(f"Searching for '{patient_name}'... (Attempt {attempt+1})")
-                    search_box = WebDriverWait(self.driver, 2).until(EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']")))
-                    search_box.click(); time.sleep(0.5); search_box.clear(); time.sleep(0.5)
-                    search_box.send_keys(patient_name); search_successful = True; break
-                except Exception: time.sleep(1)
-            if not search_successful: raise Exception("Failed to search for patient in Refund.")
-            time.sleep(3)
-            self.micro_status(f"Performing refund action for '{patient_name}'...")
-            time.sleep(2)
-            return "Done"
-        except Exception as e:
-            print(f"An error occurred during REFUND for {patient_name}: {e}"); return 'Error'
-
-    def shutdown(self):
-        try:
-            if self.driver: self.driver.quit()
-            print("[Bot Log] Chrome session closed.")
-        except Exception as e: print(f"[Bot Log] Error during shutdown: {e}")
-
-
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret-key-for-hillside-automation'
 FRONTEND_ORIGIN = os.getenv('FRONTEND_URL', 'https://quantbot.netlify.app')
@@ -252,7 +30,6 @@ socketio = SocketIO(app, cors_allowed_origins=[FRONTEND_ORIGIN, "http://localhos
 bot_instance = None
 session_data = {}
 
-
 class GmailApiService:
     def __init__(self):
         self.sender_email = os.getenv('EMAIL_SENDER'); self.service = None
@@ -266,11 +43,70 @@ class GmailApiService:
             self.service = build('gmail', 'v1', credentials=credentials)
             print("[Server Log] Gmail API Service initialized successfully")
         except Exception as e: print(f"[Server Log] Gmail API Error: {e}")
-
-    def create_professional_email_template(self, subject, status_text, stats, custom_name, process_type):
+    
+    def create_professional_email_template(self, subject, status_text, stats, custom_name):
         status_color = "#28a745" if "completed" in status_text else "#ffc107" if "terminated" in status_text else "#dc3545"
         current_date = datetime.now().strftime("%B %d, %Y at %I:%M %p")
-        html_template = f"""<!DOCTYPE html><html lang="en"><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>{subject}</title><style> * {{ margin: 0; padding: 0; box-sizing: border-box; }} body {{ font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; line-height: 1.6; color: #2c2c2c; background-color: #f8f8f8; }} .email-container {{ max-width: 700px; margin: 20px auto; background-color: #ffffff; box-shadow: 0 8px 32px rgba(139, 0, 0, 0.15); border-radius: 12px; overflow: hidden; }} .header {{ background: linear-gradient(135deg, #8A0303 0%, #4c00ff 100%); color: white; padding: 40px 30px; text-align: center; }} .header h1 {{ font-size: 32px; font-weight: 700; margin-bottom: 8px; text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.3); }} .header p {{ font-size: 18px; opacity: 0.95; font-weight: 300; letter-spacing: 1px; }} .status-banner {{ background: {status_color}; color: white; text-align: center; padding: 18px; font-size: 16px; font-weight: 600; text-transform: uppercase; letter-spacing: 1.5px; }} .content {{ padding: 40px 30px; }} .report-info {{ background: #fdfdff; border-left: 6px solid #4c00ff; padding: 25px; margin-bottom: 35px; border-radius: 8px; box-shadow: 0 4px 12px rgba(76, 0, 255, 0.1); }} .info-grid {{ display: grid; grid-template-columns: 1fr 1fr; gap: 15px 25px; }} .info-item {{ display: flex; justify-content: space-between; align-items: center; padding: 10px 0; border-bottom: 1px solid #f0f0f0; }} .stats-grid {{ display: grid; grid-template-columns: repeat(auto-fit, minmax(140px, 1fr)); gap: 20px; margin: 35px 0; }} .stat-card {{ background: #f8f9fa; border: 1px solid #e0e0e0; border-radius: 12px; padding: 25px 15px; text-align: center; box-shadow: 0 4px 15px rgba(0, 0, 0, 0.05); }} .stat-number {{ font-size: 36px; font-weight: 700; color: #4c00ff; margin-bottom: 10px; }} .attachments-section {{ background: #f8f5ff; border: 1px solid #e0e0e0; border-radius: 12px; padding: 25px; margin: 35px 0; }} .footer {{ background: #2c2c2c; color: white; padding: 35px 30px; text-align: center; }} h3 {{ color: #8A0303; margin-bottom: 20px; font-size: 20px; font-weight: 600; }} </style></head><body> <div class="email-container"> <div class="header"><h1>Hillside's Quantum Automation</h1><p>Patient Processing Report</p></div> <div class="status-banner">Process {status_text}</div> <div class="content"> <div class="report-info"> <h3>Report Summary</h3> <div class="info-grid"> <div class="info-item"><span><b>Report Name</b></span><span>{custom_name}</span></div> <div class="info-item"><span><b>Process Type</b></span><span>{process_type}</span></div> <div class="info-item"><span><b>Generated</b></span><span>{current_date}</span></div> <div class="info-item"><span><b>Status</b></span><span style="font-weight: bold; color: {status_color};">{status_text}</span></div> </div> </div> <h3>Processing Results</h3> <div class="stats-grid"> <div class="stat-card"><div class="stat-number">{stats['total']}</div><div>Total in File</div></div> <div class="stat-card"><div class="stat-number">{stats['processed']}</div><div>Attempted</div></div> <div class="stat-card"><div class="stat-number">{stats['successful']}</div><div>Done</div></div> <div class="stat-card"><div class="stat-number">{stats['bad']}</div><div>Bad State</div></div> <div class="stat-card"><div class="stat-number">{stats['skipped']}</div><div>Skipped</div></div> </div> <div class="attachments-section"> <h3>Attached Reports</h3> <p style=\"margin-bottom: 1rem;\">The following files are attached and have been uploaded to Google Drive:</p> <ul style=\"list-style-position: inside; padding-left: 0;\"> <li><b>{custom_name}_Full.csv:</b> The complete report with the final status for every patient.</li> <li><b>{custom_name}_Bad.csv:</b> A filtered list of patients that resulted in a \"Bad\" state.</li> <li><b>{custom_name}_Skipped.csv:</b> A filtered list of patients that were skipped.</li> </ul> </div> </div> <div class=\"footer\"><h4>Hillside Automation</h4><p>This is an automated report. Please do not reply.</p></div> </div> </body></html>"""
+        
+        html_template = f"""
+        <!DOCTYPE html>
+        <html lang="en">
+        <head>
+            <meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>{subject}</title>
+            <style>
+                * {{ margin: 0; padding: 0; box-sizing: border-box; }}
+                body {{ font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; line-height: 1.6; color: #2c2c2c; background-color: #f8f8f8; }}
+                .email-container {{ max-width: 700px; margin: 20px auto; background-color: #ffffff; box-shadow: 0 8px 32px rgba(139, 0, 0, 0.15); border-radius: 12px; overflow: hidden; }}
+                .header {{ background: linear-gradient(135deg, #8A0303 0%, #4c00ff 100%); color: white; padding: 40px 30px; text-align: center; }}
+                .header h1 {{ font-size: 32px; font-weight: 700; margin-bottom: 8px; text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.3); }}
+                .header p {{ font-size: 18px; opacity: 0.95; font-weight: 300; letter-spacing: 1px; }}
+                .status-banner {{ background: {status_color}; color: white; text-align: center; padding: 18px; font-size: 16px; font-weight: 600; text-transform: uppercase; letter-spacing: 1.5px; }}
+                .content {{ padding: 40px 30px; }}
+                .report-info {{ background: #fdfdff; border-left: 6px solid var(--violet); padding: 25px; margin-bottom: 35px; border-radius: 8px; box-shadow: 0 4px 12px rgba(76, 0, 255, 0.1); }}
+                .info-grid {{ display: grid; grid-template-columns: 1fr 1fr; gap: 15px 25px; }}
+                .info-item {{ display: flex; justify-content: space-between; align-items: center; padding: 10px 0; border-bottom: 1px solid #f0f0f0; }}
+                .stats-grid {{ display: grid; grid-template-columns: repeat(auto-fit, minmax(140px, 1fr)); gap: 20px; margin: 35px 0; }}
+                .stat-card {{ background: #f8f9fa; border: 1px solid #e0e0e0; border-radius: 12px; padding: 25px 15px; text-align: center; box-shadow: 0 4px 15px rgba(0, 0, 0, 0.05); }}
+                .stat-number {{ font-size: 36px; font-weight: 700; color: var(--violet); margin-bottom: 10px; }}
+                .attachments-section {{ background: #f8f5ff; border: 1px solid #e0e0e0; border-radius: 12px; padding: 25px; margin: 35px 0; }}
+                .footer {{ background: #2c2c2c; color: white; padding: 35px 30px; text-align: center; }}
+                h3 {{ color: var(--blood-red); margin-bottom: 20px; font-size: 20px; font-weight: 600; }}
+            </style>
+        </head>
+        <body>
+            <div class="email-container">
+                <div class="header"><h1>Hillside's Quantum Automation</h1><p>Patient Processing Report</p></div>
+                <div class="status-banner">Process {status_text}</div>
+                <div class="content">
+                    <div class="report-info">
+                        <h3>Report Summary</h3>
+                        <div class="info-grid">
+                            <div class="info-item"><span><b>Report Name</b></span><span>{custom_name}</span></div>
+                            <div class="info-item"><span><b>Generated</b></span><span>{current_date}</span></div>
+                        </div>
+                    </div>
+                    <h3>Processing Results</h3>
+                    <div class="stats-grid">
+                        <div class="stat-card"> <div class="stat-number">{stats['total']}</div> <div>Total in File</div> </div>
+                        <div class="stat-card"> <div class="stat-number">{stats['processed']}</div> <div>Processed</div> </div>
+                        <div class="stat-card"> <div class="stat-number">{stats['successful']}</div> <div>Done</div> </div>
+                        <div class="stat-card"> <div class="stat-number">{stats['bad']}</div> <div>Bad State</div> </div>
+                        <div class="stat-card"> <div class="stat-number">{stats['skipped']}</div> <div>Skipped (No PRN)</div> </div>
+                    </div>
+                    <div class="attachments-section">
+                        <h3>Attached Reports</h3>
+                        <p>The following files are attached to this email and have been uploaded to Google Drive:</p>
+                        <ul>
+                            <li><b>{custom_name}_Full.csv:</b> The complete report with the final status for every patient.</li>
+                            <li><b>{custom_name}_Bad.csv:</b> A filtered list of patients that resulted in a "Bad" state.</li>
+                            <li><b>{custom_name}_Skipped.csv:</b> A filtered list of patients that were skipped due to having no PRN.</li>
+                        </ul>
+                    </div>
+                </div>
+                <div class="footer"><h4>Hillside Automation</h4><p>This is an automated report. Please do not reply.</p></div>
+            </div>
+        </body></html>
+        """
         return html_template
 
     def send_report(self, recipients, subject, body, attachments=None):
@@ -289,7 +125,6 @@ class GmailApiService:
             print(f"[Server Log] Email sent successfully! Message ID: {sent_message['id']}"); return True
         except Exception as e: print(f"[Server Log] Failed to send email: {e}"); return False
 
-
 class GoogleDriveService:
     def __init__(self):
         self.creds = None; self.service = None; self.folder_id = os.getenv('GOOGLE_DRIVE_FOLDER_ID')
@@ -302,7 +137,7 @@ class GoogleDriveService:
             self.service = build('drive', 'v3', credentials=self.creds)
             print("[Server Log] Google Drive Service initialized successfully.")
         except Exception as e: print(f"[Server Log] G-Drive ERROR: Could not initialize service: {e}")
-
+    
     def upload_file(self, filename, file_content):
         if not self.service: return False
         try:
@@ -313,26 +148,22 @@ class GoogleDriveService:
             print(f"[Server Log] File '{filename}' uploaded to Google Drive."); return True
         except Exception as e: print(f"[Server Log] G-Drive ERROR: File upload failed: {e}"); return False
 
-
 email_service = GmailApiService()
 drive_service = GoogleDriveService()
 
-
 def get_email_list():
     try:
         with open('config/emails.conf', 'r') as f: return [line.strip() for line in f if line.strip()]
     except FileNotFoundError: return []
 
-
 def run_automation_process(session_id):
     global bot_instance
     results = []; is_terminated = False; is_crash = False
     try:
-        data = session_data.get(session_id, {})
-        patient_data = data.get('patient_data'); mode = data.get('mode')
+        data = session_data.get(session_id, {}); patient_data = data.get('patient_data')
         if not patient_data: raise ValueError("No patient data prepared for automation.")
         socketio.emit('initial_stats', {'total': len(patient_data)})
-        results = bot_instance.start_processing(mode, patient_data, start_date=data.get('start_date'), end_date=data.get('end_date'))
+        results = bot_instance.process_patient_list(patient_data)
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
         print(f"[Server Log] Fatal error in automation thread: {e}"); is_crash = True
@@ -343,169 +174,106 @@ def run_automation_process(session_id):
         if bot_instance: bot_instance.shutdown(); bot_instance = None
         if session_id in session_data: del session_data[session_id]
 
-
 def generate_and_send_reports(session_id, results, is_crash_report=False, is_terminated=False):
+    print("[Server Log] Preparing final reports...")
     data = session_data.get(session_id, {})
-    if not data: print("[Server Log] Session data not found for reporting."); return
+    if not data: print("[Server Log] Session data not found, cannot generate report."); return
 
     full_df = pd.DataFrame(data.get('patient_data_for_report'))
+    
     if results:
-        result_df = pd.DataFrame(results).set_index('Name')
-        full_df.set_index('Name', inplace=True)
-        full_df['Status'] = full_df['Status'].astype('object')
-        full_df.update(result_df); full_df.reset_index(inplace=True)
+        result_df = pd.DataFrame(results)
+        if not result_df.empty:
+            result_df.set_index('Name', inplace=True)
+            full_df.set_index('Name', inplace=True)
+            full_df.update(result_df)
+            full_df.reset_index(inplace=True)
+    
     full_df['Status'].fillna('Not Processed', inplace=True)
 
+    # --- Definitive Reporting Fix: Select ONLY the required columns ---
     final_report_df = full_df[['Name', 'PRN', 'Status']]
     bad_df = final_report_df[final_report_df['Status'] == 'Bad']
     skipped_df = final_report_df[final_report_df['Status'] == 'Skipped - No PRN']
 
     timestamp = datetime.now().strftime("%d_%b_%Y"); custom_name = data.get('filename') or timestamp
     full_report_name = f"{custom_name}_Full.csv"; bad_report_name = f"{custom_name}_Bad.csv"; skipped_report_name = f"{custom_name}_Skipped.csv"
+    
     full_report_content = final_report_df.to_csv(index=False)
     drive_service.upload_file(full_report_name, full_report_content)
-
+    
     attachments = {
         full_report_name: full_report_content,
         bad_report_name: bad_df.to_csv(index=False),
         skipped_report_name: skipped_df.to_csv(index=False)
     }
-    status_text = "Terminated by User" if is_terminated else "Crashed" if is_crash_report else "Completed Successfully"
-
+    status_text = "terminated by user" if is_terminated else "crashed" if is_crash_report else "completed successfully"
+    
     stats = {
-        'total': len(full_df), 'processed': len(results),
+        'total': len(full_df),
+        'processed': len(results),
         'successful': len(full_df[full_df['Status'] == 'Done']),
-        'bad': len(bad_df), 'skipped': len(skipped_df)
+        'bad': len(bad_df),
+        'skipped': len(skipped_df)
     }
-    process_type_str = data.get('mode', 'Unknown').title()
-    subject = f"{process_type_str} Automation Report [{status_text.upper()}]: {custom_name}"
-
-    professional_body = email_service.create_professional_email_template(subject, status_text, stats, custom_name, process_type_str)
-
+    
+    subject = f"Automation Report [{status_text.upper()}]: {custom_name}"
+    
+    professional_body = email_service.create_professional_email_template(subject, status_text, stats, custom_name)
+    
     email_service.send_report(data.get('emails'), subject, professional_body, attachments)
     socketio.emit('process_complete', {'message': f'Process {status_text}. Report sent.'})
 
-
 @app.route('/')
 def status_page():
     APP_STATUS_HTML = """<!DOCTYPE html><html lang="en"><head><title>API Status</title><style>body{font-family:sans-serif;display:flex;justify-content:center;align-items:center;height:100vh;margin:0;background:#f0f2f5;}.status-box{text-align:center;padding:40px 60px;background:white;border-radius:12px;box-shadow:0 8px 30px rgba(0,0,0,0.1);}h1{font-size:24px;color:#333;margin-bottom:10px;} .indicator{font-size:18px;font-weight:600;padding:8px 16px;border-radius:20px;}.active{color:#28a745;background-color:#e9f7ea;}</style></head><body><div class="status-box"><h1>Hillside Automation API</h1><div class="indicator active">● Active</div></div></body></html>"""
     return Response(APP_STATUS_HTML)
 
-
 def extract_patient_name(raw_name):
     if not isinstance(raw_name, str): return ""
-    name_only = raw_name.split('DOB').strip()
+    name_only = raw_name.split('DOB')[0].strip()
     return re.sub(r'[:\d\-\s]+$', '', name_only).strip()
 
-
-# Existing HTTP route (kept for multipart workflows)
-@app.route('/process_and_initialize', methods=['POST'])
-def handle_file_processing_and_init():
+@app.route('/process_files_for_automation', methods=['POST'])
+def handle_file_processing():
     session_id = 'user_session'
     try:
-        data = {
-            'emails': request.form.getlist('emails[]'),
-            'filename': request.form.get('filename'),
-            'mode': request.form.get('mode'),
-            'start_date': request.form.get('start_date'),
-            'end_date': request.form.get('end_date')
-        }
-        session_data[session_id] = data
-
         if 'app_data' not in request.files or 'quantum_data' not in request.files:
             return jsonify({"error": "Both files are required."}), 400
-
+        
         df_app = pd.read_excel(request.files['app_data']) if request.files['app_data'].filename.endswith('.xlsx') else pd.read_csv(request.files['app_data'])
         df_quantum = pd.read_excel(request.files['quantum_data']) if request.files['quantum_data'].filename.endswith('.xlsx') else pd.read_csv(request.files['quantum_data'])
-
+        
         if 'Patient Name' not in df_app.columns or 'PRN' not in df_app.columns: raise ValueError("App Data must contain 'Patient Name' and 'PRN' columns.")
         if 'Name' not in df_quantum.columns: raise ValueError("Quantum Data must contain a 'Name' column.")
-
+        
         df_app_filtered = df_app.dropna(subset=['PRN']); df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
         prn_lookup_dict = {extract_patient_name(row['Patient Name']): row['PRN'] for _, row in df_app_filtered.iterrows()}
-
+        
         df_quantum['PRN'] = df_quantum['Name'].apply(lambda name: prn_lookup_dict.get(name, ""))
-
+        
         master_df = df_quantum.copy()
         master_df['Status'] = ''
-
-        data['patient_data_for_report'] = master_df
-        data['patient_data'] = master_df.to_dict('records')
-
+        
+        session_data[session_id]['patient_data_for_report'] = master_df
+        session_data[session_id]['patient_data'] = master_df.to_dict('records')
+        
         socketio.emit('data_processed')
+        print(f"[Server Log] Data prepared. Total records: {len(master_df)}")
         return jsonify({"message": "Data processed successfully."})
     except Exception as e:
         print(f"[Server Log] ERROR during file processing: {e}")
         return jsonify({"error": str(e)}), 500
 
-
-# Socket.IO Handlers
 @socketio.on('connect')
 def handle_connect():
     print(f'Frontend connected.')
     emit('email_list', {'emails': get_email_list()})
 
-
-@socketio.on('get_email_list')
-def handle_get_email_list():
-    emit('email_list', {'emails': get_email_list()})
-
-
-@socketio.on('initialize_and_process_files')
-def initialize_and_process_files(payload):
+@socketio.on('initialize_session')
+def handle_init(data):
     session_id = 'user_session'
-    try:
-        data = {
-            'emails': payload.get('emails', []),
-            'filename': payload.get('filename'),
-            'mode': payload.get('mode'),
-            'start_date': payload.get('start_date'),
-            'end_date': payload.get('end_date')
-        }
-        session_data[session_id] = data
-
-        app_b64 = payload.get('app_data_content')
-        app_name = payload.get('app_data_filename', '')
-        quantum_b64 = payload.get('quantum_data_content')
-        quantum_name = payload.get('quantum_data_filename', '')
-        if not app_b64 or not quantum_b64:
-            emit('error', {'message': 'Both files are required.'})
-            return
-
-        app_buf = io.BytesIO(base64.b64decode(app_b64))
-        quantum_buf = io.BytesIO(base64.b64decode(quantum_b64))
-
-        df_app = pd.read_excel(app_buf) if app_name.endswith('.xlsx') else pd.read_csv(app_buf)
-        df_quantum = pd.read_excel(quantum_buf) if quantum_name.endswith('.xlsx') else pd.read_csv(quantum_buf)
-
-        if 'Patient Name' not in df_app.columns or 'PRN' not in df_app.columns:
-            emit('error', {'message': "App Data must contain 'Patient Name' and 'PRN' columns."})
-            return
-        if 'Name' not in df_quantum.columns:
-            emit('error', {'message': "Quantum Data must contain a 'Name' column."})
-            return
-
-        df_app_filtered = df_app.dropna(subset=['PRN'])
-        df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
-
-        prn_lookup_dict = {
-            extract_patient_name(row['Patient Name']): row['PRN']
-            for _, row in df_app_filtered.iterrows()
-        }
-
-        df_quantum['PRN'] = df_quantum['Name'].apply(lambda name: prn_lookup_dict.get(name, ""))
-
-        master_df = df_quantum.copy()
-        master_df['Status'] = ''
-
-        data['patient_data_for_report'] = master_df
-        data['patient_data'] = master_df.to_dict('records')
-
-        socketio.emit('data_processed')
-    except Exception as e:
-        print(f"[Server Log] ERROR during file processing: {e}")
-        emit('error', {'message': str(e)})
-
+    session_data[session_id] = {'emails': data['emails'], 'filename': data['filename']}
 
 @socketio.on('start_login')
 def handle_login(credentials):
@@ -520,7 +288,6 @@ def handle_login(credentials):
     else:
         emit('error', {'message': f'Failed to initialize bot: {error_message}'})
 
-
 @socketio.on('submit_otp')
 def handle_otp(data):
     if not bot_instance: return emit('error', {'message': 'Bot not initialized.'})
@@ -531,16 +298,14 @@ def handle_otp(data):
         socketio.start_background_task(run_automation_process, session_id)
     else: emit('error', {'message': f'OTP failed: {error_message}'})
 
-
 @socketio.on('terminate_process')
 def handle_terminate():
     if bot_instance: print("Termination signal received."); bot_instance.stop()
 
-
 if __name__ == '__main__':
     print("====================================================================")
-    print("  🤗 Hillside Automation - Definitive Multi-Workflow Platform")
+    print("  🤗 Hillside Automation - Definitive Version")
     print(f"  Frontend URL: {FRONTEND_ORIGIN}")
     print(f"  Port: {os.getenv('PORT', 7860)}")
     print("====================================================================")
-    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))
+    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))

worker.py

@@ -1,5 +1,7 @@
 import time
 import threading
+import subprocess
+import pandas as pd
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options as ChromeOptions
@@ -8,47 +10,36 @@ 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 datetime import datetime
-import os
 
 class QuantumBot:
     def __init__(self, socketio, app):
-        self.socketio = socketio
-        self.app = app
-        self.driver = None
-        self.DEFAULT_TIMEOUT = 30
-        self.termination_event = threading.Event()
+        self.socketio = socketio; self.app = app; self.driver = None
+        self.DEFAULT_TIMEOUT = 30; self.termination_event = threading.Event()
 
-    def initialize_driver(self):
+    def _kill_chrome_processes(self):
         try:
-            self.micro_status("Initializing headless browser with stable configuration...")
-            options = ChromeOptions()
+            subprocess.run(['pkill', '-f', 'chromium'], check=True, timeout=5)
+            time.sleep(1)
+        except Exception: pass
 
-            # Minimal, stable args (unchanged)
-            options.add_argument("--headless=new")
-            options.add_argument("--no-sandbox")
-            options.add_argument("--disable-dev-shm-usage")
+    def initialize_driver(self):
+        try:
+            self.micro_status("Initializing headless browser...")
+            self._kill_chrome_processes()
+            options = ChromeOptions(); options.binary_location = "/usr/bin/chromium"
+            options.add_argument("--headless=new"); options.add_argument("--no-sandbox")
+            options.add_argument("--disable-dev-shm-usage"); options.add_argument("--disable-gpu")
             options.add_argument("--window-size=1920,1080")
-
-            # Pick first available Chrome/Chromium binary (robust fallback)
-            candidates = [
-                "/usr/bin/chromium",
-                "/usr/bin/chromium-browser",
-                "/usr/bin/google-chrome",
-                "/usr/bin/google-chrome-stable"
-            ]
-            for p in candidates:
-                if os.path.exists(p):
-                    options.binary_location = p
-                    break
-
+            user_agent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+            options.add_argument(f"--user-agent={user_agent}")
             service = ChromeService(executable_path="/usr/bin/chromedriver")
             self.driver = webdriver.Chrome(service=service, options=options)
-            self.micro_status("Browser initialized successfully.")
+            self.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
+                'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
+            })
             return True, None
         except Exception as e:
-            error_message = f"Message: {str(e)}"
-            print(f"CRITICAL ERROR in WebDriver Initialization: {error_message}")
+            error_message = f"Message: {str(e)}"; print(f"CRITICAL ERROR in WebDriver Initialization: {error_message}")
             return False, error_message
 
     def micro_status(self, message):
@@ -57,133 +48,56 @@ class QuantumBot:
             self.socketio.emit('micro_status_update', {'message': message})
 
     def stop(self):
-        self.micro_status("Termination signal received...")
-        self.termination_event.set()
+        self.micro_status("Termination signal received..."); self.termination_event.set()
 
     def login(self, username, password):
         try:
-            self.micro_status("Navigating to login page...")
-            self.driver.get("https://gateway.quantumepay.com/account/login")
-            time.sleep(2)
-            self.micro_status("Entering credentials...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "Username"))
-            ).send_keys(username)
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "Password"))
-            ).send_keys(password)
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "login"))
-            ).click()
+            self.micro_status("Navigating to login page..."); self.driver.get("https://gateway.quantumepay.com/")
+            time.sleep(2); self.micro_status("Entering credentials...")
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "Username"))).send_keys(username)
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "Password"))).send_keys(password)
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login"))).click()
             self.micro_status("Waiting for OTP screen...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.presence_of_element_located((By.ID, "code1"))
-            )
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "code1")))
             return True, None
         except Exception as e:
-            error_message = f"Error during login: {str(e)}"
-            print(f"[Bot Log] ERROR during login: {error_message}")
+            error_message = f"Error during login: {str(e)}"; print(f"[Bot Log] ERROR during login: {error_message}")
             return False, error_message
 
     def submit_otp(self, otp):
         try:
-            self.micro_status(f"Submitting OTP...")
-            otp_digits = list(otp)
-            for i in range(6):
-                self.driver.find_element(By.ID, f"code{i+1}").send_keys(otp_digits[i])
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "login"))
-            ).click()
+            self.micro_status(f"Submitting OTP..."); otp_digits = list(otp)
+            for i in range(6): self.driver.find_element(By.ID, f"code{i+1}").send_keys(otp_digits[i])
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login"))).click()
             self.micro_status("Verifying login success...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.XPATH, "//span[text()='Payments']"))
-            )
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//span[text()='Payments']")))
             return True, None
         except Exception as e:
-            error_message = f"Error during OTP submission: {str(e)}"
-            print(f"[Bot Log] ERROR during OTP submission: {error_message}")
+            error_message = f"Error during OTP submission: {str(e)}"; print(f"[Bot Log] ERROR during OTP submission: {error_message}")
             return False, error_message
 
-    def start_processing(self, process_type, patient_data, **kwargs):
-        if process_type == 'void':
-            return self.process_void_list(patient_data)
-        elif process_type == 'refund':
-            return self.process_refund_list(
-                patient_data, kwargs.get('start_date'), kwargs.get('end_date')
-            )
-        else:
-            return []
-
-    def process_void_list(self, patient_data):
+    def process_patient_list(self, patient_data):
         results = []
         for index, record in enumerate(patient_data):
-            if self.termination_event.is_set():
-                print("[Bot Log] Termination detected.")
-                break
-            patient_name = record['Name']
-            patient_prn = record.get('PRN', '')
-            status = 'Skipped - No PRN'
-            if patient_prn and str(patient_prn).strip():
-                self.micro_status(
-                    f"Processing VOID for '{patient_name}' ({index + 1}/{len(patient_data)})..."
-                )
-                status = self._process_single_void(patient_name, patient_prn)
-            else:
+            if self.termination_event.is_set(): print("[Bot Log] Termination detected."); break
+            patient_name = record['Name']; patient_prn = record.get('PRN', '')
+            
+            # --- Definitive Conditional Logic ---
+            if not patient_prn or not str(patient_prn).strip():
                 self.micro_status(f"Skipping '{patient_name}' (No PRN).")
-                time.sleep(0.5)
-
-            results.append({'Name': patient_name, 'PRN': patient_prn, 'Status': status})
-            with self.app.app_context():
-                self.socketio.emit('log_update', {
-                    'name': patient_name, 'prn': patient_prn, 'status': status
-                })
-                self.socketio.emit('stats_update', {
-                    'processed': index + 1,
-                    'remaining': len(patient_data) - (index + 1),
-                    'status': status
-                })
-        return results
-
-    def process_refund_list(self, patient_data, start_date_str, end_date_str):
-        results = []
-        try:
-            self.micro_status("Navigating to Refund page to set date range...")
-            self.driver.get("https://gateway.quantumepay.com/credit-card/refund")
-            if not self._set_date_range_on_page(start_date_str, end_date_str):
-                raise Exception("Failed to set date range.")
-        except Exception as e:
-            self.micro_status(f"Critical error setting date range: {e}. Aborting refund process.")
-            for record in patient_data:
-                results.append({
-                    'Name': record['Name'],
-                    'PRN': record.get('PRN', ''),
-                    'Status': 'Error - Date Setup Failed'
-                })
-            return results
+                status = 'Skipped - No PRN'
+                time.sleep(0.5) # A brief pause to make the skip visible on the UI
+            else:
+                self.micro_status(f"Processing '{patient_name}' ({index + 1}/{len(patient_data)})...")
+                status = self._process_single_patient(patient_name, patient_prn)
 
-        for index, record in enumerate(patient_data):
-            if self.termination_event.is_set():
-                print("[Bot Log] Termination detected.")
-                break
-            patient_name = record['Name']
-            patient_prn = record.get('PRN', '')
-            self.micro_status(
-                f"Processing REFUND for '{patient_name}' ({index + 1}/{len(patient_data)})..."
-            )
-            status = self._process_single_refund(patient_name, patient_prn)
             results.append({'Name': patient_name, 'PRN': patient_prn, 'Status': status})
             with self.app.app_context():
-                self.socketio.emit('log_update', {
-                    'name': patient_name, 'prn': patient_prn, 'status': status
-                })
-                self.socketio.emit('stats_update', {
-                    'processed': index + 1,
-                    'remaining': len(patient_data) - (index + 1),
-                    'status': status
-                })
+                self.socketio.emit('log_update', {'name': patient_name, 'prn': patient_prn, 'status': status})
+                self.socketio.emit('stats_update', {'processed': len(results), 'remaining': len(patient_data) - len(results), 'status': status})
         return results
 
-    def _process_single_void(self, patient_name, patient_prn):
+    def _process_single_patient(self, patient_name, patient_prn):
         try:
             self.micro_status(f"Navigating to Void page for '{patient_name}'")
             self.driver.get("https://gateway.quantumepay.com/credit-card/void")
@@ -191,173 +105,43 @@ class QuantumBot:
             for attempt in range(15):
                 try:
                     self.micro_status(f"Searching... (Attempt {attempt + 1})")
-                    WebDriverWait(self.driver, 2).until(
-                        EC.presence_of_element_located((By.XPATH, "//div[contains(@class, 'table-wrapper')]"))
-                    )
-                    search_box = WebDriverWait(self.driver, 2).until(
-                        EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']"))
-                    )
-                    search_box.click()
-                    time.sleep(0.5)
-                    search_box.clear()
-                    time.sleep(0.5)
-                    search_box.send_keys(patient_name)
-                    search_successful = True
-                    break
-                except Exception:
-                    time.sleep(1)
-            if not search_successful:
-                raise Exception("Failed to search for patient.")
-
+                    WebDriverWait(self.driver, 2).until(EC.presence_of_element_located((By.XPATH, "//div[contains(@class, 'table-wrapper')]")))
+                    search_box = WebDriverWait(self.driver, 2).until(EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']")))
+                    search_box.click(); time.sleep(0.5); search_box.clear(); time.sleep(0.5)
+                    search_box.send_keys(patient_name); search_successful = True; break
+                except Exception: time.sleep(1)
+            if not search_successful: raise Exception("Failed to search for patient.")
             time.sleep(3)
             self.micro_status("Opening transaction details...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                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.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()
             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...")
-                company_input = WebDriverWait(self.driver, 10).until(
-                    EC.element_to_be_clickable((By.NAME, "company_name"))
-                )
-                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"))
-                )
+                company_input = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.NAME, "company_name")))
+                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")))
                 self.micro_status("Clearing Contact Name field...")
-                contact_input.click()
-                contact_input.send_keys(Keys.CONTROL + "a")
-                contact_input.send_keys(Keys.BACK_SPACE)
-
+                contact_input.click(); contact_input.send_keys(Keys.CONTROL + "a"); contact_input.send_keys(Keys.BACK_SPACE)
                 self.micro_status(f"Entering PRN: {patient_prn}...")
                 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()
-                time.sleep(5)
-                return 'Done'
+                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(5); return 'Done'
             except TimeoutException:
                 self.micro_status(f"'{patient_name}' is in a bad state, cancelling.")
-                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()
                 return 'Bad'
-
         except Exception as e:
-            print(f"An error occurred during VOID for {patient_name}: {e}")
-            return 'Error'
-
-    def _get_calendar_months(self):
-        try:
-            titles = self.driver.find_elements(By.XPATH, "//div[contains(@class, 'vc-title')]")
-            return [datetime.strptime(title.text.strip(), "%B %Y") for title in titles]
-        except Exception:
-            return []
-
-    def _select_date_in_calendar(self, target_date):
-        target_month_str = target_date.strftime("%B %Y")
-        self.micro_status(f"Navigating calendar to {target_month_str}...")
-        for _ in range(24):
-            visible_months = self._get_calendar_months()
-            if any(d.strftime("%B %Y") == target_month_str for d in visible_months):
-                self.micro_status(f"Found month. Selecting day {target_date.day}.")
-                day_format = "%#d" if os.name == 'nt' else "%-d"
-                aria_label = target_date.strftime(f"%A, %B {day_format}, %Y")
-                day_xpath = f"//span[@aria-label='{aria_label}']"
-                day_element = WebDriverWait(self.driver, 10).until(
-                    EC.element_to_be_clickable((By.XPATH, day_xpath))
-                )
-                self.driver.execute_script("arguments.click();", day_element)
-                return
-
-            arrow_button_xpath = (
-                "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-left')]"
-                if visible_months and target_date < visible_months
-                else "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-right')]"
-            )
-            self.driver.find_element(By.XPATH, arrow_button_xpath).click()
-            time.sleep(0.5)
-
-        raise Exception(f"Could not navigate to date {target_date.strftime('%Y-%m-%d')}")
-
-    def _set_date_range_on_page(self, start_date_str, end_date_str):
-        try:
-            self.micro_status("Opening calendar to set date range...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]"))
-            ).click()
-            time.sleep(1)
-
-            start_date = datetime.strptime(start_date_str, "%Y-%m-%d")
-            end_date = datetime.strptime(end_date_str, "%Y-%m-%d")
-
-            self._select_date_in_calendar(start_date)
-            time.sleep(1)
-            self._select_date_in_calendar(end_date)
-
-            self.micro_status("Date range set. Waiting for data to filter...")
-            time.sleep(5)
-            return True
-        except Exception as e:
-            self.micro_status(f"Error setting date range: {e}")
-            return False
-
-    def _process_single_refund(self, patient_name, patient_prn):
-        try:
-            search_successful = False
-            for attempt in range(15):
-                try:
-                    self.micro_status(f"Searching for '{patient_name}'... (Attempt {attempt+1})")
-                    search_box = WebDriverWait(self.driver, 2).until(
-                        EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']"))
-                    )
-                    search_box.click()
-                    time.sleep(0.5)
-                    search_box.clear()
-                    time.sleep(0.5)
-                    search_box.send_keys(patient_name)
-                    search_successful = True
-                    break
-                except Exception:
-                    time.sleep(1)
-
-            if not search_successful:
-                raise Exception("Failed to search for patient in Refund.")
-
-            time.sleep(3)
-            self.micro_status(f"Performing refund action for '{patient_name}'...")
-            time.sleep(2)
-            return "Done"
-        except Exception as e:
-            print(f"An error occurred during REFUND for {patient_name}: {e}")
-            return 'Error'
+            print(f"An error occurred while processing {patient_name}: {e}"); return 'Error'
 
     def shutdown(self):
         try:
-            if self.driver:
-                self.driver.quit()
-            print("[Bot Log] Chrome session closed.")
-        except Exception as e:
-            print(f"[Bot Log] Error during shutdown: {e}")
+            if self.driver: self.driver.quit()
+            self._kill_chrome_processes()
+            print("[Bot Log] Chrome session closed and cleaned up.")
+        except Exception as e: print(f"[Bot Log] Error during shutdown: {e}")