Spaces:
Running on Zero
Running on Zero
File size: 3,555 Bytes
6b01bf1 4cbd96b 6b01bf1 13047a3 6b01bf1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | """Click through the UI in headless Chrome and save screenshots.
Usage: python scripts/ui_check.py [--url http://127.0.0.1:7860] [--out shots/] [--dark] [--width 1440] [--run]
Without --run nothing uses the GPU: the recorded results are on the page from the start. With --run
it presses Run live once and waits for the live result.
"""
from __future__ import annotations
import argparse
import sys
from pathlib import Path
from playwright.sync_api import sync_playwright
def main() -> int:
ap = argparse.ArgumentParser()
ap.add_argument("--url", default="http://127.0.0.1:7860")
ap.add_argument("--out", default="shots")
ap.add_argument("--dark", action="store_true")
ap.add_argument("--width", type=int, default=1440)
ap.add_argument("--run", action="store_true", help="press Run live once (uses GPU time on a Space)")
ap.add_argument("--timeout", type=int, default=240)
args = ap.parse_args()
out = Path(args.out)
out.mkdir(parents=True, exist_ok=True)
tag = f"{args.width}{'-dark' if args.dark else ''}"
url = args.url.rstrip("/") + ("/?__theme=dark" if args.dark else "/?__theme=light")
errors: list[str] = []
with sync_playwright() as p:
browser = p.chromium.launch(channel="chrome", headless=True)
page = browser.new_page(viewport={"width": args.width, "height": 1000})
# A failed request counts only on the app's own origin: on a Space, the Hub's page wrapper also
# asks huggingface.co for an organisation avatar, which 404s for a user account.
origin = args.url.rstrip("/")
page.on("response", lambda r: errors.append(f"{r.status} {r.url}")
if r.status >= 400 and r.url.startswith(origin) else None)
page.on("console", lambda m: errors.append(f"console.{m.type}: {m.text}")
if m.type == "error" and not m.text.startswith("Failed to load resource") else None)
page.on("pageerror", lambda e: errors.append(f"pageerror: {e}"))
page.goto(url, wait_until="load", timeout=180_000)
slider = page.locator('input[type="range"]').first
slider.wait_for(timeout=180_000)
page.wait_for_timeout(2500)
page.screenshot(path=str(out / f"1-steer-alpha0-{tag}.png"), full_page=True)
slider.focus()
page.keyboard.press("End")
page.wait_for_timeout(1200)
page.screenshot(path=str(out / f"2-steer-alpha+2-{tag}.png"), full_page=True)
page.keyboard.press("Home")
page.wait_for_timeout(1200)
page.screenshot(path=str(out / f"3-steer-alpha-2-{tag}.png"), full_page=True)
page.get_by_label("Storyteller").check()
page.wait_for_timeout(1500)
page.screenshot(path=str(out / f"4-storyteller-{tag}.png"), full_page=True)
if args.run:
page.get_by_role("button", name="Run live").click()
page.get_by_text("measured just now").or_(page.get_by_text("from the Space's cache")).first.wait_for(
timeout=args.timeout * 1000)
page.wait_for_timeout(1200)
page.screenshot(path=str(out / f"5-live-{tag}.png"), full_page=True)
width = page.evaluate("document.documentElement.scrollWidth - document.documentElement.clientWidth")
if width > 0:
errors.append(f"horizontal overflow of {width}px")
browser.close()
for e in errors:
print(e)
print(f"screenshots in {out.resolve()} ({len(errors)} problems)")
return 1 if errors else 0
if __name__ == "__main__":
sys.exit(main())
|