AI Painter demo

AI Painter Studio turns a drawing or image into finished artwork with a local SDXL-Turbo image-to-image model. It is tuned for the included GTX 1070 setup and runs without sending images to a hosted service.

Windows 10 setup

Requirements:

• 64-bit Python 3.12
• An NVIDIA GPU with a current driver
• About 20 GB of free disk space for PyTorch, Python packages, and model files

From PowerShell in this folder:

Set-ExecutionPolicy -Scope Process Bypass
.\setup_windows.ps1

Then double-click:

run_demo.bat

The first launch downloads SDXL-Turbo and can take several minutes. Later launches use the local model cache. The desktop launcher runs without a console window. For troubleshooting output, use:

.\launch_app.ps1 -Console

Studio workflow

1. Draw on the canvas or choose Open image.
2. Describe the finished image in the right-hand prompt panel.
3. Choose a style, calibrated creativity level, and seed.
4. Select Generate with SDXL-Turbo.
5. Review the Result tab, save it, or move it back to the canvas.

Results are automatically saved in generated, and the current session keeps the eight most recent outputs in the history panel.

The creativity control uses a fixed ten-step schedule and a bounded nonlinear strength curve. Its low end still applies a visible style, while its high end retains source composition instead of becoming a complete redraw. The default 50% setting is the balanced starting point.

Safety controls

AI Painter applies three local safety gates:

1. Prompts requesting sexual, graphic, hateful, or other inappropriate content are rejected before generation.
2. The source canvas or opened image must pass an image-safety classifier.
3. Every generated result is checked before display or saving. Rejected results are discarded.

The image gate uses Falconsai/nsfw_image_detection on the CPU with a conservative threshold. These controls also apply to pix2pix_test.py and run_demo.ps1, so the script path does not bypass the desktop policy.

Scripted use

The same SDXL-Turbo pipeline can be run without the GUI:

.\.venv\Scripts\python.exe .\pix2pix_test.py `
  --input ".\beach.png" `
  --prompt "A luminous watercolor landscape on textured paper" `
  --output ".\generated\watercolor.png" `
  --strength 0.8 `
  --steps 4 `
  --seed 42

Or use the interactive PowerShell file picker:

.\run_demo.ps1