diagnose_progress.py

import os
import time
import json
import logging
from selenium import webdriver
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.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from webdriver_manager.chrome import ChromeDriverManager
import sys

# Configure logging to debug level
logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s - %(levelname)s - %(message)s",
    handlers=[
        logging.StreamHandler(),
        logging.FileHandler("diagnose_progress.log", encoding="utf-8"),
    ],
)
logger = logging.getLogger(__name__)


class ProgressDiagnostic:
    def __init__(self, headless=False):
        self.headless = headless
        self.driver = None
        # macOS Chrome profile path
        self.profile_dir = os.path.expanduser(
            "~/Library/Application Support/Google/Chrome/SeleniumProfile"
        )
        os.makedirs(self.profile_dir, exist_ok=True)

    def setup_driver(self):
        chrome_options = Options()
        if self.headless:
            chrome_options.add_argument("--headless=new")

        chrome_options.add_argument("--no-sandbox")
        chrome_options.add_argument("--disable-dev-shm-usage")
        chrome_options.add_argument("--disable-blink-features=AutomationControlled")
        chrome_options.add_argument("--disable-extensions")
        chrome_options.add_argument("--start-maximized")
        chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
        chrome_options.add_experimental_option("useAutomationExtension", False)
        chrome_options.add_argument(f"--user-data-dir={self.profile_dir}")

        try:
            service = Service(ChromeDriverManager().install())
            self.driver = webdriver.Chrome(service=service, options=chrome_options)
            self.driver.implicitly_wait(10)
            self.driver.execute_script(
                "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
            )
            logger.info("Chrome driver setup completed successfully")
        except Exception as e:
            logger.error(f"Failed to setup Chrome driver: {e}")
            raise

    def go_to_flow(self):
        try:
            url = "https://labs.google/fx/tools/flow/project/0bdaaba0-dfa6-42ed-8a3c-e0d760d490b9"
            self.driver.get(url)
            WebDriverWait(self.driver, 30).until(
                EC.presence_of_element_located((By.TAG_NAME, "body"))
            )
            logger.info(f"Opened: {url}")
        except Exception as e:
            logger.error(f"Failed to navigate to Flow tool: {e}")
            raise

    def diagnose_progress_elements(self):
        """Diagnose progress indicator elements during generation"""
        try:
            logger.info("=== Progress Indicator Diagnosis ===")

            logger.debug("=== DOM Structure Analysis ===")
            logger.debug(f"Current URL: {self.driver.current_url}")
            logger.debug(f"Page Title: {self.driver.title}")

            logger.debug("\n=== All Elements with '%' Character ===")
            percentage_elements = self.driver.find_elements(
                By.XPATH, "//*[contains(text(), '%')]"
            )
            for i, elem in enumerate(percentage_elements):
                if elem.is_displayed():
                    logger.debug(f"Element {i}:")
                    logger.debug(f"  Tag: {elem.tag_name}")
                    logger.debug(f"  Class: {elem.get_attribute('class')}")
                    logger.debug(f"  Text: '{elem.text}'")
                    logger.debug(f"  Style: '{elem.get_attribute('style')}'")
                    logger.debug(f"  Aria: '{elem.get_attribute('aria-valuenow')}'")
                    logger.debug("---")

            logger.debug("\n=== Progress Bar Elements ===")
            progress_selectors = [
                "//*[@role='progressbar']",
                "//*[contains(@class, 'progress')]",
                "//*[contains(@class, 'progress-bar')]",
            ]

            for selector in progress_selectors:
                elements = self.driver.find_elements(By.XPATH, selector)
                logger.debug(f"Selector '{selector}': {len(elements)} elements")
                for i, elem in enumerate(elements):
                    if elem.is_displayed():
                        logger.debug(f"  Element {i}:")
                        logger.debug(f"    Tag: {elem.tag_name}")
                        logger.debug(f"    Class: {elem.get_attribute('class')}")
                        logger.debug(f"    Text: '{elem.text}'")
                        logger.debug(f"    Style: '{elem.get_attribute('style')}'")
                        logger.debug(
                            f"    Aria: '{elem.get_attribute('aria-valuenow')}'"
                        )

            logger.debug("\n=== Generation-Related Elements ===")
            generation_selectors = [
                "//*[contains(text(), 'Generating')]",
                "//*[contains(text(), 'Processing')]",
                "//*[contains(@class, 'generating')]",
                "//*[contains(@class, 'processing')]",
            ]

            for selector in generation_selectors:
                elements = self.driver.find_elements(By.XPATH, selector)
                logger.debug(f"Selector '{selector}': {len(elements)} elements")
                for i, elem in enumerate(elements):
                    if elem.is_displayed():
                        logger.debug(f"  Element {i}:")
                        logger.debug(f"    Text: '{elem.text}'")
                        logger.debug(f"    Class: '{elem.get_attribute('class')}'")

            logger.debug("\n=== Download Button ===")
            download_selectors = [
                "//*[contains(text(), 'Download')]",
                "//*[contains(@class, 'download')]",
            ]

            for selector in download_selectors:
                elements = self.driver.find_elements(By.XPATH, selector)
                logger.debug(f"Selector '{selector}': {len(elements)} elements")
                for i, elem in enumerate(elements):
                    if elem.is_displayed():
                        logger.debug(f"  Element {i}:")
                        logger.debug(f"    Text: '{elem.text}'")
                        logger.debug(f"    Class: '{elem.get_attribute('class')}'")
                        logger.debug(f"    Enabled: {elem.is_enabled()}")

            logger.info("\n=== Diagnosis Complete ===")

        except Exception as e:
            logger.error(f"Diagnosis failed: {e}")
            import traceback

            logger.error(f"Stack trace: {traceback.format_exc()}")

    def save_page_snapshot(self):
        """Save a snapshot of the current page for debugging"""
        try:
            # Save HTML source
            with open("page_snapshot.html", "w", encoding="utf-8") as f:
                f.write(self.driver.page_source)

            # Save screenshot
            self.driver.save_screenshot("page_screenshot.png")

            logger.info(
                "Page snapshot saved as page_snapshot.html and page_screenshot.png"
            )

        except Exception as e:
            logger.error(f"Failed to save snapshot: {e}")

    def close(self):
        if self.driver:
            try:
                self.driver.quit()
                logger.info("Chrome driver closed successfully")
            except Exception as e:
                logger.error(f"Error closing Chrome driver: {e}")


if __name__ == "__main__":
    print("Flow Progress Indicator Diagnostics\n")
    print("This will help diagnose why the progress percentage isn't displaying.")
    print("\nSteps:")
    print("1. Browser will open with your saved profile")
    print("2. Navigate to the image generation interface")
    print("3. Start an image generation manually")
    print("4. When you see the progress indicator, press Enter here\n")

    try:
        diag = ProgressDiagnostic(headless=False)
        diag.setup_driver()
        diag.go_to_flow()

        input(
            "\nPress Enter when you have started an image generation and see a progress indicator: "
        )

        diag.diagnose_progress_elements()
        diag.save_page_snapshot()

        print("\n✅ Diagnosis complete!")
        print("Check diagnose_progress.log for detailed debugging info")
        print("View page_screenshot.png for visual state")
        print("View page_snapshot.html for full DOM structure")

    except Exception as e:
        logger.error(f"Error: {e}")
        print(f"\nError: {e}")
    finally:
        if "diag" in locals():
            diag.close()