Spaces:

sonuprasad23
/

quantum-bot

Running

App Files Files Community

sonuprasad23 commited on Aug 28, 2025

Commit

ca62807

1 Parent(s): f03221e

Added new feature

Browse files

Files changed (2) hide show

server.py +111 -562
worker.py +57 -137

server.py CHANGED Viewed

@@ -7,8 +7,9 @@ import threading
 import os
 import base64
 import json
 from datetime import datetime
-from flask import Flask, Response
 from flask_socketio import SocketIO, emit
 from flask_cors import CORS
 from worker import QuantumBot
@@ -22,658 +23,206 @@ load_dotenv()
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret-key-for-hillside-automation'
 FRONTEND_ORIGIN = os.getenv('FRONTEND_URL', 'https://quantbot.netlify.app')
-CORS(app, resources={r"/*": {"origins": [FRONTEND_ORIGIN, "http://localhost:3000", "http://127.0.0.1:5500"]}})
-socketio = SocketIO(app, cors_allowed_origins=[FRONTEND_ORIGIN, "http://localhost:3000", "http://127.0.0.1:5500"], async_mode='eventlet')
 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("[Gmail API] 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("[Gmail API] Service initialized successfully")
-        except Exception as e:
-            print(f"[Gmail API] Error: {e}")
-    def create_professional_email_template(self, subject, status_text, stats, custom_name):
-        """Create a professional blood red, white & violet themed email template"""
-        status_color = "#8B0000" if "successfully" in status_text else "#DC143C" if "error" in status_text else "#8A2BE2"
-        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, #8B0000 0%, #DC143C 50%, #8A2BE2 100%);
-                    color: white;
-                    padding: 40px 30px;
-                    text-align: center;
-                    position: relative;
-                }}
-                .header::before {{
-                    content: '';
-                    position: absolute;
-                    top: 0;
-                    left: 0;
-                    right: 0;
-                    bottom: 0;
-                    background: rgba(255, 255, 255, 0.1);
-                    backdrop-filter: blur(10px);
-                }}
-                .header-content {{
-                    position: relative;
-                    z-index: 2;
-                }}
-                .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;
-                    box-shadow: inset 0 2px 4px rgba(0, 0, 0, 0.2);
-                }}
-                .content {{
-                    padding: 40px 30px;
-                }}
-                .report-info {{
-                    background: linear-gradient(135deg, #f8f8f8 0%, #ffffff 100%);
-                    border-left: 6px solid #8A2BE2;
-                    padding: 25px;
-                    margin-bottom: 35px;
-                    border-radius: 8px;
-                    box-shadow: 0 4px 12px rgba(138, 43, 226, 0.1);
-                }}
-                .report-info h3 {{
-                    color: #8B0000;
-                    margin-bottom: 20px;
-                    font-size: 20px;
-                    font-weight: 600;
-                }}
-                .info-grid {{
-                    display: grid;
-                    grid-template-columns: 1fr 1fr;
-                    gap: 20px;
-                    margin-bottom: 20px;
-                }}
-                .info-item {{
-                    display: flex;
-                    justify-content: space-between;
-                    align-items: center;
-                    padding: 15px 0;
-                    border-bottom: 2px solid #f0f0f0;
-                }}
-                .info-label {{
-                    font-weight: 600;
-                    color: #8B0000;
-                    font-size: 14px;
-                }}
-                .info-value {{
-                    font-weight: 700;
-                    color: #2c2c2c;
-                    font-size: 14px;
-                }}
-                .stats-section {{
-                    margin: 35px 0;
-                }}
-                .stats-section h3 {{
-                    color: #8B0000;
-                    margin-bottom: 25px;
-                    font-size: 20px;
-                    font-weight: 600;
-                    border-bottom: 3px solid #8A2BE2;
-                    padding-bottom: 10px;
-                }}
-                .stats-grid {{
-                    display: grid;
-                    grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
-                    gap: 20px;
-                    margin-bottom: 25px;
-                }}
-                .stat-card {{
-                    background: linear-gradient(135deg, #ffffff 0%, #f8f8f8 100%);
-                    border: 2px solid #e0e0e0;
-                    border-radius: 12px;
-                    padding: 25px 15px;
-                    text-align: center;
-                    box-shadow: 0 4px 15px rgba(0, 0, 0, 0.08);
-                    transition: all 0.3s ease;
-                }}
-                .stat-card:hover {{
-                    transform: translateY(-2px);
-                    box-shadow: 0 8px 25px rgba(139, 0, 0, 0.15);
-                }}
-                .stat-number {{
-                    font-size: 36px;
-                    font-weight: 800;
-                    color: #8A2BE2;
-                    margin-bottom: 10px;
-                    text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.1);
-                }}
-                .stat-label {{
-                    font-size: 12px;
-                    color: #666;
-                    text-transform: uppercase;
-                    letter-spacing: 1px;
-                    font-weight: 600;
-                }}
-                .success .stat-number {{ color: #8B0000; }}
-                .warning .stat-number {{ color: #DC143C; }}
-                .danger .stat-number {{ color: #8A2BE2; }}
-                .attachments-section {{
-                    background: linear-gradient(135deg, #8A2BE2 0%, #9370DB 100%);
-                    color: white;
-                    border-radius: 12px;
-                    padding: 25px;
-                    margin: 35px 0;
-                    box-shadow: 0 6px 20px rgba(138, 43, 226, 0.3);
-                }}
-                .attachments-section h3 {{
-                    color: white;
-                    margin-bottom: 20px;
-                    font-size: 18px;
-                    font-weight: 600;
-                }}
-                .attachment-item {{
-                    display: flex;
-                    align-items: center;
-                    padding: 15px 0;
-                    border-bottom: 1px solid rgba(255, 255, 255, 0.2);
-                }}
-                .attachment-item:last-child {{
-                    border-bottom: none;
-                }}
-                .attachment-icon {{
-                    width: 28px;
-                    height: 28px;
-                    background-color: #8B0000;
-                    border-radius: 6px;
-                    margin-right: 15px;
-                    display: flex;
-                    align-items: center;
-                    justify-content: center;
-                    color: white;
-                    font-size: 12px;
-                    font-weight: bold;
-                    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.3);
-                }}
-                .attachment-name {{
-                    font-weight: 500;
-                    color: white;
-                    font-size: 14px;
-                }}
-                .footer {{
-                    background: linear-gradient(135deg, #2c2c2c 0%, #8B0000 100%);
-                    color: white;
-                    padding: 35px 30px;
-                    text-align: center;
-                }}
-                .footer-content {{
-                    max-width: 500px;
-                    margin: 0 auto;
-                }}
-                .footer h4 {{
-                    margin-bottom: 15px;
-                    font-size: 18px;
-                    font-weight: 700;
-                }}
-                .footer p {{
-                    font-size: 14px;
-                    opacity: 0.9;
-                    line-height: 1.6;
-                }}
-                .footer .timestamp {{
-                    margin-top: 25px;
-                    padding-top: 20px;
-                    border-top: 1px solid rgba(255, 255, 255, 0.3);
-                    font-size: 12px;
-                    opacity: 0.8;
-                }}
-                @media only screen and (max-width: 600px) {{
-                    .email-container {{
-                        margin: 10px;
-                        border-radius: 8px;
-                    }}
-                    .header {{
-                        padding: 30px 20px;
-                    }}
-                    .content {{
-                        padding: 25px 20px;
-                    }}
-                    .info-grid {{
-                        grid-template-columns: 1fr;
-                        gap: 15px;
-                    }}
-                    .stats-grid {{
-                        grid-template-columns: repeat(2, 1fr);
-                    }}
-                    .header h1 {{
-                        font-size: 26px;
-                    }}
-                    .stat-number {{
-                        font-size: 28px;
-                    }}
-                }}
-            </style>
-        </head>
-        <body>
-            <div class="email-container">
-                <div class="header">
-                    <div class="header-content">
-                        <h1>Hillside Medical Group</h1>
-                        <p>Patient Processing Report</p>
-                    </div>
-                </div>
-                <div class="status-banner">
-                    Process {status_text.title()}
-                </div>
-                <div class="content">
-                    <div class="report-info">
-                        <h3>Report Summary</h3>
-                        <div class="info-grid">
-                            <div class="info-item">
-                                <span class="info-label">Report Name</span>
-                                <span class="info-value">{custom_name}</span>
-                            </div>
-                            <div class="info-item">
-                                <span class="info-label">Generated</span>
-                                <span class="info-value">{current_date}</span>
-                            </div>
-                            <div class="info-item">
-                                <span class="info-label">Status</span>
-                                <span class="info-value" style="color: {status_color};">{status_text.title()}</span>
-                            </div>
-                            <div class="info-item">
-                                <span class="info-label">Processing Type</span>
-                                <span class="info-value">Automated</span>
-                            </div>
-                        </div>
-                    </div>
-                    <div class="stats-section">
-                        <h3>Processing Results</h3>
-                        <div class="stats-grid">
-                            <div class="stat-card">
-                                <div class="stat-number">{stats['total']}</div>
-                                <div class="stat-label">Total Records</div>
-                            </div>
-                            <div class="stat-card">
-                                <div class="stat-number">{stats['processed']}</div>
-                                <div class="stat-label">Processed</div>
-                            </div>
-                            <div class="stat-card success">
-                                <div class="stat-number">{stats['successful']}</div>
-                                <div class="stat-label">Completed</div>
-                            </div>
-                            <div class="stat-card danger">
-                                <div class="stat-number">{stats['failed']}</div>
-                                <div class="stat-label">Requires Review</div>
-                            </div>
-                        </div>
-                    </div>
-                    <div class="attachments-section">
-                        <h3>Report Files</h3>
-                        <div class="attachment-item">
-                            <div class="attachment-icon">CSV</div>
-                            <div class="attachment-name">{custom_name}_Full.csv - Complete patient processing report</div>
-                        </div>
-                        <div class="attachment-item">
-                            <div class="attachment-icon">CSV</div>
-                            <div class="attachment-name">{custom_name}_Bad.csv - Records requiring manual review</div>
-                        </div>
-                    </div>
-                </div>
-                <div class="footer">
-                    <div class="footer-content">
-                        <h4>Hillside Medical Group</h4>
-                        <p>
-                            Automated patient processing system ensuring accurate and efficient
-                            record management while maintaining compliance with healthcare data standards.
-                        </p>
-                        <div class="timestamp">
-                            Report generated {current_date}
-                        </div>
-                    </div>
-                </div>
-            </div>
-        </body>
-        </html>
-        """
-        return html_template
     def send_report(self, recipients, subject, body, attachments=None):
-        if not self.service:
-            print("[Gmail API] Service not initialized")
-            return False
-        if not recipients:
-            print("[Gmail API] No recipients provided")
-            return False
         try:
             from googleapiclient.errors import HttpError
-            print(f"[Gmail API] Sending report to: {recipients}")
-            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')
-            gmail_message = {'raw': raw_message}
-            sent_message = self.service.users().messages().send(
-                userId='me',
-                body=gmail_message
-            ).execute()
-            print(f"[Gmail API] Email sent successfully! Message ID: {sent_message['id']}")
-            return True
-        except HttpError as error:
-            print(f"[Gmail API] HTTP Error: {error}")
-            return False
-        except Exception as e:
-            print(f"[Gmail API] Error: {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("[G-Drive] Service initialized successfully.")
-        except Exception as e:
-            print(f"[G-Drive] ERROR: Could not initialize Google Drive 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"[G-Drive] File '{filename}' uploaded successfully.")
-            return True
-        except Exception as e:
-            print(f"[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, {})
-        csv_content = data.get('csv_content')
-        df = pd.read_csv(io.StringIO(csv_content))
-        if 'Status' not in df.columns:
-            df.insert(1, 'Status', '')
-        patient_list = df[(df['Status'] != 'Done') & (df['Status'] != 'Bad')]['Name'].tolist()
-        socketio.emit('initial_stats', {'total': len(patient_list)})
-        results = bot_instance.process_patient_list(patient_list)
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
-        print(f"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):
     if not results:
-        socketio.emit('process_complete', {'message': 'No patients were processed.'})
-        return
-    data = session_data.get(session_id, {})
-    original_df = pd.read_csv(io.StringIO(data.get('csv_content')))
-    if 'Status' not in original_df.columns:
-        original_df.insert(1, 'Status', '')
-    result_df = pd.DataFrame(results).set_index('Name')
-    original_df.set_index('Name', inplace=True)
-    original_df.update(result_df)
-    full_df = original_df.reset_index()
-    bad_df = full_df[full_df['Status'] == 'Bad'][['Name', 'Status']]
-    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"
     full_report_content = full_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)}
-    status_text = "terminated by user" if is_terminated else "crashed due to an error" if is_crash_report else "completed successfully"
-    stats = {
-        'total': len(full_df),
-        'processed': len(results),
-        'successful': len([r for r in results if r['Status'] == 'Done']),
-        'failed': len([r for r in results if r['Status'] == 'Bad'])
-    }
-    subject = f"Patient Processing Report - {status_text.title()}: {custom_name}"
-    professional_body = email_service.create_professional_email_template(
-        subject, status_text, stats, custom_name
-    )
-    email_success = email_service.send_report(data.get('emails'), subject, professional_body, attachments)
-    if email_success:
-        socketio.emit('process_complete', {'message': f'Process {status_text}. Report sent successfully.'})
-    else:
-        socketio.emit('process_complete', {'message': f'Process {status_text}. Email sending failed.'})
 @app.route('/')
 def status_page():
-    APP_STATUS_HTML = """<!DOCTYPE html><html lang="en"><head><title>Hillside Automation</title><style>body{font-family:-apple-system,BlinkMacSystemFont,Segoe UI,Roboto,Helvetica,Arial,sans-serif;display:flex;justify-content:center;align-items:center;height:100vh;margin:0;background:linear-gradient(135deg, #8B0000 0%, #DC143C 50%, #8A2BE2 100%);color:white;}.status-box{text-align:center;padding:40px 60px;background:rgba(255,255,255,0.15);border-radius:12px;box-shadow:0 8px 30px rgba(0,0,0,0.3);backdrop-filter:blur(10px);}h1{font-size:28px;margin-bottom:10px;} .indicator{font-size:18px;font-weight:600;padding:12px 20px;border-radius:25px;background:rgba(255,255,255,0.2);border:2px solid white;color:white;margin:15px 0;}.info{margin-top:20px;opacity:0.9;}</style></head><body><div class="status-box"><h1>Hillside Automation</h1><div class="indicator">Patient Processing System</div><div class="info">Chrome: Ready<br>Email: Gmail API<br>Drive: Integrated<br><br><a href="https://quantbot.netlify.app" style="color:#fff;">quantbot.netlify.app</a></div></div></body></html>"""
     return Response(APP_STATUS_HTML)
 @socketio.on('connect')
 def handle_connect():
-    print(f'Frontend connected from: {FRONTEND_ORIGIN}')
     emit('email_list', {'emails': get_email_list()})
 @socketio.on('initialize_session')
 def handle_init(data):
     session_id = 'user_session'
-    global bot_instance
-    session_data[session_id] = {'csv_content': data['content'], 'emails': data['emails'], 'filename': data['filename']}
-    if bot_instance:
-        bot_instance.shutdown()
-    bot_instance = QuantumBot(socketio, app)
-    is_success, error_message = bot_instance.initialize_driver()
-    if is_success:
-        emit('bot_initialized')
-    else:
-        emit('error', {'message': f'Failed to initialize automation bot: {error_message}'})
 @socketio.on('start_login')
 def handle_login(credentials):
-    if not bot_instance:
-        return emit('error', {'message': 'Bot not initialized.'})
-    is_success, error_message = bot_instance.login(credentials['username'], credentials['password'])
     if is_success:
-        emit('otp_required')
     else:
-        emit('error', {'message': f'Login failed: {error_message}'})
 @socketio.on('submit_otp')
 def handle_otp(data):
-    if not bot_instance:
-        return emit('error', {'message': 'Bot not initialized.'})
     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 Medical Group - Patient Processing System")
-    print("  Email: Professional Reports via Gmail API")
-    print(f"  Frontend: {FRONTEND_ORIGIN}")
     print("====================================================================")
-    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))

 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
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret-key-for-hillside-automation'
 FRONTEND_ORIGIN = os.getenv('FRONTEND_URL', 'https://quantbot.netlify.app')
+CORS(app, resources={r"/*": {"origins": [FRONTEND_ORIGIN, "http://localhost:3000", "http://127.0.0.1:5500", "null"]}})
+socketio = SocketIO(app, cors_allowed_origins=[FRONTEND_ORIGIN, "http://localhost:3000", "http://127.0.0.1:5500", "null"], async_mode='eventlet')
 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("[Gmail API] 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("[Gmail API] Service initialized successfully")
+        except Exception as e: print(f"[Gmail API] Error: {e}")
     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"[Gmail API] Email sent successfully! Message ID: {sent_message['id']}"); return True
+        except Exception as e: print(f"[Gmail API] Error: {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("[G-Drive] Service initialized successfully.")
+        except Exception as e: print(f"[G-Drive] ERROR: Could not initialize Google Drive 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"[G-Drive] File '{filename}' uploaded successfully."); return True
+        except Exception as e: print(f"[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')
+        socketio.emit('initial_stats', {'total': len(patient_data)})
+        results = bot_instance.process_patient_list(patient_data)
         is_terminated = bot_instance.termination_event.is_set()
     except Exception as e:
+        print(f"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):
     if not results:
+        socketio.emit('process_complete', {'message': 'No patients were processed.'}); return
+    data = session_data.get(session_id, {});
+    # Use the prepared data as the base for the final report
+    full_df = pd.DataFrame(data.get('patient_data_for_report'))
+    result_df = pd.DataFrame(results)
+    if not result_df.empty:
+        result_df = result_df.set_index(['Name', 'PRN'])
+        full_df = full_df.set_index(['Name', 'PRN'])
+        full_df.update(result_df)
+        full_df.reset_index(inplace=True)
+    bad_df = full_df[full_df['Status'] == 'Bad'][['Name', 'PRN', 'Status']]
+    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"
     full_report_content = full_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)}
+    status_text = "terminated by user" if is_terminated else "crashed" if is_crash_report else "completed"
+    subject = f"Automation Report [{status_text.upper()}]: {custom_name}"
+    body = f"The process was {status_text}. The full report is in Google Drive and also attached."
+    email_service.send_report(data.get('emails'), subject, 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 standardize_name(name):
+    if not isinstance(name, str): return ""
+    match = re.match(r'^([\w\s,-]+?)\s*(?:ID|$)', name)
+    if not match: return ""
+    clean_name = match.group(1).strip().lower()
+    if ',' in clean_name:
+        parts = [p.strip() for p in clean_name.split(',')]
+        clean_name = "".join(reversed(parts))
+    return re.sub(r'[^a-z0-9]', '', clean_name)
+@app.route('/process_files_for_automation', methods=['POST'])
+def process_files_for_automation():
+    session_id = 'user_session'
+    if 'app_data' not in request.files or 'quantum_data' not in request.files:
+        return jsonify({"error": "Both files are required."}), 400
+    try:
+        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'])
+        df_app_filtered = df_app.dropna(subset=['PRN']); df_app_filtered = df_app_filtered[df_app_filtered['PRN'].astype(str).str.strip() != '']
+        prn_lookup = {standardize_name(row['Patient Name']): row['PRN'] for _, row in df_app_filtered.iterrows()}
+        patient_data_for_automation = []
+        for _, row in df_quantum.iterrows():
+            prn = prn_lookup.get(standardize_name(row['Name']), "")
+            if prn:
+                patient_data_for_automation.append({'Name': row['Name'], 'PRN': prn})
+        # Save the original quantum data for the final report
+        session_data[session_id]['patient_data_for_report'] = df_quantum[['Name']].to_dict('records')
+        session_data[session_id]['patient_data'] = patient_data_for_automation
+        socketio.emit('data_processed')
+        return jsonify({"message": "Data processed successfully."})
+    except Exception as 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('initialize_session')
 def handle_init(data):
     session_id = 'user_session'
+    session_data[session_id] = {'emails': data['emails'], 'filename': data['filename']}
+    # Bot creation is now handled by the /process endpoint
 @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 - Two-Phase Processing")
+    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,7 +1,6 @@
 import time
 import threading
 import subprocess
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options as ChromeOptions
@@ -10,64 +9,36 @@ from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException
 class QuantumBot:
     def __init__(self, socketio, app):
-        self.socketio = socketio
-        self.app = app
-        self.driver = None
-        self.DEFAULT_TIMEOUT = 30
-        self.termination_event = threading.Event()
     def _kill_chrome_processes(self):
-        """Kill any existing Chrome processes to prevent conflicts"""
         try:
-            subprocess.run(['pkill', '-f', 'chrome'], capture_output=True, text=True, timeout=5)
-            subprocess.run(['pkill', '-f', 'chromium'], capture_output=True, text=True, timeout=5)
             time.sleep(1)
-        except:
             pass
     def initialize_driver(self):
         try:
             self.micro_status("Initializing headless browser...")
-            # Kill any existing Chrome processes first
             self._kill_chrome_processes()
-            options = ChromeOptions()
-            options.binary_location = "/usr/bin/chromium"
-            # DON'T set --user-data-dir - let Chrome handle it automatically
-            # This prevents the "already in use" error
-            # Minimal required flags for HuggingFace Spaces
-            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("--remote-debugging-port=0")
             options.add_argument("--window-size=1920,1080")
-            options.add_argument("--no-first-run")
-            options.add_argument("--disable-extensions")
-            # User agent for better compatibility
             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)
-            # Anti-detection
             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):
@@ -76,144 +47,93 @@ class QuantumBot:
             self.socketio.emit('micro_status_update', {'message': message})
     def stop(self):
-        self.micro_status("Termination signal received. Finishing current patient...")
-        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()
             self.micro_status("Waiting for OTP screen...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.presence_of_element_located((By.ID, "code1"))
-            )
             return True, None
         except Exception as e:
-            error_message = f"Error during login: {str(e)}"
-            print(f"[Bot] 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] ERROR during OTP submission: {error_message}")
             return False, error_message
-    def process_patient_list(self, patient_list):
         results = []
-        for index, patient_name in enumerate(patient_list):
-            if self.termination_event.is_set():
-                print("[Bot] Termination detected. Stopping process.")
-                break
             with self.app.app_context():
                 self.socketio.emit('stats_update', {
-                    'processed': len(results),
-                    'remaining': len(patient_list) - len(results)
                 })
-            self.micro_status(f"Processing '{patient_name}' ({index + 1}/{len(patient_list)})...")
-            status = self._process_single_patient(patient_name)
-            results.append({'Name': patient_name, 'Status': status})
-            with self.app.app_context():
-                self.socketio.emit('log_update', {'name': patient_name, 'status': status})
         return results
-    def _process_single_patient(self, patient_name):
         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 for patient (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)
-                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 while processing {patient_name}: {e}")
-            return 'Error'
     def shutdown(self):
         try:
-            if self.driver:
-                self.driver.quit()
             self._kill_chrome_processes()
             print("[Bot] Chrome session closed and cleaned up.")
-        except Exception as e:
-            print(f"[Bot] Error during shutdown: {e}")

 import time
 import threading
 import subprocess
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options as ChromeOptions
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException
 class QuantumBot:
     def __init__(self, socketio, app):
+        self.socketio = socketio; self.app = app; self.driver = None
+        self.DEFAULT_TIMEOUT = 30; self.termination_event = threading.Event()
     def _kill_chrome_processes(self):
         try:
+            subprocess.run(['pkill', '-f', 'chromium'], check=True, timeout=5)
             time.sleep(1)
+        except Exception:
             pass
     def initialize_driver(self):
         try:
             self.micro_status("Initializing headless browser...")
             self._kill_chrome_processes()
+            options = ChromeOptions(); options.binary_location = "/usr/bin/chromium"
+            options.add_argument("--headless=new"); options.add_argument("--no-sandbox")
+            options.add_argument("--disable-dev-shm-usage"); options.add_argument("--disable-gpu")
             options.add_argument("--window-size=1920,1080")
             user_agent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
             options.add_argument(f"--user-agent={user_agent}")
             service = ChromeService(executable_path="/usr/bin/chromedriver")
             self.driver = webdriver.Chrome(service=service, options=options)
             self.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
                 'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
             })
             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):
             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()
             self.micro_status("Waiting for OTP screen...")
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "code1")))
             return True, None
         except Exception as e:
+            error_message = f"Error during login: {str(e)}"; print(f"[Bot] 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] ERROR during OTP submission: {error_message}")
             return False, error_message
+    def process_patient_list(self, patient_data):
         results = []
+        for index, record in enumerate(patient_data):
+            if self.termination_event.is_set(): print("[Bot] Termination detected."); break
+            patient_name = record['Name']; patient_prn = record['PRN']
+            self.micro_status(f"Processing '{patient_name}' ({index + 1}/{len(patient_data)})...")
+            status = self._process_single_patient(patient_name, patient_prn)
+            results.append({'Name': patient_name, 'PRN': patient_prn, 'Status': status})
             with self.app.app_context():
+                self.socketio.emit('log_update', {'name': patient_name, 'prn': patient_prn, 'status': status})
                 self.socketio.emit('stats_update', {
+                    'processed': len(results), 'remaining': len(patient_data) - len(results), 'status': status
                 })
         return results
+    def _process_single_patient(self, patient_name, patient_prn):
         try:
             self.micro_status(f"Navigating to Void page for '{patient_name}'")
             self.driver.get("https://gateway.quantumepay.com/credit-card/void")
             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)
+                if patient_prn and pd.notna(patient_prn):
+                    self.micro_status(f"Entering PRN: {patient_prn}...")
+                    contact_input = WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.NAME, "last_name")))
+                    contact_input.clear(); contact_input.send_keys(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 while processing {patient_name}: {e}"); return 'Error'
     def shutdown(self):
         try:
+            if self.driver: self.driver.quit()
             self._kill_chrome_processes()
             print("[Bot] Chrome session closed and cleaned up.")
+        except Exception as e: print(f"[Bot] Error during shutdown: {e}")