Spaces:

sonuprasad23
/

quantum-bot

Running

App Files Files Community

sonuprasad23 commited on Sep 13, 2025

Commit

96e57ad

1 Parent(s): 59861d6

Fixing this

Browse files

Files changed (2) hide show

server.py +318 -117
worker.py +277 -154

server.py CHANGED Viewed

@@ -1,23 +1,24 @@
 import eventlet
 eventlet.monkey_patch()
-import pandas as pd
-import io
-import threading
 import os
-import base64
-import json
 import re
 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
 from email import encoders
-from dotenv import load_dotenv
 load_dotenv()
@@ -30,212 +31,412 @@ 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
         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 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):
-        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()
-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')
-        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:
-        print(f"[Server Log] Fatal error in automation thread: {e}"); is_crash = True
         socketio.emit('error', {'message': f'A fatal error occurred: {e}'})
     finally:
         socketio.emit('micro_status_update', {'message': 'Generating final reports...'})
         generate_and_send_reports(session_id, results, is_crash_report=is_crash, is_terminated=is_terminated)
-        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):
-    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'].fillna('Not Processed', inplace=True)
     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 ""
-    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
-        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'] = ''
-        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."})
-    except Exception as e:
-        print(f"[Server Log] ERROR during file processing: {e}")
-        return jsonify({"error": str(e)}), 500
-@socketio.on('connect')
-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.'})
-    is_success, error_message = bot_instance.submit_otp(data['otp'])
     if is_success:
         emit('login_successful')
         session_id = 'user_session'
         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(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)))

 import eventlet
 eventlet.monkey_patch()
 import os
+import io
 import re
+import json
+import base64
+import pandas as pd
 from datetime import datetime
+from flask import Flask, Response, jsonify, request
 from flask_socketio import SocketIO, emit
 from flask_cors import CORS
+from dotenv import load_dotenv
 from worker import QuantumBot
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 from email.mime.base import MIMEBase
 from email import encoders
 load_dotenv()
 bot_instance = None
 session_data = {}
+# =========================
+# Utilities
+# =========================
+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 extract_patient_name(raw_name):
+    if not isinstance(raw_name, str):
+        return ""
+    name_only = raw_name.split('DOB').strip()
+    return re.sub(r'[:\d\-\s]+$', '', name_only).strip()
+def _read_tabular_from_b64(filename: str, b64_content: str) -> pd.DataFrame:
+    raw = base64.b64decode(b64_content)
+    ext = (filename or '').lower()
+    bio = io.BytesIO(raw)
+    if ext.endswith('.xlsx'):
+        return pd.read_excel(bio)
+    # default csv
+    return pd.read_csv(io.StringIO(raw.decode('utf-8', errors='ignore')))
+# =========================
+# Google APIs
+# =========================
 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_BASE4 not found for Gmail.")
+                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):
         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 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):
+        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()
+# =========================
+# Core Automation
+# =========================
 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')
+        if not patient_data:
+            raise ValueError("No patient data prepared for automation.")
+        mode = data.get('mode', 'void')
+        start_date = data.get('start_date')
+        end_date = data.get('end_date')
         socketio.emit('initial_stats', {'total': len(patient_data)})
+        results = bot_instance.process_patient_list(
+            patient_data,
+            mode=mode,
+            start_date=start_date,
+            end_date=end_date
+        )
         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
         socketio.emit('error', {'message': f'A fatal error occurred: {e}'})
     finally:
         socketio.emit('micro_status_update', {'message': 'Generating final reports...'})
         generate_and_send_reports(session_id, results, is_crash_report=is_crash, is_terminated=is_terminated)
+        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, cannot generate report.")
+        return
     full_df = pd.DataFrame(data.get('patient_data_for_report'))
     if results:
+        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)
     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'].str.startswith('Skipped')]
+    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(final_report_df[final_report_df['Status'] == 'Done']),
+        'bad': len(bad_df),
+        'skipped': len(skipped_df)
     }
+    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.'})
+# =========================
+# Routes
+# =========================
 @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)
+# =========================
+# Socket Handlers (align with frontend)
+# =========================
+@socketio.on('connect')
+def handle_connect():
+    print('Frontend connected.')
+@socketio.on('get_email_list')
+def handle_get_email_list():
+    emit('email_list', {'emails': get_email_list()})
+@socketio.on('initialize_and_process_files')
+def handle_initialize_and_process_files(payload):
+    """
+    Frontend sends:
+    - emails: list[str]
+    - filename: str (optional)
+    - mode: 'void' or 'refund'
+    - start_date, end_date (YYYY-MM-DD) for refund mode
+    - app_data_content (base64), app_data_filename
+    - quantum_data_content (base64), quantum_data_filename
+    """
     session_id = 'user_session'
     try:
+        required_keys = ['app_data_content', 'app_data_filename', 'quantum_data_content', 'quantum_data_filename', 'mode', 'emails']
+        for k in required_keys:
+            if k not in payload:
+                raise ValueError(f"Missing required field: {k}")
+        # Read files
+        df_app = _read_tabular_from_b64(payload['app_data_filename'], payload['app_data_content'])
+        df_quantum = _read_tabular_from_b64(payload['quantum_data_filename'], payload['quantum_data_content'])
+        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.")
+        # Build PRN lookup from App Data
+        df_app_filtered = df_app.dropna(subset=['PRN'])
+        df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
+        prn_lookup = {
+            extract_patient_name(row['Patient Name']): row['PRN']
+            for _, row in df_app_filtered.iterrows()
+        }
+        # Assign PRN to Quantum by exact Name match (expected to be cleaned already)
+        df_quantum['PRN'] = df_quantum['Name'].apply(lambda name: prn_lookup.get(name, ""))
+        # Prepare master with Status
         master_df = df_quantum.copy()
+        if 'PRN' not in master_df.columns:
+            master_df['PRN'] = ''
         master_df['Status'] = ''
+        # Save to session
+        session_data[session_id] = {
+            'emails': payload.get('emails', []),
+            'filename': (payload.get('filename') or '').strip(),
+            'mode': payload.get('mode', 'void'),
+            'start_date': payload.get('start_date') or None,
+            'end_date': payload.get('end_date') or None,
+            'patient_data_for_report': master_df,
+            'patient_data': master_df.to_dict('records')
+        }
+        # Signal UI to open login modal
+        emit('data_processed')
+        print(f"[Server Log] Data prepared. Total records: {len(master_df)}")
+    except Exception as e:
+        print(f"[Server Log] ERROR during initialize_and_process_files: {e}")
+        emit('error', {'message': str(e)})
 @socketio.on('start_login')
 def handle_login(credentials):
     global bot_instance
+    try:
+        if bot_instance:
+            bot_instance.shutdown()
+        bot_instance = QuantumBot(socketio, app)
+        is_success, error_message = bot_instance.initialize_driver()
+        if not is_success:
+            return emit('error', {'message': f'Failed to initialize bot: {error_message}'})
+        ok, login_error = bot_instance.login(credentials.get('username'), credentials.get('password'))
+        if ok:
+            emit('otp_required')
+        else:
+            emit('error', {'message': f'Login failed: {login_error}'})
+    except Exception as e:
+        emit('error', {'message': f'Login init error: {e}'})
 @socketio.on('submit_otp')
 def handle_otp(data):
+    if not bot_instance:
+        return emit('error', {'message': 'Bot not initialized.'})
+    is_success, error_message = bot_instance.submit_otp(data.get('otp'))
     if is_success:
         emit('login_successful')
         session_id = 'user_session'
         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()
+# =========================
+# Entrypoint
+# =========================
 if __name__ == '__main__':
     print("====================================================================")
+    print("  🤗 Hillside Automation - Unified (Void + Refund)")
     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)))

worker.py CHANGED Viewed

@@ -1,9 +1,8 @@
 import time
 import threading
-import subprocess
-import pandas as pd
-import os
 from datetime import datetime
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options as ChromeOptions
@@ -15,48 +14,83 @@ 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"); options.add_argument("--no-first-run")
-            options.add_argument("--disable-extensions");
-            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})"
             })
             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/")
-            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()
@@ -64,111 +98,38 @@ class QuantumBot:
             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'))
-        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 = []
-        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_str, end_date_str)
-            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'
-            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, "%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}.")
@@ -176,69 +137,231 @@ class QuantumBot:
                 expected_aria_label = target_date.strftime(f"%A, %B {day_format}, %Y")
                 day_xpath = f"//span[@aria-label='{expected_aria_label}']"
                 day_element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.XPATH, day_xpath)))
-                self.driver.execute_script("arguments[0].click();", day_element)
                 return
-            if target_date < visible_months[0]:
                 self.driver.find_element(By.XPATH, "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-left')]").click()
             else:
                 self.driver.find_element(By.XPATH, "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-right')]").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):
         self.micro_status("Opening calendar...")
         date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]")))
-        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, start_date_str, end_date_str):
         try:
-            self.micro_status(f"Resetting state for '{patient_name}'...")
-            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)
-            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()
-            # This is the corrected, more complete refund logic
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.XPATH, "//input[@name='amount']")))
             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}")

+import os
 import time
 import threading
 from datetime import datetime
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options as ChromeOptions
 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 micro_status(self, message):
+        print(f"[Bot] {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 initialize_driver(self):
+        """
+        Headless on Linux/HF (HEADLESS=1 default), desktop Chrome when HEADLESS=0.
+        Optionally define CHROME_BINARY and CHROMEDRIVER for custom paths.
+        """
         try:
+            self.micro_status("Initializing browser...")
+            options = ChromeOptions()
+            headless = os.getenv("HEADLESS", "1") == "1"
+            if headless:
+                # Headless / CI / HF runner
+                chrome_binary = os.getenv("CHROME_BINARY", "/usr/bin/chromium")
+                chromedriver_path = os.getenv("CHROMEDRIVER", "/usr/bin/chromedriver")
+                options.binary_location = chrome_binary
+                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")
+                ua = os.getenv("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={ua}")
+                service = ChromeService(executable_path=chromedriver_path)
+            else:
+                # Local desktop with installed Chrome
+                from webdriver_manager.chrome import ChromeDriverManager
+                options.add_argument("--start-maximized")
+                service = ChromeService(ChromeDriverManager().install())
+            # Anti-automation flags
+            options.add_argument('--disable-blink-features=AutomationControlled')
+            options.add_experimental_option("excludeSwitches", ["enable-automation"])
+            options.add_experimental_option('useAutomationExtension', False)
             self.driver = webdriver.Chrome(service=service, options=options)
             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}")
             return False, error_message
+    def _wait_for_page_load(self, timeout=None):
+        if timeout is None:
+            timeout = self.DEFAULT_TIMEOUT
+        self.micro_status("Waiting for page activity to cease...")
+        loading_overlay_xpath = "//div[contains(@class, 'vld-background')]"
+        try:
+            WebDriverWait(self.driver, timeout).until(
+                EC.invisibility_of_element_located((By.XPATH, loading_overlay_xpath))
+            )
+            self.micro_status("...Page is ready.")
+        except TimeoutException:
+            self.micro_status("Warning: Loading overlay did not disappear in time.")
     def login(self, username, password):
         try:
+            self.micro_status("Navigating to login page...")
+            self.driver.get("https://gateway.quantumepay.com/")
+            self._wait_for_page_load()
             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()
             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] ERROR: {error_message}")
             return False, error_message
     def submit_otp(self, otp):
         try:
+            self.micro_status("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._wait_for_page_load()
             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] ERROR: {error_message}")
             return False, error_message
+    # ===== Refund helpers (calendar) =====
     def _get_calendar_months(self):
         try:
             titles = self.driver.find_elements(By.XPATH, "//div[contains(@class, 'vc-title')]")
             return [datetime.strptime(title.text, "%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}.")
                 expected_aria_label = target_date.strftime(f"%A, %B {day_format}, %Y")
                 day_xpath = f"//span[@aria-label='{expected_aria_label}']"
                 day_element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.XPATH, day_xpath)))
+                self.driver.execute_script("arguments.click();", day_element)
                 return
+            if visible_months and target_date < visible_months:
                 self.driver.find_element(By.XPATH, "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-left')]").click()
             else:
                 self.driver.find_element(By.XPATH, "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-right')]").click()
+            self._wait_for_page_load(timeout=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):
         self.micro_status("Opening calendar...")
         date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]")))
+        time.sleep(2)
+        date_button.click()
+        self._wait_for_page_load(timeout=10)
+        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)
+        self._wait_for_page_load(timeout=10)
         self._select_date_in_calendar(end_date)
+        time.sleep(2)
         self.driver.find_element(By.TAG_NAME, "body").click()
+        self._wait_for_page_load(timeout=10)
+        self._wait_for_page_load()
     def _navigate_and_verify(self, url, verification_xpath):
+        self.micro_status("Navigating to page and verifying...")
         self.driver.get(url)
         time.sleep(3)
+        self._wait_for_page_load()
         try:
+            WebDriverWait(self.driver, 15).until(EC.element_to_be_clickable((By.XPATH, verification_xpath)))
+            self.micro_status("Page loaded successfully.")
         except TimeoutException:
+            self.micro_status("Page did not load correctly. Attempting a hard refresh...")
+            self.driver.refresh()
+            time.sleep(3)
+            self._wait_for_page_load()
+            WebDriverWait(self.driver, 15).until(EC.element_to_be_clickable((By.XPATH, verification_xpath)))
+            self.micro_status("Page loaded successfully after refresh.")
+    # ===== Main dispatcher =====
+    def process_patient_list(self, patient_data, mode='void', start_date=None, end_date=None):
+        """
+        mode: 'void' or 'refund'
+        For 'refund', start_date and end_date must be provided (YYYY-MM-DD).
+        """
+        results = []
+        idx = 0
+        total = len(patient_data)
+        while idx < total:
+            if self.termination_event.is_set():
+                print("[Bot] Termination detected.")
+                break
+            record = patient_data[idx]
+            patient_name = record.get('Name', '')
+            patient_prn = record.get('PRN', '')
+            with self.app.app_context():
+                self.socketio.emit('stats_update', {
+                    'processed': len(results),
+                    'remaining': total - len(results),
+                    'status': 'Starting'
+                })
+            if not patient_prn or not str(patient_prn).strip():
+                status = 'Skipped - No PRN'
+                self.micro_status(f"Skipping '{patient_name}' (No PRN).")
+                time.sleep(0.4)
+                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})
+                idx += 1
+                continue
+            self.micro_status(f"Processing '{patient_name}' ({len(results) + 1}/{total})...")
+            if mode == 'refund':
+                status = self._process_single_patient_refund(patient_name, patient_prn, start_date, end_date)
+            else:
+                status = self._process_single_patient_void(patient_name, patient_prn)
+            if status != "RETRY":
+                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})
+                idx += 1
+        return results
+    # ===== Void workflow =====
+    def _process_single_patient_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 with retries for stability
+            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.3); search_box.clear(); time.sleep(0.2)
+                    search_box.send_keys(patient_name)
+                    break
+                except Exception:
+                    time.sleep(0.6)
+            time.sleep(2)
             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")))
+                # Clear reliably then fill PRN
+                contact_input.click(); contact_input.send_keys(Keys.CONTROL + "a"); contact_input.send_keys(Keys.BACK_SPACE)
+                contact_input.send_keys(str(patient_prn))
+                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Save Changes']"))).click()
+                WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Confirm']]"))).click()
+                time.sleep(2)
+                return 'Done'
+            except TimeoutException:
+                self.micro_status(f"'{patient_name}' is in a bad state, cancelling.")
+                try:
+                    WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]"))).click()
+                except Exception:
+                    pass
+                return 'Bad'
         except Exception as e:
+            print(f"An error occurred while processing {patient_name}: {e}")
             return 'Error'
+    # ===== Refund workflow =====
+    def _process_single_patient_refund(self, patient_name, patient_prn, start_date_str, end_date_str):
+        try:
+            refund_page_url = "https://gateway.quantumepay.com/credit-card/refund"
+            date_button_xpath = "//button[.//span[contains(text(), '-')]]"
+            self._navigate_and_verify(refund_page_url, date_button_xpath)
+            self._set_date_range_on_page(start_date_str, end_date_str)
+            self.micro_status(f"Searching for '{patient_name}'...")
+            search_box = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.XPATH, "//input[@placeholder='Search']"))
+            )
+            time.sleep(0.5)
+            search_box.click(); search_box.clear(); time.sleep(0.2); search_box.send_keys(patient_name)
+            self._wait_for_page_load()
+            row_xpath = f"//tr[contains(., \"{patient_name}\")]"
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.XPATH, row_xpath)))
+            self.micro_status("Opening transaction details...")
+            details_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
+                EC.element_to_be_clickable((By.XPATH, f"{row_xpath}//button[@data-v-b6b33fa0]"))
+            )
+            time.sleep(0.6); details_button.click()
+            self._wait_for_page_load(timeout=5)
+            transaction_link = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.LINK_TEXT, "Transaction Detail")))
+            time.sleep(0.6); transaction_link.click()
+            self._wait_for_page_load()
+            self.micro_status("Adding to Vault...")
+            vault_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Add to Vault']")))
+            time.sleep(0.6); vault_button.click()
+            self._wait_for_page_load()
+            confirm_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//div[@class='modal-footer']//button/span[normalize-space()='Confirm']")))
+            time.sleep(0.6); confirm_button.click()
+            self._wait_for_page_load()
+            try:
+                self.micro_status("Verifying success and saving...")
+                company_input = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.NAME, "company_name")))
+                time.sleep(0.5); company_input.clear(); company_input.send_keys(patient_name)
+                if patient_prn and str(patient_prn).strip() and str(patient_prn).lower() != 'nan':
+                    contact_input = WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.NAME, "company_contact")))
+                    time.sleep(0.5)
+                    if contact_input.get_attribute('value') == 'N/A':
+                        contact_input.clear()
+                    contact_input.click(); contact_input.send_keys(Keys.CONTROL + "a"); contact_input.send_keys(Keys.BACK_SPACE)
+                    contact_input.send_keys(str(patient_prn))
+                save_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button/span[normalize-space()='Save Changes']")))
+                time.sleep(0.6); save_button.click()
+                self._wait_for_page_load()
+                final_confirm = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Confirm']]")))
+                time.sleep(0.6); final_confirm.click()
+                time.sleep(1.2)
+                self._wait_for_page_load()
+                return 'Done'
+            except TimeoutException:
+                self.micro_status(f"'{patient_name}' is in a bad state, cancelling.")
+                try:
+                    cancel_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[normalize-space()='Cancel']]")))
+                    time.sleep(0.5); cancel_button.click()
+                except Exception:
+                    pass
+                self._wait_for_page_load()
+                return 'Bad'
+        except Exception as e:
+            print(f"An error occurred while processing {patient_name}: {e}")
+            self.micro_status(f"An error occurred for '{patient_name}'. Retrying this patient...")
+            return 'RETRY'
     def shutdown(self):
         try:
+            if self.driver:
+                self.driver.quit()
+            print("[Bot] Chrome session closed.")
+        except Exception as e:
+            print(f"[Bot] Error during shutdown: {e}")