trying fixes for loading

src/proxy_lite/browser/browser.py

@@ -185,48 +185,48 @@ class BrowserSession:
     )
     async def update_poi(self) -> None:
         try:
-            # Added for robustness based on previous discussions
-            await self.current_page.wait_for_load_state("networkidle", timeout=180000)
-            logger.debug("wait_for_load_state('networkidle') completed.")
+            # We will use "domcontentloaded" as a base and then wait for specific elements
+            await self.current_page.wait_for_load_state("domcontentloaded", timeout=60000) # Reduced timeout for initial load
+            logger.debug("wait_for_load_state('domcontentloaded') completed.")
+
+            # --- MODIFICATION START ---
+            # Wait for the "Account Forecasting" heading to be visible
+            # Adjust the selector below based on the actual HTML of the Salesforce page.
+            # Common selectors could be:
+            # - `h1:has-text('Account Forecasting')`
+            # - `h2:has-text('Account Forecasting')`
+            # - `div.some-class-name:has-text('Account Forecasting')`
+            # - `[data-qa-id="account-forecasting-heading"]` (if Salesforce uses data-qa attributes)
+            # You might need to inspect the Salesforce page to get the exact selector.
+            # For now, let's assume it's an h1 or h2 tag containing the text.
+            try:
+                await self.current_page.wait_for_selector(
+                    "h1:has-text('Account Forecasting'), h2:has-text('Account Forecasting')",
+                    timeout=60000, # Set a reasonable timeout for this specific element
+                    state="visible"
+                )
+                logger.debug("Successfully waited for 'Account Forecasting' heading.")
+            except PlaywrightTimeoutError as e:
+                logger.error(f"Timeout waiting for 'Account Forecasting' heading on URL: {self.current_page.url}")
+                # You might want to log more specific HTML/screenshot here if this still times out often
+                raise # Re-raise if this critical element doesn't appear
 
-            # This is the line that was previously timing out, now with increased timeout.
-            # Adding explicit try/except around it for specific debugging.
+            # It's still good to wait for the body to be visible, but with a shorter timeout
+            # if the previous specific heading check passed.
             try:
-                await self.current_page.wait_for_selector("body", timeout=180000, state="visible")
+                await self.current_page.wait_for_selector("body", timeout=30000, state="visible")
                 logger.debug("wait_for_selector('body', state='visible') completed.")
             except PlaywrightTimeoutError as e:
-                # --- START TEMPORARY DEBUGGING CODE ---
-                current_url = self.current_page.url if self.current_page else "N/A"
-                logger.error(f"DEBUGGING: Playwright Timeout (180s) on body selector at URL: {current_url}")
-
-                html_content = None
-                try:
-                    if self.current_page:
-                        html_content = await self.current_page.content()
-                        # Log only a snippet of HTML to avoid excessively large logs
-                        logger.error(f"DEBUGGING: HTML Content (first 1000 chars) when timeout occurred:\n{html_content[:1000]}...")
-                except Exception as html_e:
-                    logger.error(f"DEBUGGING: Could not get HTML content: {html_e}")
-
-                screenshot_b64 = "N/A"
-                try:
-                    if self.current_page:
-                        # Capture screenshot at lower quality to keep log size manageable
-                        screenshot_bytes = await self.current_page.screenshot(type="jpeg", quality=50)
-                        screenshot_b64 = base64.b64encode(screenshot_bytes).decode("utf-8")
-                        # Log only a very short snippet of base64 string
-                        logger.error(f"DEBUGGING: Base64 Screenshot (truncated) when timeout occurred:\ndata:image/jpeg;base64,{screenshot_b64[:100]}... (full string is much longer)")
-                except Exception as ss_e:
-                    logger.error(f"DEBUGGING: Could not take screenshot: {ss_e}")
-
-                # Re-raise the original exception to ensure the task still fails,
-                # but now with crucial debugging information in the logs.
-                raise e
-                # --- END TEMPORARY DEBUGGING CODE ---
+                logger.warning(f"DEBUGGING: Playwright Timeout (30s) on body selector, but 'Account Forecasting' heading was found. This might be acceptable if the page is usable.")
+                # We can choose to suppress this specific timeout if the critical element is found,
+                # or re-raise it if a fully loaded body is strictly necessary for further actions.
+                # For now, let's just log and continue, as the primary issue was full page load.
+                pass # Do not re-raise if we've found our key indicator.
+
 
         except PlaywrightTimeoutError: # This outer catch is for the wait_for_load_state timeout
-            logger.error(f"Timeout waiting for website load state (networkidle): {self.current_url}")
-            raise # Re-raise if load_state itself times out
+            logger.error(f"Timeout waiting for website load state (domcontentloaded): {self.current_url}")
+            raise # Re-raise if initial load_state itself times out
 
         except Exception as e:
             logger.error(f"An unexpected error occurred during page readiness check: {e}")