Bug

server.py

@@ -23,51 +23,14 @@ load_dotenv()
 
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret-key-for-hillside-automation'
-CORS(app, resources={r"/*": {"origins": "*"}})
-socketio = SocketIO(app, cors_allowed_origins="*", async_mode='eventlet')
+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')
 
 bot_instance = None
 session_data = {}
 automation_thread = None
 
-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
-            
-            # --- Definitive Secure Credential Loading ---
-            base64_creds = os.getenv('GDRIVE_SA_KEY_BASE64')
-            if not base64_creds:
-                raise ValueError("GDRIVE_SA_KEY_BASE64 secret not found.")
-            if not self.folder_id:
-                raise ValueError("GOOGLE_DRIVE_FOLDER_ID secret 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}")
-
-    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
-
 class EmailService:
     def __init__(self):
         self.sender_email = os.getenv('EMAIL_SENDER')
@@ -89,6 +52,31 @@ class EmailService:
             print(f"Email sent successfully to {', '.join(recipients)}"); return True
         except Exception as e: print(f"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("[G-Drive] Service initialized securely from secrets.")
+        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:
+            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 = EmailService()
 drive_service = GoogleDriveService()
 
@@ -138,44 +126,43 @@ def generate_and_send_reports(session_id, results, is_crash_report=False, is_ter
 
 @app.route('/')
 def status_page():
-    APP_STATUS_HTML = """
-    <!DOCTYPE html><html lang="en"><head><title>API Status</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:#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's Automation API</h1><div class="indicator active">● Active</div></div></body></html>
-    """
+    APP_STATUS_HTML = """<!DOCTYPE html><html lang="en"><head><title>API Status</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:#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's Automation API</h1><div class="indicator active">● Active</div></div></body></html>"""
     return Response(APP_STATUS_HTML)
 
 @socketio.on('connect')
 def handle_connect():
-    print('Frontend connected.'); 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'
+    session_id = 'user_session'; global bot_instance
     session_data[session_id] = {'csv_content': data['content'], 'emails': data['emails'], 'filename': data['filename']}
-    global bot_instance
     if bot_instance: bot_instance.shutdown()
     bot_instance = QuantumBot(socketio, app)
-    if bot_instance.initialize_driver(): emit('bot_initialized')
-    else: emit('error', {'message': 'Failed to initialize the automation bot.'})
+    
+    # --- Definitive Fix: Capture and send detailed error message ---
+    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.'})
-    if bot_instance.login(credentials['username'], credentials['password']): emit('otp_required')
-    else: emit('error', {'message': 'Login failed. Please check credentials.'})
+    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.'})
-    if bot_instance.submit_otp(data['otp']):
+    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, session_id=session_id)
-    else: emit('error', {'message': 'OTP was incorrect or timed out.'})
+        socketio.start_background_task(target=run_automation_process, args=(session_id,))
+    else: emit('error', {'message': f'OTP failed: {error_message}'})
 
 @socketio.on('terminate_process')
 def handle_terminate():

worker.py

@@ -6,9 +6,6 @@ import threading
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from selenium.webdriver.chrome.options import Options as ChromeOptions
-from webdriver_manager.chrome import ChromeDriverManager
-# --- Definitive Fix: Import ChromeType from its new, correct location ---
-from webdriver_manager.core.os_manager import ChromeType
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
@@ -31,23 +28,25 @@ class QuantumBot:
             options.add_argument("--window-size=1920,1080")
             options.add_argument("--no-sandbox")
             options.add_argument("--disable-dev-shm-usage")
+            options.add_argument("--disable-gpu") # Added for stability in Linux environments
             options.add_argument('--disable-blink-features=AutomationControlled')
             options.add_experimental_option("excludeSwitches", ["enable-automation"])
             options.add_experimental_option('useAutomationExtension', False)
             
-            os.environ['WDM_LOCAL'] = '/home/appuser/.wdm'
-            service = ChromeService(ChromeDriverManager(chrome_type=ChromeType.CHROMIUM).install())
+            # --- Definitive Fix: Use the explicit driver path installed by Docker ---
+            # This removes the dependency on webdriver-manager for production.
+            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
+            return True, None
         except Exception as e:
-            print(f"CRITICAL ERROR in WebDriver Initialization: {e}")
-            self.micro_status(f"Error: Could not start browser. {e}")
-            return False
+            error_message = f"Error during WebDriver initialization: {str(e)}"
+            print(f"CRITICAL ERROR in WebDriver Initialization: {error_message}")
+            return False, error_message
 
     def micro_status(self, message):
         print(f"[Bot Action] {message}")
@@ -69,10 +68,11 @@ class QuantumBot:
             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
+            return True, None
         except Exception as e:
-            print(f"[Bot] ERROR during login: {e}")
-            return False
+            error_message = f"Error during login process: {str(e)}"
+            print(f"[Bot] ERROR during login: {error_message}")
+            return False, error_message
 
     def submit_otp(self, otp):
         try:
@@ -83,10 +83,11 @@ class QuantumBot:
             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
+            return True, None
         except Exception as e:
-            print(f"[Bot] ERROR during OTP submission: {e}")
-            return False
+            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 = []
@@ -94,17 +95,11 @@ class QuantumBot:
             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.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
@@ -113,7 +108,6 @@ 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:
@@ -127,16 +121,13 @@ 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(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")))