Spaces:
Runtime error
Runtime error
Refactor VideoTemplateEditor component: remove AI editing features and related state management, streamline template loading, and update styles for improved layout. Enhance CSS for responsive design and visual consistency across various components, including dropdowns, sync stats, and quiz elements. Remove deprecated premium badge styles and adjust layout for chat components.
66de9c8 | """An infographic's PNG: its stored HTML page drawn by a headless Chrome at the page's own size. No Remotion, no | |
| browser library: one `chrome --headless --screenshot` run per download (the result is kept for the next one).""" | |
| import functools | |
| import glob | |
| import hashlib | |
| import os | |
| import re | |
| import shutil | |
| import subprocess | |
| import tempfile | |
| from collections import OrderedDict | |
| from pathlib import Path | |
| from threading import Lock | |
| from app.infographics.render import SIZES | |
| TIMEOUT = 90 # seconds for one drawing | |
| SCALE = 2 # pixels per page pixel: a 1600 x 2260 page is drawn as 3200 x 4520, sharp on any screen and in print | |
| KEEP = 6 # drawings kept in memory | |
| _cache: OrderedDict[str, bytes] = OrderedDict() | |
| _lock = Lock() | |
| class ImageUnavailable(Exception): | |
| """No Chrome to draw with, or it failed.""" | |
| def _runs(path: str) -> bool: | |
| """Whether the binary actually starts (a package manager's wrapper can outlive the browser it points at).""" | |
| try: | |
| return subprocess.run([path, "--version"], capture_output=True, timeout=20).returncode == 0 | |
| except (subprocess.SubprocessError, OSError): | |
| return False | |
| def chrome() -> str | None: | |
| """CHROME, or a working Chrome / Chromium found on the machine.""" | |
| if env := os.environ.get("CHROME"): | |
| return env if Path(env).exists() else None | |
| found = [p for name in ("chromium", "chromium-browser", "google-chrome", "google-chrome-stable", "chrome") | |
| if (p := shutil.which(name))] | |
| # Playwright's headless shell first: desktop Chrome on macOS can hang taking a headless screenshot. | |
| shell = "chromium_headless_shell-*/chrome-headless-shell-*/chrome-headless-shell" | |
| patterns = [os.path.expanduser(f"~/Library/Caches/ms-playwright/{shell}"), os.path.expanduser(f"~/.cache/ms-playwright/{shell}"), | |
| "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome", | |
| os.path.expanduser("~/.cache/puppeteer/chrome/*/chrome-mac-*/" | |
| "Google Chrome for Testing.app/Contents/MacOS/Google Chrome for Testing")] | |
| found += [p for pat in patterns for p in glob.glob(pat)] | |
| return next((p for p in found if _runs(p)), None) | |
| def render_png(html: str, layout: str = "portrait") -> bytes: | |
| """The page as a PNG, drawn at SCALE times its size: tall (1600 x 2260) or wide (2260 x 1600) before scaling.""" | |
| width, height = SIZES.get(layout, SIZES["portrait"]) | |
| key = hashlib.sha256(f"{width}x{height}@{SCALE}{html}".encode()).hexdigest() | |
| with _lock: | |
| if key in _cache: | |
| _cache.move_to_end(key) | |
| return _cache[key] | |
| binary = chrome() | |
| if not binary: | |
| raise ImageUnavailable("No Chrome is installed here to draw the image.") | |
| page = f'<!doctype html><meta charset="utf-8"><body style="margin:0">{html}</body>' | |
| with tempfile.TemporaryDirectory() as tmp: | |
| src, out = Path(tmp) / "page.html", Path(tmp) / "out.png" | |
| src.write_text(page, encoding="utf-8") | |
| cmd = [binary, "--headless=new", "--disable-gpu", "--no-sandbox", "--hide-scrollbars", f"--force-device-scale-factor={SCALE}", | |
| f"--user-data-dir={tmp}/profile", f"--window-size={width},{height}", "--virtual-time-budget=8000", | |
| f"--screenshot={out}", f"file://{src}"] | |
| try: | |
| subprocess.run(cmd, check=True, capture_output=True, timeout=TIMEOUT) | |
| except (subprocess.SubprocessError, OSError) as exc: | |
| raise ImageUnavailable("The image could not be drawn.") from exc | |
| if not out.exists(): | |
| raise ImageUnavailable("The image could not be drawn.") | |
| data = out.read_bytes() | |
| with _lock: | |
| _cache[key] = data | |
| while len(_cache) > KEEP: | |
| _cache.popitem(last=False) | |
| return data | |
| _FIT = re.compile(r"FIT=([0-9.]+)") | |
| _MEASURE = ("<script>document.fonts.ready.then(function(){var b=document.querySelector('.ig-body'),i=document.querySelector('.ig-inner');" | |
| "var s=1;if(b&&i){var inner=i.scrollHeight*parseFloat(getComputedStyle(i).getPropertyValue('--fit')||1);s=Math.min(1,b.clientHeight/Math.max(inner,1));}" | |
| "document.title='FIT='+s.toFixed(3);});</script>") | |
| def measure_fit(html: str, layout: str) -> float: | |
| """The scale (1 or less) that makes an AI-composed page's body fit its space: the page drawn once, in Chrome, with our own | |
| small script measuring it (the page's markup is sanitised, so no other script is in it). 1 when Chrome is missing.""" | |
| width, height = SIZES.get(layout, SIZES["portrait"]) | |
| binary = chrome() | |
| if not binary: | |
| return 1.0 | |
| page = f'<!doctype html><meta charset="utf-8"><body style="margin:0">{html}{_MEASURE}</body>' | |
| with tempfile.TemporaryDirectory() as tmp: | |
| src = Path(tmp) / "page.html" | |
| src.write_text(page, encoding="utf-8") | |
| cmd = [binary, "--headless=new", "--disable-gpu", "--no-sandbox", "--hide-scrollbars", f"--user-data-dir={tmp}/profile", | |
| f"--window-size={width},{height}", "--virtual-time-budget=10000", "--dump-dom", f"file://{src}"] | |
| try: | |
| out = subprocess.run(cmd, check=True, capture_output=True, timeout=TIMEOUT, text=True).stdout | |
| except (subprocess.SubprocessError, OSError): | |
| return 1.0 | |
| found = _FIT.search(out) | |
| return max(0.55, min(1.0, float(found.group(1)))) if found else 1.0 | |