root/ubuntu-post-install
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

paintplus: vendor the app source and rename from EditmaskwithAI

Browse Source 
Bring the full EditmaskwithAI application into the repo under paintplus/
(429 files) so the service is self-contained — the installer copies the
vendored source to ~/docker/paintplus/src instead of cloning at runtime.

Rename to PaintPlus (service + branding; app logic untouched):
- services/editmaskwithai.sh -> services/paintplus.sh (register_service
  paintplus, install_paintplus, ~/docker/paintplus, Caddy paintplus:8000,
  Authelia option preserved)
- container names -> paintplus across docker-compose*.yml; dev network
  -> paintplus-network
- browser <title> -> "PaintPlus - AI Image Editor"; README heading ->
  PaintPlus with upstream provenance note
- README utilities table: editmaskwithai -> paintplus

Backend/frontend code (help strings referencing the old container name,
the ai_photo_edit.db filename) is intentionally left as-is to avoid
touching application logic.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Nb2vJ8W7bHKx1JXVvpCraH
This commit is contained in:
Claude
2026-06-26 05:48:43 +00:00
parent b4e8ba2a79
commit 084922afaa
431 changed files with 87396 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
services/editmaskwithai.sh → services/paintplus.sh
+88 -76
View File
@@ -1,74 +1,81 @@
#!/bin/bash
# services/editmaskwithai.sh — Self-hosted AI photo editor: paint a mask over an
# object, describe the change, and AI replaces only that region.
# Source: https://github.com/outis1one/EditmaskwithAI
# services/paintplus.sh — PaintPlus: self-hosted AI photo editor. Paint a mask over
# an object, describe the change, and AI replaces only that region.
#
# The full application source is vendored in this repo under ./paintplus and is
# copied to ~/docker/paintplus/src at install time (no network clone).
# Based on EditmaskwithAI (github.com/outis1one/EditmaskwithAI).
# Part of the modular post-install system (sourced by setup.sh).
#
# Two deployment modes (chosen at install):
# Cloud — no GPU; uses a cloud provider (OpenAI gpt-image / Replicate). Lightweight.
# GPU — local inference via the repo's own installer (NVIDIA, downloads ~13 GB).
# GPU — local inference via the app's own installer (NVIDIA, downloads ~13 GB).
#
# The app reads config from a .env file the compose file interpolates
# The app reads config from a .env the compose file interpolates
# (${AI_PROVIDER}, ${OPENAI_API_KEY}, ${SECRET_KEY}, ...). No env_file: directive.
# No built-in auth — protect with Authelia via Caddy.
register_service editmaskwithai utilities "AI photo editor — mask a region, AI replaces it (EditmaskwithAI)" 3080
register_service paintplus utilities "AI photo editor — mask a region, AI replaces it (PaintPlus)" 3080
install_editmaskwithai() {
install_paintplus() {
require_docker || return 1
log_info "Installing EditmaskwithAI (mask-based AI photo editor)..."
log_info "Installing PaintPlus (mask-based AI photo editor)..."
local EMA_DIR="$DOCKER_DIR/editmaskwithai"
local REPO_URL="https://github.com/outis1one/EditmaskwithAI.git"
# Vendored application source lives in this repo at <repo>/paintplus
local SELF_DIR SRC_DIR
SELF_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
SRC_DIR="$(cd "$SELF_DIR/.." && pwd)/paintplus"
local PP_DIR="$DOCKER_DIR/paintplus"
local APP_DIR="$PP_DIR/src"
if [ "$DRY_RUN" = true ]; then
echo "[DRY-RUN] Would clone $REPO_URL to $EMA_DIR"
echo "[DRY-RUN] Would copy vendored source $SRC_DIR -> $APP_DIR"
echo "[DRY-RUN] Would prompt for deployment mode (Cloud API / Local GPU)"
echo "[DRY-RUN] Would create .env from .env.example (AI_PROVIDER, API key, SECRET_KEY)"
echo "[DRY-RUN] Would create .env (AI_PROVIDER, API key, SECRET_KEY)"
echo "[DRY-RUN] Cloud: docker compose up -d --build | GPU: install-local-gpu.sh + bring-up-local-gpu.sh"
echo "[DRY-RUN] Would offer Authelia SSO and configure Caddy (ai-photo-edit:8000, host port 3080)"
echo "[DRY-RUN] Would offer Authelia SSO and configure Caddy (paintplus:8000, host port 3080)"
return 0
fi
# ── Clone / update ────────────────────────────────────────────────────────
if [ -d "$EMA_DIR/.git" ]; then
log_info "Updating EditmaskwithAI repo..."
git -C "$EMA_DIR" pull --ff-only 2>/dev/null \
&& log_success "Repo updated" \
|| log_warning "Could not pull latest — using existing checkout"
else
log_info "Cloning EditmaskwithAI..."
git clone --depth 1 "$REPO_URL" "$EMA_DIR" \
|| { log_error "Clone failed — check network and git access"; return 1; }
if [ ! -d "$SRC_DIR" ]; then
log_error "Vendored PaintPlus source not found at $SRC_DIR"
return 1
fi
ensure_docker_dir_ownership "$EMA_DIR"
cd "$EMA_DIR" || return 1
# ── Copy vendored source into the docker dir ──────────────────────────────
# Refreshes app files; leaves a user-edited .env and the override untouched
# (neither is part of the vendored source).
mkdir -p "$APP_DIR"
cp -a "$SRC_DIR/." "$APP_DIR/"
ensure_docker_dir_ownership "$PP_DIR"
cd "$APP_DIR" || return 1
# ── Deployment mode ───────────────────────────────────────────────────────
echo ""
log_info "Deployment mode:"
log_info " 1) Cloud API No GPU needed. Uses OpenAI (gpt-image) or Replicate. Lightweight."
log_info " 2) Local GPU NVIDIA GPU; runs the repo's installer and downloads ~13 GB of models."
log_info " 2) Local GPU NVIDIA GPU; runs the app's installer and downloads ~13 GB of models."
echo ""
local EMA_MODE=""
prompt_text "Mode [1=Cloud]:" "1" EMA_MODE
local PP_MODE=""
prompt_text "Mode [1=Cloud]:" "1" PP_MODE
# ── .env from the repo's template ─────────────────────────────────────────
# ── .env from the app's template ──────────────────────────────────────────
if [ -f .env.example ]; then
cp .env.example .env
else
log_warning ".env.example missing from repo — writing a fresh .env"
log_warning ".env.example missing — writing a fresh .env"
: > .env
fi
# Upsert KEY=VALUE into ./.env (drop any existing/commented line, then append)
_ema_set_env() {
_pp_set_env() {
sed -i -E "/^#?[[:space:]]*$1=/d" .env
printf '%s=%s\n' "$1" "$2" >> .env
}
local EMA_PROVIDER="local_gpu"
if [[ "$EMA_MODE" != "2" ]]; then
local PP_PROVIDER="local_gpu"
if [[ "$PP_MODE" != "2" ]]; then
echo ""
log_info "Cloud provider:"
log_info " 1) OpenAI gpt-image editing. Key: https://platform.openai.com/api-keys"
@@ -78,34 +85,34 @@ install_editmaskwithai() {
prompt_text "Provider [1=OpenAI]:" "1" _prov
local _key=""
if [[ "$_prov" == "2" ]]; then
EMA_PROVIDER="replicate"
PP_PROVIDER="replicate"
prompt_text "Replicate API key (enter to set later):" "" _key
if [ -n "$_key" ]; then _ema_set_env REPLICATE_API_KEY "$_key"
if [ -n "$_key" ]; then _pp_set_env REPLICATE_API_KEY "$_key"
else log_warning "No key entered — set REPLICATE_API_KEY in .env later"; fi
else
EMA_PROVIDER="openai"
PP_PROVIDER="openai"
prompt_text "OpenAI API key (enter to set later):" "" _key
if [ -n "$_key" ]; then _ema_set_env OPENAI_API_KEY "$_key"
if [ -n "$_key" ]; then _pp_set_env OPENAI_API_KEY "$_key"
else log_warning "No key entered — set OPENAI_API_KEY in .env later"; fi
fi
else
local _hf=""
prompt_text "HuggingFace token (optional — for gated models, enter to skip):" "" _hf
[ -n "$_hf" ] && _ema_set_env HF_TOKEN "$_hf"
[ -n "$_hf" ] && _pp_set_env HF_TOKEN "$_hf"
fi
_ema_set_env AI_PROVIDER "$EMA_PROVIDER"
_ema_set_env SECRET_KEY "$(generate_password 48)"
_pp_set_env AI_PROVIDER "$PP_PROVIDER"
_pp_set_env SECRET_KEY "$(generate_password 48)"
chmod 600 .env
mkdir -p data
ensure_docker_dir_ownership "$EMA_DIR"
ensure_docker_dir_ownership "$PP_DIR"
# ── Caddy network override (cloud mode) ───────────────────────────────────
# The base compose has no networks, so Caddy (on caddy_net) can't reach the
# container by name. An override file — auto-merged with the default compose —
# attaches the app to caddy_net. The GPU compose is run with an explicit -f and
# container by name. This override — auto-merged with the default compose —
# attaches the app to caddy_net. The GPU compose runs with an explicit -f and
# does NOT merge overrides, so GPU mode is wired with `docker network connect`.
if [[ "$EMA_MODE" != "2" ]] && [ -d "$DOCKER_DIR/caddy" ]; then
if [[ "$PP_MODE" != "2" ]] && [ -d "$DOCKER_DIR/caddy" ]; then
cat > docker-compose.override.yml << OVR
# Added by ubuntu-post-install so Caddy (on caddy_net) can reach this app by name.
services:
@@ -117,64 +124,64 @@ networks:
external: true
name: ${SITE_CADDY_NET:-caddy_net}
OVR
ensure_docker_dir_ownership "$EMA_DIR/docker-compose.override.yml"
fi
# ── Bring the stack up ────────────────────────────────────────────────────
local START_EMA=""
if [[ "$EMA_MODE" == "2" ]]; then
local START_PP=""
if [[ "$PP_MODE" == "2" ]]; then
echo ""
log_warning "GPU mode runs the repo's installer (NVIDIA toolkit, DNS fix) and downloads ~13 GB."
prompt_yn "Run the local-GPU installer now? (y/n):" "y" START_EMA
if [[ "$START_EMA" =~ ^[Yy]$ ]]; then
log_warning "GPU mode runs the app's installer (NVIDIA toolkit, DNS fix) and downloads ~13 GB."
prompt_yn "Run the local-GPU installer now? (y/n):" "y" START_PP
if [[ "$START_PP" =~ ^[Yy]$ ]]; then
if [ -f install-local-gpu.sh ] && [ -f bring-up-local-gpu.sh ]; then
chmod +x install-local-gpu.sh bring-up-local-gpu.sh 2>/dev/null || true
bash install-local-gpu.sh || log_warning "install-local-gpu.sh reported an error — see output above"
bash bring-up-local-gpu.sh || log_warning "bring-up-local-gpu.sh failed — check: docker compose -f docker-compose.gpu.yml logs"
else
log_warning "GPU scripts not found in repo — start manually per the README"
log_warning "GPU scripts not found — start manually per src/README.md"
fi
fi
else
prompt_yn "Build and start EditmaskwithAI now? (y/n):" "y" START_EMA
if [[ "$START_EMA" =~ ^[Yy]$ ]]; then
prompt_yn "Build and start PaintPlus now? (y/n):" "y" START_PP
if [[ "$START_PP" =~ ^[Yy]$ ]]; then
docker compose up -d --build \
&& log_success "EditmaskwithAI started" \
&& log_success "PaintPlus started" \
|| log_warning "Start failed — check: docker compose logs"
fi
fi
# Ensure the running container is on caddy_net (covers GPU mode, where the
# override file above is not merged by the repo's bring-up script).
if [ -d "$DOCKER_DIR/caddy" ] && [[ "$START_EMA" =~ ^[Yy]$ ]]; then
docker network connect "$SITE_CADDY_NET" ai-photo-edit 2>/dev/null || true
# override file above is not merged by the app's bring-up script).
if [ -d "$DOCKER_DIR/caddy" ] && [[ "$START_PP" =~ ^[Yy]$ ]]; then
docker network connect "$SITE_CADDY_NET" paintplus 2>/dev/null || true
fi
# ── Auth + Caddy ──────────────────────────────────────────────────────────
local EXTRA_BLOCK=""
if [ -d "$DOCKER_DIR/authelia" ]; then
local _use_auth=""
prompt_yn "Protect EditmaskwithAI with Authelia SSO? (y/n):" "y" _use_auth
prompt_yn "Protect PaintPlus with Authelia SSO? (y/n):" "y" _use_auth
[[ "$_use_auth" =~ ^[Yy]$ ]] && EXTRA_BLOCK=" import authelia"
fi
configure_caddy_for_service "EditmaskwithAI" "ai-photo-edit:8000" "editmask" "$EXTRA_BLOCK"
configure_caddy_for_service "PaintPlus" "paintplus:8000" "paintplus" "$EXTRA_BLOCK"
# ── README ────────────────────────────────────────────────────────────────
write_readme "$EMA_DIR" << MD
# EditmaskwithAI
# ── README (deploy notes; the app's own docs are at src/README.md) ────────
write_readme "$PP_DIR" << MD
# PaintPlus (deployment)
Self-hosted AI photo editor: paint a mask over any object, describe what you
want, and the AI replaces just that region — pixels outside the mask are left
untouched. Source: https://github.com/outis1one/EditmaskwithAI
want, and the AI replaces just that region. The application source is vendored
in ubuntu-post-install and copied here to \`src/\` (no network clone).
App docs: \`src/README.md\`. Based on EditmaskwithAI.
- URL: http://localhost:3080
- No built-in login — protect via Authelia SSO if exposed.
- Provider: set \`AI_PROVIDER\` in \`.env\` (openai, replicate, local_gpu, stability, comfyui, invokeai).
- Provider: set \`AI_PROVIDER\` in \`src/.env\` (openai, replicate, local_gpu, stability, comfyui, invokeai).
## Configure providers / keys
Edit \`.env\` then restart (the compose file interpolates these — there is no env_file):
Edit \`src/.env\` then restart (the compose file interpolates these — no env_file):
\`\`\`bash
cd $EMA_DIR
cd $APP_DIR
nano .env # AI_PROVIDER, OPENAI_API_KEY / REPLICATE_API_KEY, HF_TOKEN
docker compose up -d --build
\`\`\`
@@ -182,7 +189,7 @@ Keys: OpenAI https://platform.openai.com/api-keys · Replicate https://replicate
## Cloud mode (no GPU)
\`\`\`bash
cd $EMA_DIR
cd $APP_DIR
docker compose up -d --build # starts on http://localhost:3080
docker compose logs -f
docker compose down
@@ -190,23 +197,28 @@ docker compose down
## Local GPU mode (NVIDIA, ~13 GB of models)
\`\`\`bash
cd $EMA_DIR
cd $APP_DIR
./install-local-gpu.sh # toolkit + DNS fix + model prefetch
./bring-up-local-gpu.sh # docker compose -f docker-compose.gpu.yml up -d --build
./bring-up-local-gpu.sh logs -f
\`\`\`
GPU auto-selects models by VRAM (FLUX >=24 GB, SDXL 12-24 GB, SD 1.5 <2 GB).
## Update the app
Re-run the PaintPlus installer (copies the latest vendored source over \`src/\`,
keeping your \`src/.env\`), then rebuild:
\`\`\`bash
cd $APP_DIR && docker compose up -d --build
\`\`\`
## Caddy
Reverse-proxied as \`ai-photo-edit:8000\` on \`${SITE_CADDY_NET:-caddy_net}\`.
Cloud mode joins that network via \`docker-compose.override.yml\`; GPU mode is
attached with \`docker network connect\` after start (the GPU compose file does
not merge the override).
Reverse-proxied as \`paintplus:8000\` on \`${SITE_CADDY_NET:-caddy_net}\`. Cloud mode
joins that network via \`src/docker-compose.override.yml\`; GPU mode is attached
with \`docker network connect\` after start.
MD
echo ""
echo " URL: http://localhost:3080"
echo " Dir: $EMA_DIR"
echo " Provider: $EMA_PROVIDER (change AI_PROVIDER in $EMA_DIR/.env)"
echo " App dir: $APP_DIR"
echo " Provider: $PP_PROVIDER (change AI_PROVIDER in $APP_DIR/.env)"
echo ""
}