Spaces:

sonuprasad23
/

quantum-bot

Sleeping

App Files Files Community

sonuprasad23 commited on Sep 12, 2025

Commit

2d2ac3e

1 Parent(s): b48d5cc

Updated Logic for Refund

Browse files

Files changed (2) hide show

server.py +109 -22
worker.py +72 -24

server.py CHANGED Viewed

@@ -31,12 +31,63 @@ bot_instance = None
 session_data = {}
 class GmailApiService:
-    # This class remains unchanged
-    pass
 class GoogleDriveService:
-    # This class remains unchanged
-    pass
 email_service = GmailApiService()
 drive_service = GoogleDriveService()
@@ -66,13 +117,50 @@ def run_automation_process(session_id):
         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):
-    # This function remains unchanged
-    pass
 @app.route('/')
 def status_page():
-    # This function remains unchanged
-    pass
 def extract_patient_name(raw_name):
     if not isinstance(raw_name, str): return ""
@@ -83,18 +171,7 @@ def extract_patient_name(raw_name):
 def handle_file_processing_and_init():
     session_id = 'user_session'
     try:
-        # --- Definitive Fix: Get all data from FormData ---
-        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] = {
-            'emails': emails, 'filename': filename, 'mode': mode,
-            'start_date': start_date, 'end_date': end_date
-        }
         if 'app_data' not in request.files or 'quantum_data' not in request.files:
             return jsonify({"error": "Both files are required."}), 400
@@ -112,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."})
@@ -126,6 +205,14 @@ 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

 session_data = {}
 class GmailApiService:
+    def __init__(self):
+        self.sender_email = os.getenv('EMAIL_SENDER'); self.service = None
+        try:
+            from google.oauth2 import service_account; from googleapiclient.discovery import build
+            base64_creds = os.getenv('GDRIVE_SA_KEY_BASE64')
+            if not base64_creds: print("[Server Log] WARNING: GDRIVE_SA_KEY_BASE64 not found."); return
+            creds_json = base64.b64decode(base64_creds).decode('utf-8'); creds_dict = json.loads(creds_json)
+            credentials = service_account.Credentials.from_service_account_info(creds_dict, scopes=['https://www.googleapis.com/auth/gmail.send'])
+            if self.sender_email: credentials = credentials.with_subject(self.sender_email)
+            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):
+        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>"""
+        return html_template
+    def send_report(self, recipients, subject, body, attachments=None):
+        if not self.service or not recipients: return False
+        try:
+            from googleapiclient.errors import HttpError
+            message = MIMEMultipart(); message['From'] = self.sender_email; message['To'] = ', '.join(recipients); message['Subject'] = subject
+            message.attach(MIMEText(body, 'html'))
+            if attachments:
+                for filename, content in attachments.items():
+                    part = MIMEBase('application', 'octet-stream'); part.set_payload(content.encode('utf-8'))
+                    encoders.encode_base64(part); part.add_header('Content-Disposition', f'attachment; filename="{filename}"')
+                    message.attach(part)
+            raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode('utf-8')
+            sent_message = self.service.users().messages().send(userId='me', body={'raw': raw_message}).execute()
+            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')
+        try:
+            from google.oauth2 import service_account; from googleapiclient.discovery import build
+            base64_creds = os.getenv('GDRIVE_SA_KEY_BASE64')
+            if not base64_creds or not self.folder_id: raise ValueError("Google Drive secrets not found.")
+            creds_json = base64.b64decode(base64_creds).decode('utf-8'); creds_dict = json.loads(creds_json)
+            self.creds = service_account.Credentials.from_service_account_info(creds_dict, scopes=['https://www.googleapis.com/auth/drive'])
+            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:
+            from googleapiclient.http import MediaIoBaseUpload
+            file_metadata = {'name': filename, 'parents': [self.folder_id]}
+            media = MediaIoBaseUpload(io.BytesIO(file_content.encode('utf-8')), mimetype='text/csv', resumable=True)
+            self.service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+            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()
         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):
+    data = session_data.get(session_id, {});
+    if not data: print("[Server Log] Session data not found for reporting."); 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.update(result_df); full_df.reset_index(inplace=True)
+    full_df['Status'] = full_df['Status'].fillna('Not Processed')
+    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"
+    stats = {
+        'total': len(full_df), 'processed': len(results),
+        'successful': len(full_df[full_df['Status'] == 'Done']),
+        '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)
+    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 ""
 def handle_file_processing_and_init():
     session_id = 'user_session'
     try:
+        data = session_data.get(session_id, {})
         if 'app_data' not in request.files or 'quantum_data' not in request.files:
             return jsonify({"error": "Both files are required."}), 400
         master_df = df_quantum.copy()
         master_df['Status'] = ''
+        data['patient_data_for_report'] = master_df
+        data['patient_data'] = master_df.to_dict('records')
+        data['start_date'] = request.form.get('start_date')
+        data['end_date'] = request.form.get('end_date')
         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] = {
+        'emails': data['emails'], 'filename': data['filename'], 'mode': data.get('mode'),
+        'start_date': data.get('start_date'), 'end_date': data.get('end_date')
+    }
 @socketio.on('start_login')
 def handle_login(credentials):
     global bot_instance

worker.py CHANGED Viewed

@@ -121,26 +121,48 @@ class QuantumBot:
             time.sleep(1)
             self._select_date_in_calendar(end_date)
             self.driver.find_element(By.TAG_NAME, "body").click()
-            self.micro_status("Date range set. Waiting up to 1 minute for data to filter...")
-            WebDriverWait(self.driver, 60).until(EC.invisibility_of_element_located((By.XPATH, "//div[contains(@class, 'is-loading')]")))
-            self.micro_status("Data has finished loading.")
             return True
         except Exception as e:
             self.micro_status(f"Error setting date range: {e}"); return False
     def process_void_list(self, patient_data):
-        # This method remains unchanged as it is stable
-        pass
     def process_refund_list(self, patient_data, start_date, end_date):
         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 REFUND for '{patient_name}' ({index + 1}/{len(patient_data)})...")
-                status = self._process_single_refund(patient_name, patient_prn, start_date, end_date)
             else:
                 self.micro_status(f"Skipping '{patient_name}' (No PRN).")
                 time.sleep(0.5)
@@ -150,42 +172,37 @@ class QuantumBot:
                 self.socketio.emit('stats_update', {'processed': index + 1, 'remaining': len(patient_data) - (index + 1), 'status': status})
         return results
-    def _process_single_refund(self, patient_name, patient_prn):
         try:
-            self.micro_status(f"Navigating to Refund page for '{patient_name}'")
-            self.driver.get("https://gateway.quantumepay.com/credit-card/refund")
-            # Date range must be set for each patient in a robust loop
-            if not self._set_date_range(start_date, end_date):
-                 raise Exception("Date range could not be set.")
             search_successful = False
             for attempt in range(15):
                 try:
                     self.micro_status(f"Searching... (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("Opening transaction details...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, f"//tr[contains(., \"{patient_name}\")]//button"))).click()
-            self.micro_status("Initiating Refund...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Refund']"))).click()
             try:
                 self.micro_status("Verifying success and saving...")
-                # Assuming the fields are the same as the Void workflow
                 company_input = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.NAME, "company_name")))
                 company_input.click(); company_input.send_keys(Keys.CONTROL + "a"); company_input.send_keys(Keys.BACK_SPACE)
                 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)
                 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'
@@ -193,6 +210,37 @@ class QuantumBot:
                 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 REFUND for {patient_name}: {e}"); return 'Error'

             time.sleep(1)
             self._select_date_in_calendar(end_date)
             self.driver.find_element(By.TAG_NAME, "body").click()
+            self.micro_status("Date range set. Waiting dynamically for up to 1 minute for data to filter...")
+            WebDriverWait(self.driver, 60).until_not(EC.presence_of_element_located((By.XPATH, "//*[contains(@class, 'is-loading')]")))
+            time.sleep(2) # Extra buffer
             return True
         except Exception as e:
             self.micro_status(f"Error setting date range: {e}"); return False
     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, end_date):
         results = []
+        # Set date range once at the beginning
+        self.driver.get("https://gateway.quantumepay.com/credit-card/refund")
+        if not self._set_date_range(start_date, end_date):
+            self.micro_status("FATAL: Could not set date range. Terminating process.")
+            for record in patient_data:
+                results.append({'Name': record['Name'], 'PRN': record.get('PRN',''), 'Status': 'Error'})
+            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', '')
             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)
                 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.click(); company_input.send_keys(Keys.CONTROL + "a"); company_input.send_keys(Keys.BACK_SPACE)
                 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'
                 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 _process_single_refund(self, patient_name, patient_prn):
+        # This is the new, robust Refund logic.
+        try:
+            # We are already on the correct page with the date range set
+            search_successful = False
+            for attempt in range(15):
+                try:
+                    self.micro_status(f"Searching for '{patient_name}' in date range... (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("Opening transaction details...")
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, f"//tr[contains(., \"{patient_name}\")]//button"))).click()
+            self.micro_status("Initiating Refund...")
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Refund']"))).click()
+            try:
+                self.micro_status("Verifying refund success...")
+                # We expect a confirmation modal or a success message.
+                # Here we will assume a final confirm button appears, similar to Void.
+                WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.XPATH, "//div[@class='modal-footer']//button/span[normalize-space()='Confirm']"))).click()
+                time.sleep(5); return 'Done'
+            except TimeoutException:
+                self.micro_status(f"'{patient_name}' is in a bad state for refund, 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 REFUND for {patient_name}: {e}"); return 'Error'