Skip to main content

AI-driven cross-platform UI automation engine with test script generation

Project description

ScreenForge

CI Python 3.11+ License: MIT

中文 | English

Your AI agent runs the test. You keep the pytest file.

Describe the flow in plain language. Watch the agent click through a live site. Walk away with a self-healing pytest script that survives UI changes — and runs in CI. (Drives real iOS + Android devices too, not just Chrome.)

ScreenForge is an AI-driven UI automation engine that turns natural language into executable test scripts. Unlike record-and-replay tools, you don't perform the actions yourself — the AI does it for you.

ScreenForge Live Mirror — the generated pytest grows beside the live page as the agent runs

Why ScreenForge?

Playwright Codegen Browser Use Midscene.js ScreenForge
Need to perform actions yourself? Yes No No No
Generates replayable test scripts? Yes No No Yes (pytest)
Self-healing when UI changes? No No No Yes
Works as AI Agent tool (MCP)? No Yes No Yes
Drives real iOS / Android devices? No No No Yes

Core architecture: Your AI Agent is the brain (understands requirements, makes decisions). ScreenForge is the hands (executes UI actions, generates code).

Quick Start

pip install screenforge

# See the magic without any API key:
screenforge --demo

# For real usage, set your LLM key:
export OPENAI_API_KEY=sk-...

# Inspect the current page (returns DOM tree for your Agent to analyze):
echo '{"operation":"inspect_ui","platform":"web"}' | screenforge --tool-stdin

# Execute a single action:
screenforge --action click --platform web --locator-type text --locator-value "Login"

How It Works

You (or your AI Agent)          ScreenForge
        │                            │
        ├──── "Test the login" ─────►│
        │                            ├── inspect_ui (get DOM tree)
        │◄── DOM tree ──────────────┤
        │                            │
        ├──── click #email ─────────►│
        ├──── input "user@..." ─────►│
        ├──── click "Sign In" ──────►│
        │                            │
        │◄── pytest script ─────────┤
        │◄── Allure report ─────────┤

Each step: inspect → decide → act → verify. The AI decides, ScreenForge executes.

Features

  • Keep a real pytest file: Every run emits a replayable pytest script (Allure-instrumented) you can drop straight into CI — not a black-box agent run.
  • Self-healing engine: When the UI changes and a locator breaks, the engine auto-repairs it with confidence scoring and AST validation, so your committed test survives.
  • The agent does the clicking: You (or your AI agent) describe intent; ScreenForge inspects the DOM, acts, and verifies — you never record by hand.
  • Real devices, not just browsers: Drives Android (uiautomator2) and iOS (wda) physical devices over the same protocol — the one thing pure-web agents (Playwright MCP, Browser Use) can't do.
  • L1/L2 semantic cache: Same page + same instruction = instant response, no LLM call needed
  • Visual fallback: When DOM can't locate elements (Canvas, games), VLM parses screenshots
  • MCP server: Any MCP-compatible Agent can drive ScreenForge natively
  • Structured output: JSON Lines events + report/runs/<id>/ artifacts for CI integration
  • Live Mirror playground: Watch the generated pytest code grow line-by-line beside a live screenshot as the test runs — screenforge --playground. See the Playground Guide
  • Execution-replay review report: Run any test with REVIEW_RECORD=1 and get a self-contained, shareable offline report.html — a drag-timeline of each step's screenshot + the test source line that triggered it + the DOM tree, plus a video.gif filmstrip. Default off, zero overhead. (Web verified; mobile registry/recording seam in place but unverified.)
  • Case memory hints on inspect_ui: pass an optional task label and the payload's memory field carries what the last run of that task did — locator, counts, and outcome — a hint the Agent weighs against the live tree, seen before it acts, never auto-replayed. It records failures too, so check last_status/success_count before trusting a hint. The label must match the one the earlier run recorded (--action-name, else "{action}:{locator_value}"); load_case_memory lists them. See the Agent Guide.
  • Plain-language element lookup on inspect_ui: pass an optional intent phrase (e.g. "登录按钮") and the payload's candidates field returns up to 3 ranked elements with ready-to-use locators — additive to the full ui_tree, never a replacement. Matching is literal string similarity, not semantic, so phrase intent in the same language as the UI. See the Agent Guide.

Agent Integration (Claude Code / Cursor / Codex)

ScreenForge exposes itself as a tool for AI Agents. The standard loop:

# 1. Get page structure (your Agent analyzes it)
echo '{"operation":"inspect_ui","platform":"web"}' | screenforge --tool-stdin

# 2. Your Agent decides what to do, sends precise actions
screenforge --action click --platform web --locator-type text --locator-value "Login"

# 3. Verify the result, repeat
echo '{"operation":"inspect_ui","platform":"web"}' | screenforge --tool-stdin

For batch operations, use workflows:

screenforge --workflow ./workflows/login.yaml --platform web --json

Or start the MCP server for native Agent integration:

screenforge --mcp-server

GitHub Actions

Add ScreenForge to your CI pipeline:

- uses: jhinzzz/ScreenForge@v1
  with:
    platform: web
    workflow: ./workflows/login.yaml
    openai-api-key: ${{ secrets.OPENAI_API_KEY }}

Results are auto-uploaded as Allure artifacts. See action.yml for all inputs.

See Agent Integration Guide for the complete protocol.

Installation (from source)

git clone https://github.com/jhinzzz/ScreenForge.git
cd ScreenForge
python -m venv .venv && source .venv/bin/activate
pip install -e ".[all]"

Platform-specific extras:

  • Web only: pip install -e . (default, includes Playwright)
  • Android: pip install -e ".[android]"
  • iOS: pip install -e ".[ios]"
  • ML/cache: pip install -e ".[ml]" (sentence-transformers for semantic cache)

Configuration

# Required: LLM API key (OpenAI-compatible endpoint)
export OPENAI_API_KEY=sk-...

# Optional: custom endpoint (defaults to api.openai.com)
export OPENAI_BASE_URL=https://api.openai.com/v1

# Optional: model (defaults to gpt-4o)
export MODEL_NAME=gpt-4o

Or create a .env file (copy from .env_template).

Badge

If ScreenForge generates tests for your project, add this badge to your README:

[![Tests by ScreenForge](https://img.shields.io/badge/tests%20by-ScreenForge-blue?logo=pytest)](https://github.com/jhinzzz/ScreenForge)

Tests by ScreenForge

Learn More

Resource Description
Mobile Setup Android & iOS device connection guide
MCP Setup (3 min) Connect to Claude Desktop / Cursor / Cline / Claude Code
Agent Guide Integration protocol for AI Agents
Capability Matrix Supported platforms, actions, and locators
Architecture Deep-Dive Brain/hands split, semantic cache, self-heal AST gates, hygiene-as-feature
Examples Real committed workflows + the green pytest they generated
Playground Guide Live Mirror — watch code + screenshots grow as the test runs
Workflow Examples YAML workflow templates
CHANGELOG Version history

Contributing

See CONTRIBUTING.md for guidelines. Issues and PRs welcome!

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

screenforge-0.10.0.tar.gz (233.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

screenforge-0.10.0-py3-none-any.whl (151.1 kB view details)

Uploaded Python 3

File details

Details for the file screenforge-0.10.0.tar.gz.

File metadata

  • Download URL: screenforge-0.10.0.tar.gz
  • Upload date:
  • Size: 233.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for screenforge-0.10.0.tar.gz
Algorithm Hash digest
SHA256 2758ac40ae1c8637bdfbf99b51632f4493359b162f15c08f0a41c8ec3672ed4a
MD5 c8bdf2a75ae313cfbb214d4dc5527fd8
BLAKE2b-256 b7af7fd72334f1faabcb70cd0c7148b56df2ddb38c28dea7dc6efdf5e13c4cc7

See more details on using hashes here.

Provenance

The following attestation bundles were made for screenforge-0.10.0.tar.gz:

Publisher: release.yml on jhinzzz/ScreenForge

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file screenforge-0.10.0-py3-none-any.whl.

File metadata

  • Download URL: screenforge-0.10.0-py3-none-any.whl
  • Upload date:
  • Size: 151.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for screenforge-0.10.0-py3-none-any.whl
Algorithm Hash digest
SHA256 9220a15206307077b9a38ab2a6bbe829dc3c3b9649214a5e43c3eb25f81b6aa6
MD5 a0af735a4d732e26373d1ab6c6f17451
BLAKE2b-256 d778f08c936cbcae41ec9b7a945c15c7b540077a29dcc8e5ec6e1739f41c7532

See more details on using hashes here.

Provenance

The following attestation bundles were made for screenforge-0.10.0-py3-none-any.whl:

Publisher: release.yml on jhinzzz/ScreenForge

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page