Bug-Fixed

Dockerfile

@@ -1,3 +1,4 @@
+# Dockerfile
 FROM python:3.11-slim
 
 WORKDIR /app
@@ -7,11 +8,25 @@ ENV TZ=Etc/UTC
 ENV HOME=/home/appuser
 ENV PYTHONUSERBASE=/home/appuser/.local
 
+# System deps + Chromium + ChromeDriver + common headless libs
 RUN apt-get update && apt-get install -y --no-install-recommends \
+    wget \
+    gnupg \
+    unzip \
+    ca-certificates \
     chromium \
     chromium-driver \
+    libnss3 \
+    libxss1 \
+    libasound2 \
+    libxtst6 \
+    libatk-bridge2.0-0 \
+    libgtk-3-0 \
+    libgbm1 \
+    fonts-liberation \
     && rm -rf /var/lib/apt/lists/*
 
+# Create unprivileged user
 RUN addgroup --system appuser && adduser --system --ingroup appuser appuser
 RUN mkdir -p /home/appuser/.local && chown -R appuser:appuser /app /home/appuser
 
@@ -26,4 +41,4 @@ COPY --chown=appuser:appuser . .
 
 EXPOSE 7860
 
-CMD ["python", "server.py"]
+CMD ["python", "server.py"]

requirements.txt

@@ -7,4 +7,4 @@ python-dotenv
 selenium
 google-api-python-client
 google-auth-httplib2
-google-auth-oauthlib
+google-auth-oauthlib

server.py

@@ -1,3 +1,4 @@
+# server.py
 import eventlet
 eventlet.monkey_patch()
 
@@ -23,7 +24,7 @@ load_dotenv()
 
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret-key-for-hillside-automation'
-FRONTEND_ORIGIN = os.getenv('FRONTEND_URL', 'http://127.0.0.1:5500') 
+FRONTEND_ORIGIN = os.getenv('FRONTEND_URL', 'http://127.0.0.1:5500')
 CORS(app, resources={r"/*": {"origins": FRONTEND_ORIGIN}})
 socketio = SocketIO(app, cors_allowed_origins=FRONTEND_ORIGIN, async_mode='eventlet')
 
@@ -36,7 +37,8 @@ class EmailService:
         self.sender_email = os.getenv('EMAIL_SENDER')
         self.password = os.getenv('EMAIL_PASSWORD')
         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.")
+        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:
@@ -139,8 +141,7 @@ def handle_init(data):
     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)
-    
-    # --- Definitive Fix: Capture and send detailed error message ---
+
     is_success, error_message = bot_instance.initialize_driver()
     if is_success:
         emit('bot_initialized')
@@ -174,4 +175,4 @@ if __name__ == '__main__':
     print("  Make sure all secrets are set in your Hugging Face Space.")
     print("  Listening on http://127.0.0.1:5000")
     print("====================================================================")
-    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))
+    socketio.run(app, host='0.0.0.0', port=int(os.getenv('PORT', 7860)))

worker.py

@@ -1,17 +1,18 @@
-import pandas as pd
+# worker.py
 import time
-import io
 import os
 import threading
 import tempfile
 import shutil
+
 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.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
-from selenium.common.exceptions import TimeoutException, NoSuchElementException, ElementNotInteractableException
+from selenium.common.exceptions import TimeoutException
+
 
 class QuantumBot:
     def __init__(self, socketio, app):
@@ -20,31 +21,39 @@ class QuantumBot:
         self.driver = None
         self.DEFAULT_TIMEOUT = 30
         self.termination_event = threading.Event()
-        self.temp_dir = None
+        self.chrome_user_data_dir = None
+        self.chrome_cache_dir = None
 
     def initialize_driver(self):
         try:
-            self.micro_status("Initializing isolated headless browser...")
+            self.micro_status("Initializing headless browser...")
             options = ChromeOptions()
+
+            # Use system Chromium installed in the image
             options.binary_location = "/usr/bin/chromium"
-            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"
-            
-            self.temp_dir = tempfile.mkdtemp()
 
-            options.add_argument(f"user-agent={user_agent}")
+            # 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}")
+
+            # 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"
+            options.add_argument(f"--user-agent={user_agent}")
             options.add_argument("--headless=new")
             options.add_argument("--window-size=1920,1080")
+            options.add_argument("--no-first-run")
+            options.add_argument("--no-default-browser-check")
             options.add_argument("--no-sandbox")
             options.add_argument("--disable-dev-shm-usage")
             options.add_argument("--disable-gpu")
-            options.add_argument(f"--user-data-dir={self.temp_dir}")
-            options.add_argument('--disable-blink-features=AutomationControlled')
-            options.add_experimental_option("excludeSwitches", ["enable-automation"])
-            options.add_experimental_option('useAutomationExtension', False)
-            
+            options.add_argument("--disable-features=Translate,AutomationControlled")
+
             service = ChromeService(executable_path="/usr/bin/chromedriver")
             self.driver = webdriver.Chrome(service=service, options=options)
-            
+
+            # De-emphasize webdriver flag
             self.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
                 'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
             })
@@ -69,11 +78,19 @@ class QuantumBot:
             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()
+            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")))
+            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)}"
@@ -86,9 +103,13 @@ class QuantumBot:
             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()
+            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']")))
+            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)}"
@@ -117,50 +138,70 @@ 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:
                     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']")))
+                    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.")
-            
+                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()
-            
+            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()
-            
+            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 = 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()
+                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()
+                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):
         if self.driver:
             self.driver.quit()
-        if self.temp_dir and os.path.exists(self.temp_dir):
-            shutil.rmtree(self.temp_dir)
-            print(f"[Bot] Cleaned up temporary directory: {self.temp_dir}")
+        # 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.")