dsh-plugin-image-workbench 0.0.0-stage → 1.1.5

package/vendor/chatgpt2api/source/assets/kunrouai.svg

@@ -0,0 +1,50 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="24 20 378 110" role="img" aria-labelledby="kri-horizontal-title">
+    <title id="kri-horizontal-title">坤肉AI 横版标志</title>
+
+    <defs>
+        <linearGradient id="kri-horizontal-upper" gradientUnits="userSpaceOnUse" x1="64" y1="29" x2="78" y2="103">
+            <stop offset="0" stop-color="#CA47F4"/>
+            <stop offset=".44" stop-color="#8650FC"/>
+            <stop offset=".75" stop-color="#5855FF"/>
+            <stop offset="1" stop-color="#326CFF"/>
+        </linearGradient>
+        <linearGradient id="kri-horizontal-shine" gradientUnits="userSpaceOnUse" x1="82" y1="60" x2="139" y2="60">
+            <stop offset="0" stop-color="#74CEF1" stop-opacity="0"/>
+            <stop offset="1" stop-color="#74CEF1"/>
+        </linearGradient>
+        <linearGradient id="kri-horizontal-lower" gradientUnits="userSpaceOnUse" x1="76" y1="88" x2="119" y2="133">
+            <stop offset="0" stop-color="#1640C2"/>
+            <stop offset=".5" stop-color="#176BEA"/>
+            <stop offset="1" stop-color="#269DF5"/>
+        </linearGradient>
+        <linearGradient id="kri-horizontal-back" gradientUnits="userSpaceOnUse" x1="86" y1="57" x2="145" y2="59">
+            <stop offset="0" stop-color="#9854FC"/>
+            <stop offset=".6" stop-color="#6FBCF7"/>
+            <stop offset="1" stop-color="#2DC9EF"/>
+        </linearGradient>
+        <linearGradient id="kri-horizontal-orbit" gradientUnits="userSpaceOnUse" x1="45" y1="73" x2="68" y2="118">
+            <stop offset="0" stop-color="#6849FF"/>
+            <stop offset=".26" stop-color="#3D7CFF"/>
+            <stop offset=".59" stop-color="#30C5F5"/>
+            <stop offset=".8" stop-color="#B2F0FF"/>
+        </linearGradient>
+        <linearGradient id="kri-horizontal-star" gradientUnits="userSpaceOnUse" x1="143" y1="24" x2="156" y2="39">
+            <stop offset="0" stop-color="#348BF2"/>
+            <stop offset="1" stop-color="#42C4F6"/>
+        </linearGradient>
+        <linearGradient id="kri-horizontal-text" gradientUnits="userSpaceOnUse" x1="184" y1="75" x2="398" y2="75">
+            <stop offset="0" stop-color="#8650FC"/>
+            <stop offset=".55" stop-color="#326CFF"/>
+            <stop offset="1" stop-color="#269DF5"/>
+        </linearGradient>
+    </defs>
+
+    <path fill="url(#kri-horizontal-back)" d="M85.2 55.2C90.6 53 96.2 50.8 101.7 49.4L97.6 53.4C93.6 55.3 89.4 57.4 85.2 59ZM111.8 44.3C127.9 42.4 141.2 42.5 146.1 49.4C153.3 59.7 132.4 73 119.3 77.4C130.8 68.1 143.2 56.8 136.4 51.6C132.4 48.5 124.2 48.5 115.8 50Z"/>
+    <path fill="url(#kri-horizontal-lower)" d="M56.4 101.4C74.9 97.1 95.3 89.7 111.5 82.6L146.9 121.8H118.3C115.5 121.8 113.7 120.9 111.8 118.7L90.1 94.2L86.4 98.3C83.3 101.2 82.2 103.8 82.2 107.2V110.8C82.2 121.3 74.3 129.3 64.1 129.3H56.4Z"/>
+    <path fill="url(#kri-horizontal-upper)" d="M56.4 99.4V47.5C56.4 37.2 63.5 29.5 74.1 29.5H82.2V74.8L109.8 46.5C111.7 44.5 113.1 43.7 116 43.7H137.2L108.4 78.9C91.7 87.1 74.1 94.3 56.4 99.4Z"/>
+    <path fill="url(#kri-horizontal-shine)" d="M56.4 99.4V47.5C56.4 37.2 63.5 29.5 74.1 29.5H82.2V74.8L109.8 46.5C111.7 44.5 113.1 43.7 116 43.7H137.2L108.4 78.9C91.7 87.1 74.1 94.3 56.4 99.4Z"/>
+    <path fill="url(#kri-horizontal-orbit)" d="M52.4 70.6C39.3 78.8 24.6 91.5 24.7 98C24.9 112.4 67.4 106.5 110.3 80.6L108.4 78.9C76.7 93.5 43.8 102.9 36.6 96.1C31.1 90.9 44.2 79.7 52.4 75Z"/>
+    <path fill="url(#kri-horizontal-star)" d="M149.4 20.5C150.5 27.7 153.5 30.5 160.5 31.7C153.5 32.9 150.5 36 149.4 43.1C148.2 36 145.3 32.9 138.3 31.7C145.3 30.5 148.2 27.7 149.4 20.5Z"/>
+
+    <text x="184" y="100" fill="url(#kri-horizontal-text)" font-family="'PingFang SC', 'Microsoft YaHei', sans-serif" font-size="66" font-weight="700" letter-spacing="1">坤肉AI</text>
+</svg>

package/vendor/chatgpt2api/source/config.default.json

@@ -0,0 +1,108 @@
+{
+  "auth-key": "",
+  "refresh_account_interval_minute": 5,
+  "image_retention_days": 15,
+  "image_poll_timeout_secs": 120,
+  "auto_remove_rate_limited_accounts": false,
+  "auto_remove_invalid_accounts": false,
+  "log_levels": [
+    "debug",
+    "error",
+    "info",
+    "warning"
+  ],
+  "proxy": "",
+  "proxy_runtime": {
+    "enabled": false,
+    "egress_mode": "single_proxy",
+    "proxy_url": "",
+    "resource_proxy_url": "",
+    "skip_ssl_verify": false,
+    "reset_session_status_codes": [
+      403
+    ],
+    "clearance": {
+      "enabled": false,
+      "mode": "flaresolverr",
+      "cf_cookies": "",
+      "cf_clearance": "",
+      "user_agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/145.0.0.0 Safari/537.36",
+      "browser": "chrome",
+      "flaresolverr_url": "http://flaresolverr:8191",
+      "timeout_sec": 60,
+      "refresh_interval": 3600,
+      "warm_up_on_start": true
+    }
+  },
+  "base_url": "",
+  "sensitive_words": [],
+  "global_system_prompt": "",
+  "default_upstream_model_name": "gpt-5-5",
+  "default_thinking_effort": "auto",
+  "ai_review": {
+    "enabled": false,
+    "base_url": "",
+    "api_key": "",
+    "model": "",
+    "prompt": ""
+  },
+  "backup": {
+    "enabled": false,
+    "provider": "cloudflare_r2",
+    "account_id": "admin",
+    "access_key_id": "",
+    "secret_access_key": "",
+    "bucket": "",
+    "prefix": "backups",
+    "interval_minutes": 1440,
+    "rotation_keep": 10,
+    "encrypt": false,
+    "passphrase": "",
+    "include": {
+      "config": true,
+      "cpa": true,
+      "sub2api": true,
+      "logs": true,
+      "image_tasks": true,
+      "accounts_snapshot": true,
+      "auth_keys_snapshot": true,
+      "images": false
+    }
+  },
+  "image_account_concurrency": 3,
+  "image_parallel_generation": true,
+  "image_poll_interval_secs": 10,
+  "image_poll_initial_wait_secs": 10,
+  "image_storage": {
+    "enabled": false,
+    "mode": "local",
+    "webdav_url": "",
+    "webdav_username": "",
+    "webdav_password": "",
+    "webdav_root_path": "chatgpt2api/images",
+    "public_base_url": ""
+  },
+  "image_min_free_mb": 500,
+  "chat_completion_cache": {
+    "enabled": true,
+    "ttl_seconds": 60,
+    "max_entries": 256,
+    "dedupe_inflight": true,
+    "stream_cache": true,
+    "normalize_messages": true,
+    "drop_adjacent_duplicates": true,
+    "drop_assistant_history": false
+  },
+  "image_settle_enabled": false,
+  "image_check_before_hit_enabled": false,
+  "image_settle_secs": 2,
+  "auto_relogin_after_refresh": false,
+  "image_timeout_retry_secs": 30,
+  "third_party_apps": {
+    "infinite_canvas": {
+      "enabled": true,
+      "url": "https://canvas.best"
+    }
+  },
+  "image_remove_conversation_after_result": true
+}

package/vendor/chatgpt2api/source/main.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+import uvicorn
+from api import create_app
+
+app = create_app()
+
+if __name__ == "__main__":
+    uvicorn.run(app, access_log=False, log_level="info")

package/vendor/chatgpt2api/source/pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "chatgpt2api"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = [
+    "curl-cffi>=0.15.0",
+    "fastapi>=0.136.0",
+    "pillow>=12.2.0",
+    "pybase64>=1.4.3",
+    "python-multipart>=0.0.26",
+    "tiktoken>=0.12.0",
+    "uvicorn>=0.44.0",
+    "sqlalchemy>=2.0.0",
+    "psycopg2-binary>=2.9.0",
+    "gitpython>=3.1.0",
+]
+
+[dependency-groups]
+dev = [
+    "httpx>=0.28.1",
+]
+
+[[tool.uv.index]]
+url = "https://mirrors.aliyun.com/pypi/simple"
+default = true

package/vendor/chatgpt2api/source/scripts/init_proxy_config.py

@@ -0,0 +1,186 @@
+#!/usr/bin/env python3
+"""Idempotently merge WARP/FlareSolverr proxy_runtime defaults into config.json."""
+
+from __future__ import annotations
+
+import copy
+import json
+import os
+import re
+import sys
+from pathlib import Path
+from typing import Any
+
+DEFAULT_USER_AGENT = (
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+    "AppleWebKit/537.36 (KHTML, like Gecko) "
+    "Chrome/145.0.0.0 Safari/537.36"
+)
+
+DEFAULT_PROXY_RUNTIME: dict[str, Any] = {
+    "enabled": False,
+    "egress_mode": "direct",
+    "proxy_url": "",
+    "resource_proxy_url": "",
+    "skip_ssl_verify": False,
+    "reset_session_status_codes": [403],
+    "clearance": {
+        "enabled": False,
+        "mode": "none",
+        "cf_cookies": "",
+        "cf_clearance": "",
+        "user_agent": DEFAULT_USER_AGENT,
+        "browser": "chrome",
+        "flaresolverr_url": "",
+        "timeout_sec": 60,
+        "refresh_interval": 3600,
+        "warm_up_on_start": False,
+    },
+}
+
+
+def _env_bool(name: str, default: bool) -> bool:
+    value = os.getenv(name)
+    if value is None:
+        return default
+    lowered = value.strip().lower()
+    if lowered in {"1", "true", "yes", "on"}:
+        return True
+    if lowered in {"0", "false", "no", "off"}:
+        return False
+    return default
+
+
+def _env_int(name: str, default: int, minimum: int = 0) -> int:
+    try:
+        return max(minimum, int(os.getenv(name, str(default))))
+    except (TypeError, ValueError, OverflowError):
+        return default
+
+
+def _warp_runtime_defaults() -> dict[str, Any]:
+    clearance_enabled = _env_bool("CHATGPT2API_PROXY_RUNTIME_CLEARANCE_ENABLED", True)
+    clearance_mode = os.getenv("CHATGPT2API_PROXY_RUNTIME_CLEARANCE_MODE", "flaresolverr").strip().lower()
+    if clearance_mode not in {"none", "manual", "flaresolverr"}:
+        clearance_mode = "flaresolverr" if clearance_enabled else "none"
+    if not clearance_enabled:
+        clearance_mode = "none"
+
+    egress_mode = os.getenv("CHATGPT2API_PROXY_RUNTIME_EGRESS_MODE", "single_proxy").strip().lower()
+    if egress_mode not in {"direct", "single_proxy"}:
+        egress_mode = "single_proxy"
+
+    runtime = copy.deepcopy(DEFAULT_PROXY_RUNTIME)
+    runtime.update(
+        {
+            "enabled": _env_bool("CHATGPT2API_PROXY_RUNTIME_ENABLED", True),
+            "egress_mode": egress_mode,
+            "proxy_url": os.getenv("CHATGPT2API_PROXY_RUNTIME_PROXY_URL", "http://privoxy:8118").strip(),
+            "resource_proxy_url": os.getenv("CHATGPT2API_PROXY_RUNTIME_RESOURCE_PROXY_URL", "").strip(),
+            "skip_ssl_verify": _env_bool("CHATGPT2API_PROXY_RUNTIME_SKIP_SSL_VERIFY", False),
+            "reset_session_status_codes": [
+                int(part.strip())
+                for part in os.getenv("CHATGPT2API_PROXY_RUNTIME_RESET_STATUS_CODES", "403").split(",")
+                if part.strip().isdigit() and 100 <= int(part.strip()) <= 599
+            ] or [403],
+        }
+    )
+    runtime["clearance"].update(
+        {
+            "enabled": clearance_enabled,
+            "mode": clearance_mode,
+            "user_agent": os.getenv("CHATGPT2API_PROXY_RUNTIME_USER_AGENT", DEFAULT_USER_AGENT).strip() or DEFAULT_USER_AGENT,
+            "browser": os.getenv("CHATGPT2API_PROXY_RUNTIME_BROWSER", "chrome").strip() or "chrome",
+            "flaresolverr_url": os.getenv("CHATGPT2API_FLARESOLVERR_URL", "http://flaresolverr:8191").strip(),
+            "timeout_sec": _env_int("CHATGPT2API_PROXY_RUNTIME_CLEARANCE_TIMEOUT_SEC", 60, 1),
+            "refresh_interval": _env_int("CHATGPT2API_PROXY_RUNTIME_CLEARANCE_REFRESH_INTERVAL", 3600, 60),
+            "warm_up_on_start": _env_bool("CHATGPT2API_PROXY_RUNTIME_WARM_UP_ON_START", False),
+        }
+    )
+    return runtime
+
+
+def _deep_fill_missing(target: dict[str, Any], defaults: dict[str, Any]) -> bool:
+    changed = False
+    for key, value in defaults.items():
+        if key not in target:
+            target[key] = copy.deepcopy(value)
+            changed = True
+            continue
+        if isinstance(target[key], dict) and isinstance(value, dict):
+            changed = _deep_fill_missing(target[key], value) or changed
+    return changed
+
+
+def _looks_like_repository_default(runtime: Any) -> bool:
+    if not isinstance(runtime, dict):
+        return True
+    candidate = copy.deepcopy(runtime)
+    _deep_fill_missing(candidate, DEFAULT_PROXY_RUNTIME)
+    return candidate == DEFAULT_PROXY_RUNTIME
+
+
+def _mask_url(value: str) -> str:
+    return re.sub(r"(https?://)([^\s/@:]+):([^\s/@]+)@", r"\1[REDACTED]@", value or "", flags=re.I)
+
+
+def main() -> int:
+    config_path = Path(os.getenv("CHATGPT2API_CONFIG_FILE", "/app/config.json"))
+    if not config_path.exists():
+        print(f"Config file not found, creating {config_path}")
+        data: dict[str, Any] = {}
+    else:
+        try:
+            data = json.loads(config_path.read_text(encoding="utf-8"))
+        except json.JSONDecodeError as exc:
+            print(f"Invalid JSON in {config_path}: {exc}", file=sys.stderr)
+            return 1
+        if not isinstance(data, dict):
+            print(f"Config root must be an object: {config_path}", file=sys.stderr)
+            return 1
+
+    desired = _warp_runtime_defaults()
+    existing = data.get("proxy_runtime")
+    changed = False
+
+    if _looks_like_repository_default(existing) or _env_bool("CHATGPT2API_PROXY_RUNTIME_FORCE", False):
+        data["proxy_runtime"] = desired
+        changed = True
+        print("Created proxy_runtime defaults")
+    else:
+        runtime = existing if isinstance(existing, dict) else {}
+        changed = _deep_fill_missing(runtime, DEFAULT_PROXY_RUNTIME)
+        data["proxy_runtime"] = runtime
+        print("Proxy runtime already configured")
+
+    if changed:
+        config_path.parent.mkdir(parents=True, exist_ok=True)
+        payload = json.dumps(data, ensure_ascii=False, indent=2) + "\n"
+        tmp_path = config_path.with_suffix(config_path.suffix + ".tmp")
+        try:
+            tmp_path.write_text(payload, encoding="utf-8")
+            tmp_path.replace(config_path)
+        except OSError as exc:
+            # Docker bind-mounted single files can reject atomic rename with EBUSY.
+            # Fall back to in-place write so the init job works with both file and
+            # directory mounts.
+            if getattr(exc, "errno", None) != 16:
+                raise
+            config_path.write_text(payload, encoding="utf-8")
+            tmp_path.unlink(missing_ok=True)
+
+    runtime = data.get("proxy_runtime") if isinstance(data.get("proxy_runtime"), dict) else {}
+    clearance = runtime.get("clearance") if isinstance(runtime.get("clearance"), dict) else {}
+    print(
+        "Proxy runtime summary: "
+        f"enabled={bool(runtime.get('enabled'))}, "
+        f"egress_mode={runtime.get('egress_mode')}, "
+        f"proxy_url={_mask_url(str(runtime.get('proxy_url') or ''))}, "
+        f"clearance_mode={clearance.get('mode')}, "
+        f"flaresolverr_url={_mask_url(str(clearance.get('flaresolverr_url') or ''))}"
+    )
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

package/vendor/chatgpt2api/source/scripts/migrate_storage.py

@@ -0,0 +1,163 @@
+#!/usr/bin/env python3
+"""
+存储后端数据迁移脚本
+
+用法：
+  python scripts/migrate_storage.py --from json --to postgres
+  python scripts/migrate_storage.py --from postgres --to git
+  python scripts/migrate_storage.py --export accounts.json
+  python scripts/migrate_storage.py --import accounts.json
+"""
+
+import argparse
+import json
+import os
+import sys
+from pathlib import Path
+
+# 添加项目根目录到 Python 路径
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+DATA_DIR = Path(__file__).resolve().parents[1] / "data"
+
+from services.storage.factory import create_storage_backend
+
+
+def export_to_json(output_file: str):
+    """导出当前存储后端的数据到 JSON 文件"""
+    print(f"[migrate] Exporting data to {output_file}")
+    DATA_DIR.mkdir(parents=True, exist_ok=True)
+    storage = create_storage_backend(DATA_DIR)
+    accounts = storage.load_accounts()
+    
+    output_path = Path(output_file)
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+    output_path.write_text(
+        json.dumps(accounts, ensure_ascii=False, indent=2) + "\n",
+        encoding="utf-8",
+    )
+    
+    print(f"[migrate] Exported {len(accounts)} accounts to {output_file}")
+
+
+def import_from_json(input_file: str):
+    """从 JSON 文件导入数据到当前存储后端"""
+    print(f"[migrate] Importing data from {input_file}")
+    DATA_DIR.mkdir(parents=True, exist_ok=True)
+    input_path = Path(input_file)
+    if not input_path.exists():
+        print(f"[migrate] Error: File not found: {input_file}")
+        sys.exit(1)
+    
+    try:
+        accounts = json.loads(input_path.read_text(encoding="utf-8"))
+        if not isinstance(accounts, list):
+            print(f"[migrate] Error: Invalid JSON format, expected array")
+            sys.exit(1)
+    except json.JSONDecodeError as e:
+        print(f"[migrate] Error: Invalid JSON: {e}")
+        sys.exit(1)
+    
+    storage = create_storage_backend(DATA_DIR)
+    storage.save_accounts(accounts)
+    
+    print(f"[migrate] Imported {len(accounts)} accounts")
+
+
+def migrate_data(from_backend: str, to_backend: str):
+    """从一个存储后端迁移到另一个"""
+    print(f"[migrate] Migrating from {from_backend} to {to_backend}")
+    DATA_DIR.mkdir(parents=True, exist_ok=True)
+    # 保存原始环境变量
+    original_backend = os.environ.get("STORAGE_BACKEND")
+    
+    try:
+        # 从源后端读取数据
+        os.environ["STORAGE_BACKEND"] = from_backend
+        from_storage = create_storage_backend(DATA_DIR)
+        accounts = from_storage.load_accounts()
+        print(f"[migrate] Loaded {len(accounts)} accounts from {from_backend}")
+        
+        # 写入目标后端
+        os.environ["STORAGE_BACKEND"] = to_backend
+        to_storage = create_storage_backend(DATA_DIR)
+        to_storage.save_accounts(accounts)
+        print(f"[migrate] Saved {len(accounts)} accounts to {to_backend}")
+        
+        print(f"[migrate] Migration completed successfully!")
+        
+    finally:
+        # 恢复原始环境变量
+        if original_backend:
+            os.environ["STORAGE_BACKEND"] = original_backend
+        elif "STORAGE_BACKEND" in os.environ:
+            del os.environ["STORAGE_BACKEND"]
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="ChatGPT2API 存储后端数据迁移工具",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+示例:
+  # 从 JSON 迁移到 PostgreSQL
+  python scripts/migrate_storage.py --from json --to postgres
+  
+  # 从 PostgreSQL 迁移到 Git
+  python scripts/migrate_storage.py --from postgres --to git
+  
+  # 导出当前数据到 JSON 文件
+  python scripts/migrate_storage.py --export backup.json
+  
+  # 从 JSON 文件导入数据
+  python scripts/migrate_storage.py --import backup.json
+
+环境变量:
+  STORAGE_BACKEND  - 存储后端类型 (json, sqlite, postgres, git)
+  DATABASE_URL     - 数据库连接字符串
+  GIT_REPO_URL     - Git 仓库地址
+  GIT_TOKEN        - Git 访问令牌
+        """
+    )
+    
+    parser.add_argument(
+        "--from",
+        dest="from_backend",
+        choices=["json", "sqlite", "postgres", "git"],
+        help="源存储后端",
+    )
+    parser.add_argument(
+        "--to",
+        dest="to_backend",
+        choices=["json", "sqlite", "postgres", "git"],
+        help="目标存储后端",
+    )
+    parser.add_argument(
+        "--export",
+        dest="export_file",
+        metavar="FILE",
+        help="导出数据到 JSON 文件",
+    )
+    parser.add_argument(
+        "--import",
+        dest="import_file",
+        metavar="FILE",
+        help="从 JSON 文件导入数据",
+    )
+    
+    args = parser.parse_args()
+    
+    # 检查参数
+    if args.from_backend and args.to_backend:
+        migrate_data(args.from_backend, args.to_backend)
+    elif args.export_file:
+        export_to_json(args.export_file)
+    elif args.import_file:
+        import_from_json(args.import_file)
+    else:
+        parser.print_help()
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

package/vendor/chatgpt2api/source/scripts/privoxy-warp.conf

@@ -0,0 +1,13 @@
+listen-address  0.0.0.0:8118
+toggle  1
+enable-remote-toggle  0
+enable-remote-http-toggle  0
+enable-edit-actions  0
+enforce-blocks  0
+buffer-limit  4096
+keep-alive-timeout  300
+socket-timeout  300
+
+# Forward every HTTP/HTTPS request through the WARP SOCKS5 proxy.
+# The trailing dot is required by Privoxy syntax.
+forward-socks5t   /   warp-proxy:1080 .