Viskit Studio

自托管商品视觉工作台。

图片预览与二次编辑	商品套包概览

现在能做什么

• 商品图上传
• 商品信息识别
• 结构化 brief
• 输出计划编辑
• 支持白底图、模特图、试穿/买家秀、Banner、详情模块、14 图套包
• 生成任务队列
• 图片预览、下载、删除、二次编辑
• 模板库
• 服务商配置
• SQLite / PostgreSQL

未来事项

• 支持素材向量嵌入：为模板、历史生成图、用户上传参考图和可复用素材建立 embedding 索引。
• 支持参考素材检索：根据商品 brief、类目、目标图位、平台尺寸和风格意图，检索相似构图、色彩、场景和历史成功案例。
• 向量仅作为素材参考层：向量检索不直接决定最终画面，只用于找到可参考的素材、构图和风格线索。
• 增强提示词组合：将商品信息、模板目标、参考素材特征、品牌约束、色彩锁定和负面约束组合成更稳定的生图提示词。
• 支持参考图风格统一：筛选后的参考图可作为 image edit / reference image 输入，交给支持该能力的生图模型统一视觉风格。
• 建立反馈闭环：用户下载、采纳、二次编辑或废弃的结果回写为素材质量信号，用于后续检索排序。

规划中的参考流程：

flowchart TD
    A["商品信息 / brief / 模板目标"] --> B["向量检索参考素材"]
    B --> C["筛选素材<br/>类目、授权、品牌、色彩、构图、质量"]
    C --> D["组合提示词<br/>商品身份 + 模板目标 + 参考素材特征 + 风格锁定"]
    D --> E["参考图输入生图模型"]
    E --> F["统一风格后生成视觉套包"]

Loading

Docker 部署

docker-compose.yml：

services:
  app:
    image: mayurikano/viskit-studio:latest
    ports:
      - "3068:3000"
    environment:
      DATABASE_URL: sqlite:////app/data/viskit.db
      CONFIG_PATH: /app/data/config.yaml
      IMAGEGEN_OUTPUT_DIR: /app/data/imagegen
      VISKIT_AUTO_MIGRATE: "1"
      VISKIT_BOOTSTRAP_WORKSPACE: "1"
      NEXT_SERVER_API_BASE_URL: http://127.0.0.1:8000
    volumes:
      - ./data:/app/data
    restart: unless-stopped

源码部署

安装 make：

# Debian / Ubuntu
sudo apt update && sudo apt install -y make

# macOS
xcode-select --install

# Windows
winget install -e --id GnuWin32.Make

Windows 安装后，把这个目录加入系统 Path，然后重启 PowerShell / CMD：

C:\Program Files (x86)\GnuWin32\bin

验证：

make -v

准备：

make install-prod

构建：

make build

启动：

make start

开发者模式

make bootstrap
cp .env.example .env
mkdir -p data
cp config.yaml.example data/config.yaml
make db-migrate
make dev

检查

make lint
make typecheck
make test
make web-build

环境变量

变量	用途
VISKIT_WEB_PORT	Web 端口，默认 3068
VISKIT_DATA_DIR	宿主机数据目录，默认 ./data
VISKIT_DATABASE_URL	Docker 里的数据库地址，默认 SQLite
VISKIT_GENERATION_JOB_CONCURRENCY	单个生成任务并发数，默认 4
NEXT_PUBLIC_API_BASE_URL	前端外部 API 地址；同源部署一般留空
OPENAI_API_KEY	OpenAI 或 OpenAI 兼容接口密钥
ANTHROPIC_API_KEY	Anthropic 兼容接口密钥
APIMART_API_KEY	API Mart 密钥
CHATGPT2API_KEY	chatgpt2api 密钥
CPA_API_KEY	CPA / CLIProxyAPI 密钥
SUB2API_API_KEY	Sub2API 密钥
COMPLIANCE_SCREEN_API_KEY	合规检查服务密钥，可选

服务商配置

data/config.yaml：

providers:
  vision:
    protocol: openai_compatible
    base_url: https://api.example.com
    api_key_env: OPENAI_API_KEY
    model: gpt-4o

  llm:
    protocol: openai_compatible
    base_url: https://api.example.com
    api_key_env: OPENAI_API_KEY
    model: gpt-4.1

  image:
    protocol: openai_compatible
    base_url: https://api.example.com
    api_key_env: OPENAI_API_KEY
    model: gpt-image-2

  compliance_screen:
    protocol: openai_compatible
    base_url: https://api.example.com
    api_key_env: OPENAI_API_KEY
    model: gpt-4.1-mini

图片接口

OpenAI 兼容：

providers:
  image:
    protocol: openai_compatible
    base_url: http://your-gateway:3000
    api_key_env: OPENAI_API_KEY
    model: gpt-image-2

接口：

• POST /v1/images/generations
• POST /v1/images/edits

通用反代 / 网关：

• CPA / CLIProxyAPI: https://github.com/router-for-me/CLIProxyAPI
• Sub2API: https://github.com/Wei-Shaw/sub2api
• New API: https://github.com/QuantumNous/new-api
• chatgpt2api: https://github.com/basketikun/chatgpt2api

image_generation：

providers:
  image:
    protocol: image_generation
    adapter: chatgpt2api
    base_url: http://host.docker.internal:8317
    api_key_env: CHATGPT2API_KEY
    model: gpt-image-2

适配器：

gemini
gemini_openai
openai
chatgpt2api
volcengine_ark
z_image_gitee
jimeng2api
grok
siliconflow_adapter

PostgreSQL

UV_EXTRAS=--extra postgres
VISKIT_DATABASE_URL=postgresql+psycopg://viskit:viskit@postgres:5432/viskit
POSTGRES_DB=viskit
POSTGRES_USER=viskit
POSTGRES_PASSWORD=viskit

生成结果接口

示例：

{
  "job_id": "gen_xxx",
  "status": "succeeded",
  "outputs": [
    {
      "output_id": "out_xxx",
      "status": "succeeded",
      "image_url": "/api/generation/jobs/gen_xxx/outputs/out_xxx/image",
      "download_url": "/api/generation/jobs/gen_xxx/outputs/out_xxx/image",
      "image_id": "asset:asset_xxx"
    }
  ]
}

图片响应：

Content-Type: image/png
Cache-Control: no-store

图片 ID：

asset:{asset_id}
kit-slot:{kit_id}:{slot_id}

目录

apps/api/              FastAPI 后端
apps/web/              Next.js 前端
apps/web/public/minipaint/
                       内置 miniPaint 静态资源
services/copywriter/   文案、合规、OCR
services/editor/       图片编辑
services/imagegen/     生图编排和模板
services/providers/    服务商适配
packages/schemas/      共享 schema
infra/                 PostgreSQL 与迁移

致谢

• chatgpt2api: https://github.com/basketikun/chatgpt2api
• miniPaint: https://github.com/viliusle/miniPaint

License

MIT. See LICENSE.