Bug-Fixed

server.py

@@ -39,6 +39,7 @@ class EmailService:
         self.smtp_server = "smtp.gmail.com"; self.smtp_port = 587
         if not self.sender_email or not self.password:
             print("[Email] WARNING: Email credentials not found in secrets.")
+
     def send_report(self, recipients, subject, body, attachments=None):
         if not self.sender_email or not self.password: return False
         try:
@@ -52,7 +53,8 @@ class EmailService:
             with smtplib.SMTP(self.smtp_server, self.smtp_port) as server:
                 server.starttls(); server.login(self.sender_email, self.password); server.send_message(msg)
             print(f"Email sent successfully to {', '.join(recipients)}"); return True
-        except Exception as e: print(f"Failed to send email: {e}"); return False
+        except Exception as e:
+            print(f"Failed to send email: {e}"); return False
 
 class GoogleDriveService:
     def __init__(self):
@@ -62,12 +64,15 @@ class GoogleDriveService:
             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.")
+            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 securely from secrets.")
-        except Exception as e: print(f"[G-Drive] CRITICAL ERROR: Could not initialize Google Drive service: {e}")
+        except Exception as e:
+            print(f"[G-Drive] CRITICAL ERROR: Could not initialize Google Drive service: {e}")
+
     def upload_file(self, filename, file_content):
         if not self.service: return False
         try:
@@ -77,15 +82,18 @@ class GoogleDriveService:
             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
+        except Exception as e:
+            print(f"[G-Drive] ERROR: File upload failed: {e}"); return False
 
 email_service = EmailService()
 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 []
+        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
@@ -94,7 +102,7 @@ def run_automation_process(session_id):
         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()
+        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()
@@ -133,13 +141,15 @@ def status_page():
 
 @socketio.on('connect')
 def handle_connect():
-    print(f'Frontend connected. Allowed origin: {FRONTEND_ORIGIN}'); emit('email_list', {'emails': get_email_list()})
+    print(f'Frontend connected. Allowed origin: {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()
+    if bot_instance:
+        bot_instance.shutdown()
     bot_instance = QuantumBot(socketio, app)
 
     is_success, error_message = bot_instance.initialize_driver()
@@ -150,24 +160,31 @@ def handle_init(data):
 
 @socketio.on('start_login')
 def handle_login(credentials):
-    if not bot_instance: return emit('error', {'message': 'Bot not initialized.'})
+    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}'})
+    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.'})
+    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(target=run_automation_process, args=(session_id,))
-    else: emit('error', {'message': f'OTP failed: {error_message}'})
+    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 bot_instance:
+        print("Termination signal received.")
+        bot_instance.stop()
 
 if __name__ == '__main__':
     print("====================================================================")

worker.py

@@ -1,9 +1,7 @@
 # worker.py
 import time
-import os
 import threading
-import tempfile
-import shutil
+import os
 
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
@@ -21,8 +19,6 @@ class QuantumBot:
         self.driver = None
         self.DEFAULT_TIMEOUT = 30
         self.termination_event = threading.Event()
-        self.chrome_user_data_dir = None
-        self.chrome_cache_dir = None
 
     def initialize_driver(self):
         try:
@@ -32,11 +28,9 @@ class QuantumBot:
             # Use system Chromium installed in the image
             options.binary_location = "/usr/bin/chromium"
 
-            # Per-run, unique profile and cache to avoid "already in use" locks
-            self.chrome_user_data_dir = tempfile.mkdtemp(prefix="chrome-user-data-")
-            self.chrome_cache_dir = tempfile.mkdtemp(prefix="chrome-cache-")
-            options.add_argument(f"--user-data-dir={self.chrome_user_data_dir}")
-            options.add_argument(f"--disk-cache-dir={self.chrome_cache_dir}")
+            # Do NOT set --user-data-dir; let ChromeDriver create a temp profile
+            # Use an ephemeral DevTools port to avoid collisions
+            options.add_argument("--remote-debugging-port=0")
 
             # Stable/headless startup flags
             user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36"
@@ -138,6 +132,7 @@ class QuantumBot:
         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:
@@ -155,8 +150,10 @@ class QuantumBot:
                     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(
@@ -165,6 +162,7 @@ class QuantumBot:
             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']"))
@@ -172,6 +170,7 @@ class QuantumBot:
             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(
@@ -200,8 +199,4 @@ class QuantumBot:
     def shutdown(self):
         if self.driver:
             self.driver.quit()
-        # Cleanup per-run temp dirs
-        for d in [self.chrome_user_data_dir, self.chrome_cache_dir]:
-            if d and os.path.exists(d):
-                shutil.rmtree(d, ignore_errors=True)
-        print(f"[Bot] Cleaned up temporary Chrome profile and cache.")
+        print("[Bot] Chrome session closed.")