Bug-Fixed

server.py

@@ -39,7 +39,6 @@ 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:
@@ -53,8 +52,7 @@ 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):
@@ -64,15 +62,12 @@ 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:
@@ -82,18 +77,15 @@ 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
@@ -102,7 +94,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()
@@ -141,15 +133,13 @@ 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()
@@ -160,31 +150,24 @@ 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,7 +1,6 @@
 # worker.py
 import time
 import threading
-import os
 
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
@@ -28,8 +27,8 @@ class QuantumBot:
             # Use system Chromium installed in the image
             options.binary_location = "/usr/bin/chromium"
 
-            # Do NOT set --user-data-dir; let ChromeDriver create a temp profile
-            # Use an ephemeral DevTools port to avoid collisions
+            # Do NOT set --user-data-dir; let ChromeDriver create a fresh temp profile per session
+            # Use an ephemeral DevTools port to avoid collisions in constrained environments
             options.add_argument("--remote-debugging-port=0")
 
             # Stable/headless startup flags
@@ -47,7 +46,7 @@ class QuantumBot:
             service = ChromeService(executable_path="/usr/bin/chromedriver")
             self.driver = webdriver.Chrome(service=service, options=options)
 
-            # De-emphasize webdriver flag
+            # Reduce webdriver detection
             self.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
                 'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
             })