Spaces:

sonuprasad23
/

quantum-bot

Sleeping

App Files Files Community

sonuprasad23 commited on Sep 13, 2025

Commit

47657a7

1 Parent(s): 7dde815

Fixing this

Browse files

Files changed (2) hide show

server.py +19 -11
worker.py +71 -76

server.py CHANGED Viewed

@@ -47,7 +47,7 @@ class GmailApiService:
     def create_professional_email_template(self, subject, status_text, stats, custom_name, process_type):
         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 Patients</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>"""
         return html_template
     def send_report(self, recipients, subject, body, attachments=None):
@@ -104,7 +104,6 @@ def run_automation_process(session_id):
         data = session_data.get(session_id, {});
         patient_data = data.get('patient_data'); mode = data.get('mode')
         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'))
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
@@ -158,7 +157,7 @@ def generate_and_send_reports(session_id, results, is_crash_report=False, is_ter
 @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):
@@ -166,11 +165,10 @@ def extract_patient_name(raw_name):
     name_only = raw_name.split('DOB')[0].strip()
     return re.sub(r'[:\d\-\s]+$', '', name_only).strip()
-@app.route('/process_and_initialize', methods=['POST'])
 def handle_file_processing_and_init():
     session_id = 'user_session'
     try:
-        # Load session data from the custom header
         session_data_from_header = json.loads(request.headers.get('X-Session-Data'))
         session_data[session_id] = session_data_from_header
@@ -191,8 +189,10 @@ def handle_file_processing_and_init():
         master_df = df_quantum.copy()
         master_df['Status'] = ''
         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')
         return jsonify({"message": "Data processed successfully."})
@@ -205,19 +205,27 @@ def handle_connect():
     print(f'Frontend connected.')
     emit('email_list', {'emails': get_email_list()})
-@socketio.on('start_login')
-def handle_login(credentials):
     global bot_instance
     if bot_instance: bot_instance.shutdown()
     bot_instance = QuantumBot(socketio, app)
     is_success, error_message = bot_instance.initialize_driver()
     if is_success:
-        is_login_success, login_error = bot_instance.login(credentials['username'], credentials['password'])
-        if is_login_success: emit('otp_required')
-        else: emit('error', {'message': f'Login failed: {login_error}'})
     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.'})

     def create_professional_email_template(self, subject, status_text, stats, custom_name, process_type):
         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><head><title>{subject}</title><style>body{{font-family:sans-serif;}}</style></head><body><h2>Hillside's Quantum Automation Report</h2><p><b>Status:</b> {status_text}</p><div>...</div></body></html>"""
         return html_template
     def send_report(self, recipients, subject, body, attachments=None):
         data = session_data.get(session_id, {});
         patient_data = data.get('patient_data'); mode = data.get('mode')
         if not patient_data: raise ValueError("No patient data prepared for automation.")
         results = bot_instance.start_processing(mode, patient_data, start_date=data.get('start_date'), end_date=data.get('end_date'))
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
 @app.route('/')
 def status_page():
+    APP_STATUS_HTML = """<!DOCTYPE html><html><head><title>API Status</title></head><body><h1>Hillside Automation API is Active</h1></body></html>"""
     return Response(APP_STATUS_HTML)
 def extract_patient_name(raw_name):
     name_only = raw_name.split('DOB')[0].strip()
     return re.sub(r'[:\d\-\s]+$', '', name_only).strip()
+@app.route('/process_files_for_automation', methods=['POST'])
 def handle_file_processing_and_init():
     session_id = 'user_session'
     try:
         session_data_from_header = json.loads(request.headers.get('X-Session-Data'))
         session_data[session_id] = session_data_from_header
         master_df = df_quantum.copy()
         master_df['Status'] = ''
+        # We save the full, merged dataframe for the final report
         session_data[session_id]['patient_data_for_report'] = master_df
+        # We only process records that have a PRN
+        session_data[session_id]['patient_data'] = master_df[master_df['PRN'] != ''].to_dict('records')
         socketio.emit('data_processed')
         return jsonify({"message": "Data processed successfully."})
     print(f'Frontend connected.')
     emit('email_list', {'emails': get_email_list()})
+@socketio.on('initialize_session')
+def handle_init(data):
+    session_id = 'user_session'
+    session_data[session_id] = data # Store mode, dates, emails etc.
     global bot_instance
     if bot_instance: bot_instance.shutdown()
     bot_instance = QuantumBot(socketio, app)
     is_success, error_message = bot_instance.initialize_driver()
     if is_success:
+        emit('bot_initialized')
     else:
         emit('error', {'message': f'Failed to initialize bot: {error_message}'})
+@socketio.on('start_login')
+def handle_login(credentials):
+    if not bot_instance: return emit('error', {'message': 'Bot not initialized.'})
+    is_success, error_message = bot_instance.login(credentials['username'], credentials['password'])
+    if is_success: emit('otp_required')
+    else: emit('error', {'message': f'Login failed: {error_message}'})
 @socketio.on('submit_otp')
 def handle_otp(data):
     if not bot_instance: return emit('error', {'message': 'Bot not initialized.'})

worker.py CHANGED Viewed

@@ -12,29 +12,26 @@ 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
 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 _kill_chrome_processes(self):
-        try:
-            subprocess.run(['pkill', '-f', 'chromium'], check=True, timeout=5)
-            time.sleep(1)
-        except Exception: pass
     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")
-            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.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
                 'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
@@ -82,7 +79,7 @@ class QuantumBot:
         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'))
         return []
     def process_void_list(self, patient_data):
@@ -90,67 +87,73 @@ class QuantumBot:
         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 = []
-        self._navigate_and_verify("https://gateway.quantumepay.com/credit-card/refund", "//button[.//span[contains(text(), '-')]]")
-        self._set_date_range_on_page(start_date_str, end_date_str)
         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 REFUND for '{patient_name}' ({index + 1}/{len(patient_data)})...")
-                status = self._process_single_refund(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_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")))
-                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'
@@ -159,7 +162,7 @@ class QuantumBot:
                 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:
@@ -192,49 +195,41 @@ class QuantumBot:
         date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]")))
         time.sleep(1); date_button.click()
         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)
-        time.sleep(1)
         self.driver.find_element(By.TAG_NAME, "body").click()
         self.micro_status("Date range set. Waiting for data to filter...")
         time.sleep(3)
-    def _navigate_and_verify(self, url, verification_xpath):
-        self.micro_status(f"Navigating and verifying page...")
-        self.driver.get(url)
-        time.sleep(3)
         try:
-            WebDriverWait(self.driver, 15).until(EC.presence_of_element_located((By.XPATH, verification_xpath)))
-        except TimeoutException:
-            self.micro_status("Page did not load correctly. Refreshing...")
-            self.driver.refresh(); time.sleep(3)
-            WebDriverWait(self.driver, 15).until(EC.presence_of_element_located((By.XPATH, verification_xpath)))
-        self.micro_status("Page loaded.")
-    def _process_single_refund(self, patient_name, patient_prn):
-        try:
-            self.micro_status(f"Searching for '{patient_name}' in date range...")
-            search_box = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']")))
-            time.sleep(1); search_box.click(); search_box.clear(); time.sleep(0.5); search_box.send_keys(patient_name)
-            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("Refunding...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Refund']"))).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()
-            time.sleep(5)
-            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()
-            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}")

 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException
+from webdriver_manager.chrome import ChromeDriverManager
 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.download_dir = os.path.join(os.getcwd(), "temp_downloads")
     def initialize_driver(self):
         try:
+            self.micro_status("Initializing browser...")
+            options = ChromeOptions()
+            options.add_argument("--start-maximized")
+            options.add_argument('--disable-blink-features=AutomationControlled')
+            options.add_experimental_option("excludeSwitches", ["enable-automation"])
+            options.add_experimental_option('useAutomationExtension', False)
+            os.makedirs(self.download_dir, exist_ok=True)
+            prefs = {"download.default_directory": self.download_dir}
+            options.add_experimental_option("prefs", prefs)
+            service = ChromeService(ChromeDriverManager().install())
             self.driver = webdriver.Chrome(service=service, options=options)
             self.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
                 'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
         if process_type == 'void':
             return self.process_void_list(patient_data)
         elif process_type == 'refund':
+            return self.process_refund_list(kwargs.get('start_date'), kwargs.get('end_date'))
         return []
     def process_void_list(self, patient_data):
         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 VOID for '{patient_name}' ({index + 1}/{len(patient_data)})...")
+            status = self._process_single_void_or_refund(patient_name, patient_prn, "void")
             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, start_date_str, end_date_str):
+        self.micro_status("Starting Refund Report Download process...")
+        downloaded_df = self._download_report(start_date_str, end_date_str, "refund")
+        if downloaded_df is None: raise Exception("Failed to download the refund report.")
+        patient_data = downloaded_df.to_dict('records')
+        self.socketio.emit('initial_stats', {'total': len(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', '')
+            self.micro_status(f"Processing REFUND for '{patient_name}' ({index + 1}/{len(patient_data)})...")
+            status = self._process_single_void_or_refund(patient_name, patient_prn, "refund", start_date_str, end_date_str, is_first_record=(index==0))
             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_or_refund(self, patient_name, patient_prn, mode, start_date_str=None, end_date_str=None, is_first_record=False):
         try:
+            page_url = f"https://gateway.quantumepay.com/credit-card/{mode}"
+            self.micro_status(f"Navigating to {mode.title()} page for '{patient_name}'")
+            self.driver.get(page_url)
+            time.sleep(3)
+            if mode == 'refund' and not is_first_record:
+                self._set_date_range_on_page(start_date_str, end_date_str)
             search_successful = False
+            for attempt in range(5):
                 try:
                     self.micro_status(f"Searching... (Attempt {attempt + 1})")
+                    WebDriverWait(self.driver, 5).until(EC.presence_of_element_located((By.XPATH, "//div[contains(@class, 'table-wrapper')]")))
+                    search_box = WebDriverWait(self.driver, 5).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)
+                if patient_prn and str(patient_prn).strip() and str(patient_prn).lower() != 'nan':
+                    self.micro_status(f"Entering PRN: {patient_prn}...")
+                    contact_input = WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.NAME, "company_contact")))
+                    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()
                 time.sleep(5); return 'Done'
                 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 {mode.upper()} for {patient_name}: {e}"); return 'Error'
     def _get_calendar_months(self):
         try:
         date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]")))
         time.sleep(1); date_button.click()
         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); time.sleep(1)
         self.driver.find_element(By.TAG_NAME, "body").click()
         self.micro_status("Date range set. Waiting for data to filter...")
         time.sleep(3)
+    def _download_report(self, start_date_str, end_date_str, mode):
         try:
+            self.micro_status(f"Navigating to {mode.title()} page to download report...")
+            self.driver.get(f"https://gateway.quantumepay.com/credit-card/{mode}")
+            self._set_date_range_on_page(start_date_str, end_date_str)
+            self.micro_status("Exporting report to CSV...")
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[@class='icon']]"))).click()
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.LINK_TEXT, "Export CSV"))).click()
+            self.micro_status("Waiting for download to complete...")
+            time.sleep(10)
+            files = os.listdir(self.download_dir)
+            if not files: raise Exception("CSV file was not downloaded.")
+            downloaded_file_path = os.path.join(self.download_dir, files[0])
+            df = pd.read_csv(downloaded_file_path)
+            os.remove(downloaded_file_path)
+            return df
         except Exception as e:
+            print(f"Error downloading report: {e}")
+            return None
     def shutdown(self):
         try:
             if self.driver: self.driver.quit()
+            if os.path.exists(self.download_dir):
+                import shutil
+                shutil.rmtree(self.download_dir)
+            print("[Bot Log] Chrome session and downloads cleaned up.")
         except Exception as e: print(f"[Bot Log] Error during shutdown: {e}")