trycua--cua/libs/python/cua-sandbox/cua_sandbox/interfaces/screen.py

"""Screen interface — screenshots and screen info, backed by a Transport."""

from __future__ import annotations

import base64
from typing import Tuple

from cua_sandbox.transport.base import Transport


class Screen:
    """Screen capture and info."""

    def __init__(self, transport: Transport):
        self._t = transport

    async def screenshot(self, format: str = "png", quality: int = 95) -> bytes:
        """Capture a screenshot and return raw image bytes.

        Args:
            format: "png" (lossless, default) or "jpeg" (lossy, ~5-10x smaller).
            quality: JPEG quality 1-95, ignored for PNG.
        """
        return await self._t.screenshot(format=format, quality=quality)

    async def screenshot_base64(self, format: str = "png", quality: int = 95) -> str:
        """Capture a screenshot and return as a base64-encoded string."""
        raw = await self._t.screenshot(format=format, quality=quality)
        return base64.b64encode(raw).decode("ascii")

    async def size(self) -> Tuple[int, int]:
        """Return (width, height) of the screen."""
        d = await self._t.get_screen_size()
        return d["width"], d["height"]