nonebot-plugin-easy-aidraw 0.1.0__tar.gz

nonebot_plugin_easy_aidraw-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Agnes4m <Z735803792@163.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

nonebot_plugin_easy_aidraw-0.1.0/PKG-INFO

@@ -0,0 +1,193 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-easy-aidraw
+Version: 0.1.0
+Summary: AI绘图插件 for NoneBot2，支持多种绘图API
+Author-email: Agnes4m <Z735803792@163.com>
+License-Expression: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: nonebot2>=2.5.0
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: nonebot-plugin-alconna>=0.62.0
+Dynamic: license-file
+
+<!-- markdownlint-disable MD026 MD031 MD033 MD036 MD041 MD046 MD051 -->
+<div align="center">
+  <img src="https://raw.githubusercontent.com/Agnes4m/nonebot_plugin_l4d2_server/main/image/logo.png" width="180" height="180"  alt="AgnesDigitalLogo">
+  <br>
+  <p><img src="https://s2.loli.net/2022/06/16/xsVUGRrkbn1ljTD.png" width="240" alt="NoneBotPluginText"></p>
+</div>
+
+<div align="center">
+
+# nonebot_plugin_aidraw 0.1.0
+
+_✨Nonebot & AI 绘图 插件 ✨_
+
+<div align = "center">
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw" target="_blank">仓库</a> &nbsp; · &nbsp;
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues" target="_blank">指令 & 反馈</a> &nbsp; · &nbsp;
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues" target="_blank">常见问题</a>
+</div><br>
+
+<img src="https://img.shields.io/badge/python-3.10+-blue?logo=python&logoColor=edb641" alt="python">
+<a href ="LICENSE">
+<img src="https://img.shields.io/github/license/Agnes4m/nonebot_plugin_aidraw" alt="aidrawlogo">
+</a>
+<img src="https://img.shields.io/badge/nonebot-2.1.0+-red.svg" alt="NoneBot">
+<a href="https://pypi.python.org/pypi/nonebot-plugin-aidraw">
+<img src="https://img.shields.io/pypi/v/nonebot-plugin-aidraw?logo=python&logoColor=edb641" alt="python">
+</a>
+</br>
+<a href="https://github.com/astral-sh/ruff">
+<img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/charliermarsh/ruff/main/assets/badge/v2.json" alt="ruff">
+</a>
+<a href="https://github.com/psf/black">
+<img src="https://img.shields.io/badge/code%20style-black-000000.svg?logo=python&logoColor=edb641" alt="black">
+</a>
+
+<img src="https://img.shields.io/badge/alconna-0.50.0+-red.svg" alt="NoneBot">
+
+<a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues">
+        <img alt="GitHub issues" src="https://img.shields.io/github/issues/Agnes4m/nonebot_plugin_aidraw" alt="issues">
+</a>
+
+<a href="https://pypi.python.org/pypi/nonebot-plugin-aidraw">
+    <img src="https://img.shields.io/pypi/dm/nonebot-plugin-aidraw" alt="pypi download">
+</a>
+</br>
+<a href="https://jq.qq.com/?_wv=1027&k=HdjoCcAe">
+        <img src="https://img.shields.io/badge/QQ%E7%BE%A4-399365126-orange?style=flat-square" alt="QQ Chat Group">
+</a>
+</div>
+
+## 快速使用
+
+
+### 安装
+
+以下提到的方法 任选**其一** 即可
+
+<details open>
+<summary>[推荐] 使用 nb-cli 安装</summary>
+在 nonebot2 项目的根目录下打开命令行, 输入以下指令即可安装
+
+```bash
+nb plugin install nonebot-plugin-aidraw
+```
+
+</details>
+
+<details>
+<summary>使用包管理器安装</summary>
+在 nonebot2 项目的插件目录下, 打开命令行, 根据你使用的包管理器, 输入相应的安装命令
+
+<details>
+<summary>pip</summary>
+
+```bash
+pip install nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>pdm</summary>
+
+```bash
+pdm add nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>poetry</summary>
+
+```bash
+poetry add nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>conda</summary>
+
+```bash
+conda install nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>uv</summary>
+
+```bash
+uv install nonebot-plugin-aidraw
+```
+
+</details>
+</details>
+
+### env最简化配置
+
+draw_api_url = "http://localhost:8080"      # API 地址
+draw_api_key = ""                           # API 密钥
+draw_model = "gpt-image-2"                  # 模型名称
+
+## 功能
+
+- `绘图` 调用绘图 API 生成图片
+
+### 支持的绘图后端
+
+- 任意 OpenAI 兼容的 `/v1/images/generations` 接口（如 OpenAI、Gemini、本地自部署等）
+- 返回 `url` 时直接转发图片链接
+- 自定义 API 地址、密钥、模型、尺寸、超时
+
+### NSFW 过滤
+
+- 群聊场景下可选启用 NSFW 关键词过滤（`DRAW_NSFW_ENABLED=true` + `DRAW_NSFW_KEYWORDS=[...]`）
+- 私聊自动跳过检测
+
+### 输出容错
+
+- 一旦拿到图片 URL，即使后续发送环节出现小问题，也不会再向用户推送错误消息，避免误扰
+
+
+## 主要功能
+
+- [x] 调用 OpenAI 兼容绘图接口生成图片
+- [x] 支持自定义 API 地址、密钥、模型
+- [x] 支持 base64 / URL 两种返回格式
+- [x] 群聊 NSFW 关键词过滤（可关闭）
+- [x] 黑白名单访问控制（支持群组/用户 ID）
+
+## env 设置
+
+```bash
+draw_api_url = "http://localhost:8080"      # API 地址
+draw_api_key = ""                           # API 密钥
+draw_model = "flux"                         # 模型名称
+draw_backend = "openai"                     # 后端类型: openai/gemini/sd
+draw_default_size = "1024x1024"            # 图片尺寸
+draw_timeout = 120                          # 超时时间(秒)
+draw_nsfw_enabled = false                  # NSFW 检测（仅群聊）
+draw_nsfw_keywords = []                    # NSFW 关键词
+draw_whitelist_mode = false                # 白名单模式
+draw_whitelist = []                        # 白名单
+draw_blacklist = []                         # 黑名单
+```
+
+> `draw_api_url` 留空时使用后端默认地址；填写时作为后缀拼接到 base URL 后面
+> ID 格式：`group_123456`（群组）或 `123456`（私聊）
+
+## 其他
+
+- 如果您有发现 BUG 或者更好的建议，欢迎提 Issue & Pr
+- 如果本插件对你有帮助，不要忘了点个 Star~
+- 本项目仅供学习使用，请勿用于商业用途
+- [MIT License](https://github.com/Agnes4m/nonebot_plugin_aidraw/blob/main/LICENSE) ©[@Agnes4m](https://github.com/Agnes4m)
+
+## 🌐 感谢
+
+- [nonebot2](https://github.com/nonebot/nonebot2) - 聊天机器人的基础框架
+- [nonebot-plugin-alconna](https://github.com/nonebot-plugin-alconna/nonebot-plugin-alconna) - NoneBot2 的 Alconna 适配

nonebot_plugin_easy_aidraw-0.1.0/README.md

@@ -0,0 +1,177 @@
+<!-- markdownlint-disable MD026 MD031 MD033 MD036 MD041 MD046 MD051 -->
+<div align="center">
+  <img src="https://raw.githubusercontent.com/Agnes4m/nonebot_plugin_l4d2_server/main/image/logo.png" width="180" height="180"  alt="AgnesDigitalLogo">
+  <br>
+  <p><img src="https://s2.loli.net/2022/06/16/xsVUGRrkbn1ljTD.png" width="240" alt="NoneBotPluginText"></p>
+</div>
+
+<div align="center">
+
+# nonebot_plugin_aidraw 0.1.0
+
+_✨Nonebot & AI 绘图 插件 ✨_
+
+<div align = "center">
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw" target="_blank">仓库</a> &nbsp; · &nbsp;
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues" target="_blank">指令 & 反馈</a> &nbsp; · &nbsp;
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues" target="_blank">常见问题</a>
+</div><br>
+
+<img src="https://img.shields.io/badge/python-3.10+-blue?logo=python&logoColor=edb641" alt="python">
+<a href ="LICENSE">
+<img src="https://img.shields.io/github/license/Agnes4m/nonebot_plugin_aidraw" alt="aidrawlogo">
+</a>
+<img src="https://img.shields.io/badge/nonebot-2.1.0+-red.svg" alt="NoneBot">
+<a href="https://pypi.python.org/pypi/nonebot-plugin-aidraw">
+<img src="https://img.shields.io/pypi/v/nonebot-plugin-aidraw?logo=python&logoColor=edb641" alt="python">
+</a>
+</br>
+<a href="https://github.com/astral-sh/ruff">
+<img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/charliermarsh/ruff/main/assets/badge/v2.json" alt="ruff">
+</a>
+<a href="https://github.com/psf/black">
+<img src="https://img.shields.io/badge/code%20style-black-000000.svg?logo=python&logoColor=edb641" alt="black">
+</a>
+
+<img src="https://img.shields.io/badge/alconna-0.50.0+-red.svg" alt="NoneBot">
+
+<a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues">
+        <img alt="GitHub issues" src="https://img.shields.io/github/issues/Agnes4m/nonebot_plugin_aidraw" alt="issues">
+</a>
+
+<a href="https://pypi.python.org/pypi/nonebot-plugin-aidraw">
+    <img src="https://img.shields.io/pypi/dm/nonebot-plugin-aidraw" alt="pypi download">
+</a>
+</br>
+<a href="https://jq.qq.com/?_wv=1027&k=HdjoCcAe">
+        <img src="https://img.shields.io/badge/QQ%E7%BE%A4-399365126-orange?style=flat-square" alt="QQ Chat Group">
+</a>
+</div>
+
+## 快速使用
+
+
+### 安装
+
+以下提到的方法 任选**其一** 即可
+
+<details open>
+<summary>[推荐] 使用 nb-cli 安装</summary>
+在 nonebot2 项目的根目录下打开命令行, 输入以下指令即可安装
+
+```bash
+nb plugin install nonebot-plugin-aidraw
+```
+
+</details>
+
+<details>
+<summary>使用包管理器安装</summary>
+在 nonebot2 项目的插件目录下, 打开命令行, 根据你使用的包管理器, 输入相应的安装命令
+
+<details>
+<summary>pip</summary>
+
+```bash
+pip install nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>pdm</summary>
+
+```bash
+pdm add nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>poetry</summary>
+
+```bash
+poetry add nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>conda</summary>
+
+```bash
+conda install nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>uv</summary>
+
+```bash
+uv install nonebot-plugin-aidraw
+```
+
+</details>
+</details>
+
+### env最简化配置
+
+draw_api_url = "http://localhost:8080"      # API 地址
+draw_api_key = ""                           # API 密钥
+draw_model = "gpt-image-2"                  # 模型名称
+
+## 功能
+
+- `绘图` 调用绘图 API 生成图片
+
+### 支持的绘图后端
+
+- 任意 OpenAI 兼容的 `/v1/images/generations` 接口（如 OpenAI、Gemini、本地自部署等）
+- 返回 `url` 时直接转发图片链接
+- 自定义 API 地址、密钥、模型、尺寸、超时
+
+### NSFW 过滤
+
+- 群聊场景下可选启用 NSFW 关键词过滤（`DRAW_NSFW_ENABLED=true` + `DRAW_NSFW_KEYWORDS=[...]`）
+- 私聊自动跳过检测
+
+### 输出容错
+
+- 一旦拿到图片 URL，即使后续发送环节出现小问题，也不会再向用户推送错误消息，避免误扰
+
+
+## 主要功能
+
+- [x] 调用 OpenAI 兼容绘图接口生成图片
+- [x] 支持自定义 API 地址、密钥、模型
+- [x] 支持 base64 / URL 两种返回格式
+- [x] 群聊 NSFW 关键词过滤（可关闭）
+- [x] 黑白名单访问控制（支持群组/用户 ID）
+
+## env 设置
+
+```bash
+draw_api_url = "http://localhost:8080"      # API 地址
+draw_api_key = ""                           # API 密钥
+draw_model = "flux"                         # 模型名称
+draw_backend = "openai"                     # 后端类型: openai/gemini/sd
+draw_default_size = "1024x1024"            # 图片尺寸
+draw_timeout = 120                          # 超时时间(秒)
+draw_nsfw_enabled = false                  # NSFW 检测（仅群聊）
+draw_nsfw_keywords = []                    # NSFW 关键词
+draw_whitelist_mode = false                # 白名单模式
+draw_whitelist = []                        # 白名单
+draw_blacklist = []                         # 黑名单
+```
+
+> `draw_api_url` 留空时使用后端默认地址；填写时作为后缀拼接到 base URL 后面
+> ID 格式：`group_123456`（群组）或 `123456`（私聊）
+
+## 其他
+
+- 如果您有发现 BUG 或者更好的建议，欢迎提 Issue & Pr
+- 如果本插件对你有帮助，不要忘了点个 Star~
+- 本项目仅供学习使用，请勿用于商业用途
+- [MIT License](https://github.com/Agnes4m/nonebot_plugin_aidraw/blob/main/LICENSE) ©[@Agnes4m](https://github.com/Agnes4m)
+
+## 🌐 感谢
+
+- [nonebot2](https://github.com/nonebot/nonebot2) - 聊天机器人的基础框架
+- [nonebot-plugin-alconna](https://github.com/nonebot-plugin-alconna/nonebot-plugin-alconna) - NoneBot2 的 Alconna 适配

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw/__init__.py

@@ -0,0 +1,24 @@
+"""nonebot_plugin_easy_aidraw - AI绘图插件"""
+
+__version__ = "0.1.0"
+
+from nonebot import require
+from nonebot.plugin import PluginMetadata, inherit_supported_adapters
+
+from .config import EnvConfig
+from .handler import draw_command
+
+require("nonebot_plugin_alconna")
+
+__plugin_meta__ = PluginMetadata(
+    name="AI绘图",
+    description="AI绘图插件，支持调用本地或远程的绘图API生成图片",
+    usage="使用 /绘图 <提示词> 生成图片\n例如: /绘图 一只可爱的小猫",
+    type="application",
+    homepage="https://github.com/Agnes4m/nonebot_plugin_easy_aidraw",
+    config=EnvConfig,
+    supported_adapters=inherit_supported_adapters("nonebot_plugin_alconna"),
+    extra={"version": __version__, "author": "Agnes4m"},
+)
+
+__all__ = ["EnvConfig", "__plugin_meta__", "draw_command"]

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw/api.py

@@ -0,0 +1,145 @@
+"""绘图 API 调用与配置访问"""
+
+from __future__ import annotations
+
+import re
+
+import httpx
+from nonebot import get_driver
+from nonebot.adapters import Event
+from nonebot.log import logger
+
+_config: dict | None = None
+
+BACKEND_BASES = {
+    "openai": "https://api.openai.com",
+    "gemini": "https://generativelanguage.googleapis.com",
+    "sd": "http://localhost:7860",
+}
+
+BACKEND_PATHS = {
+    "openai": "/v1/images/generations",
+    "gemini": "/v1beta/images/generations",
+    "sd": "/sdapi/v1/txt2img",
+}
+
+
+def _get_config() -> dict:
+    """延迟获取配置"""
+    global _config
+    if _config is None:
+        config_dict = dict(get_driver().config)
+        api_url = config_dict.get("draw_api_url", "")
+        backend = config_dict.get("draw_backend", "openai")
+        base = BACKEND_BASES.get(backend, BACKEND_BASES["openai"])
+        path = api_url if api_url else BACKEND_PATHS.get(backend, BACKEND_PATHS["openai"])
+        full_url = base + path if not path.startswith("http") else path
+        _config = {
+            "api_url": full_url,
+            "api_key": config_dict.get("draw_api_key", ""),
+            "model": config_dict.get("draw_model", "flux"),
+            "default_size": config_dict.get("draw_default_size", "1024x1024"),
+            "timeout": config_dict.get("draw_timeout", 120),
+            "nsfw_enabled": config_dict.get("draw_nsfw_enabled", False),
+            "nsfw_keywords": config_dict.get("draw_nsfw_keywords", []),
+        }
+    return _config
+
+
+def is_private_message(event: Event) -> bool:
+    """判断是否为私聊消息"""
+    session_id = event.get_session_id()
+    return not session_id.startswith(("group_", "channel_"))
+
+
+def check_whitelist_blacklist(event: Event) -> tuple[bool, str]:
+    """检查黑白名单，返回 (是否通过, 原因)"""
+    config = _get_config()
+    session_id = event.get_session_id()
+
+    if session_id.startswith("group_"):
+        target_id = "group_" + session_id.split("_")[1]
+    else:
+        target_id = session_id.split("_", 1)[-1]
+
+    whitelist = config.get("draw_whitelist", [])
+    blacklist = config.get("draw_blacklist", [])
+    whitelist_mode = config.get("draw_whitelist_mode", False)
+
+    if whitelist_mode:
+        if target_id not in whitelist:
+            return False, "不在白名单中"
+        return True, ""
+    else:
+        if target_id in blacklist:
+            return False, "在黑名单中"
+        return True, ""
+
+
+def check_nsfw(prompt: str) -> tuple[bool, str | None]:
+    """检查 prompt 是否包含 NSFW 关键词"""
+    config = _get_config()
+
+    if not config.get("nsfw_enabled", False):
+        return False, None
+
+    nsfw_keywords = config.get("nsfw_keywords", [])
+    if not nsfw_keywords:
+        return False, None
+
+    words = re.findall(r"[\w\u4e00-\u9fff]+", prompt.lower())
+
+    nsfw_set = {kw.lower() for kw in nsfw_keywords}
+    for word in words:
+        if word in nsfw_set:
+            return True, word
+
+    return False, None
+
+
+async def generate_image(prompt: str, image_urls: list[str] | None = None) -> str | None:
+    """调用API生成图片，返回图片URL"""
+    config = _get_config()
+
+    if not config["api_key"]:
+        raise ValueError("未配置 DRAW_API_KEY，请检查配置文件")
+
+    headers = {
+        "Authorization": f"Bearer {config['api_key']}",
+        "Content-Type": "application/json",
+    }
+
+    payload = {
+        "model": config["model"],
+        "prompt": prompt,
+        "size": config["default_size"],
+    }
+
+    if image_urls:
+        payload["extra_body"] = {
+            "image": image_urls,
+            "response_format": "url",
+        }
+
+    async with httpx.AsyncClient(timeout=config["timeout"]) as client:
+        logger.info(
+            f"[绘图] 发起请求: url={config['api_url']}, "
+            f"model={config['model']}, prompt_len={len(prompt)}, "
+            f"size={config['default_size']}, image_count={len(image_urls) if image_urls else 0}"
+        )
+        logger.debug(f"[绘图] 请求体: {payload}")
+        logger.debug(f"[绘图] 请求头: {headers}")
+        response = await client.post(config["api_url"], json=payload, headers=headers)
+        logger.info(f"[绘图] 响应状态: {response.status_code}")
+        logger.debug(f"[绘图] 响应体: {response.text[:500]}")
+        response.raise_for_status()
+
+        data = response.json()
+        logger.debug(f"[绘图] API返回: {data}")
+
+        if "data" in data and len(data["data"]) > 0:
+            img_data = data["data"][0]
+            if url := img_data.get("url"):
+                return url
+            raise ValueError("API未返回图片URL")
+        raise ValueError(f"API返回格式异常: {data}")

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw/config.py

@@ -0,0 +1,20 @@
+"""子包配置模块"""
+
+from pydantic import BaseModel
+
+
+class EnvConfig(BaseModel):
+    draw_api_url: str = "http://localhost:8080"
+    draw_api_key: str = ""
+    draw_model: str = "flux"
+    draw_backend: str = "openai"
+    draw_default_size: str = "1024x1024"
+    draw_timeout: int = 120
+    draw_nsfw_enabled: bool = False
+    draw_nsfw_keywords: list[str] = []
+    draw_whitelist_mode: bool = False
+    draw_whitelist: list[str] = []
+    draw_blacklist: list[str] = []
+
+    class Config:
+        populate_by_name = True

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw/handler.py

@@ -0,0 +1,54 @@
+"""命令注册与业务处理"""
+
+from __future__ import annotations
+
+from arclet.alconna import Alconna, CommandMeta
+from nonebot.adapters import Event
+from nonebot.log import logger
+from nonebot_plugin_alconna import Image, UniMessage, UniMsg, on_alconna
+
+from .api import check_nsfw, check_whitelist_blacklist, generate_image, is_private_message
+
+draw_alc = Alconna(
+    "绘图",
+    meta=CommandMeta(description="AI绘图命令", example="/绘图 一只可爱的小猫"),
+)
+
+draw_command = on_alconna(draw_alc, auto_send_output=False, use_origin=True, skip_for_unmatch=False)
+
+
+@draw_command.handle()
+async def handle_draw(event: Event, unimsg: UniMsg):
+    if not (passed := check_whitelist_blacklist(event))[0]:
+        return await UniMessage.text(f"❌ 访问被拒绝：{passed[1]}").finish()
+
+    prompt = unimsg.extract_plain_text().strip()
+    for prefix in ["/绘图", "绘图", "/画", "画"]:
+        if prompt.startswith(prefix):
+            prompt = prompt[len(prefix) :].strip()
+            break
+
+    image_urls = [img.data["url"] for img in unimsg[Image] if img.data.get("url")]
+
+    logger.info(f"[绘图] 用户请求绘图: {prompt}, 附带图片: {len(image_urls)}")
+
+    if not is_private_message(event):
+        is_nsfw, keyword = check_nsfw(prompt)
+        if is_nsfw:
+            logger.warning(f"[绘图] 检测到敏感词: {keyword}")
+            return await UniMessage.text(f"❌ 检测到敏感词「{keyword}」，请修改提示词").finish()
+
+    if not prompt:
+        return await UniMessage.text("❌ 请提供绘图提示词\n例如: /绘图 一只可爱的小猫").finish()
+    await UniMessage.text("🎨 正在生成图片，请稍候...").send()
+
+    image_url = await generate_image(prompt, image_urls or None)
+
+    if not image_url:
+        return await UniMessage.text("❌ 图片生成失败").finish()
+    logger.info(f"[绘图] 图片生成成功: {image_url}")
+    try:
+        await UniMessage.image(url=image_url).send()
+    except Exception as e:
+        logger.error(f"[绘图] 发送失败: {e}")
+        await UniMessage.text(f"❌ 发送失败，可手动访问: {image_url}").finish()

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw.egg-info/PKG-INFO

@@ -0,0 +1,193 @@
+Metadata-Version: 2.4
+Name: nonebot-plugin-easy-aidraw
+Version: 0.1.0
+Summary: AI绘图插件 for NoneBot2，支持多种绘图API
+Author-email: Agnes4m <Z735803792@163.com>
+License-Expression: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: nonebot2>=2.5.0
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: nonebot-plugin-alconna>=0.62.0
+Dynamic: license-file
+
+<!-- markdownlint-disable MD026 MD031 MD033 MD036 MD041 MD046 MD051 -->
+<div align="center">
+  <img src="https://raw.githubusercontent.com/Agnes4m/nonebot_plugin_l4d2_server/main/image/logo.png" width="180" height="180"  alt="AgnesDigitalLogo">
+  <br>
+  <p><img src="https://s2.loli.net/2022/06/16/xsVUGRrkbn1ljTD.png" width="240" alt="NoneBotPluginText"></p>
+</div>
+
+<div align="center">
+
+# nonebot_plugin_aidraw 0.1.0
+
+_✨Nonebot & AI 绘图 插件 ✨_
+
+<div align = "center">
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw" target="_blank">仓库</a> &nbsp; · &nbsp;
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues" target="_blank">指令 & 反馈</a> &nbsp; · &nbsp;
+        <a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues" target="_blank">常见问题</a>
+</div><br>
+
+<img src="https://img.shields.io/badge/python-3.10+-blue?logo=python&logoColor=edb641" alt="python">
+<a href ="LICENSE">
+<img src="https://img.shields.io/github/license/Agnes4m/nonebot_plugin_aidraw" alt="aidrawlogo">
+</a>
+<img src="https://img.shields.io/badge/nonebot-2.1.0+-red.svg" alt="NoneBot">
+<a href="https://pypi.python.org/pypi/nonebot-plugin-aidraw">
+<img src="https://img.shields.io/pypi/v/nonebot-plugin-aidraw?logo=python&logoColor=edb641" alt="python">
+</a>
+</br>
+<a href="https://github.com/astral-sh/ruff">
+<img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/charliermarsh/ruff/main/assets/badge/v2.json" alt="ruff">
+</a>
+<a href="https://github.com/psf/black">
+<img src="https://img.shields.io/badge/code%20style-black-000000.svg?logo=python&logoColor=edb641" alt="black">
+</a>
+
+<img src="https://img.shields.io/badge/alconna-0.50.0+-red.svg" alt="NoneBot">
+
+<a href="https://github.com/Agnes4m/nonebot_plugin_aidraw/issues">
+        <img alt="GitHub issues" src="https://img.shields.io/github/issues/Agnes4m/nonebot_plugin_aidraw" alt="issues">
+</a>
+
+<a href="https://pypi.python.org/pypi/nonebot-plugin-aidraw">
+    <img src="https://img.shields.io/pypi/dm/nonebot-plugin-aidraw" alt="pypi download">
+</a>
+</br>
+<a href="https://jq.qq.com/?_wv=1027&k=HdjoCcAe">
+        <img src="https://img.shields.io/badge/QQ%E7%BE%A4-399365126-orange?style=flat-square" alt="QQ Chat Group">
+</a>
+</div>
+
+## 快速使用
+
+
+### 安装
+
+以下提到的方法 任选**其一** 即可
+
+<details open>
+<summary>[推荐] 使用 nb-cli 安装</summary>
+在 nonebot2 项目的根目录下打开命令行, 输入以下指令即可安装
+
+```bash
+nb plugin install nonebot-plugin-aidraw
+```
+
+</details>
+
+<details>
+<summary>使用包管理器安装</summary>
+在 nonebot2 项目的插件目录下, 打开命令行, 根据你使用的包管理器, 输入相应的安装命令
+
+<details>
+<summary>pip</summary>
+
+```bash
+pip install nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>pdm</summary>
+
+```bash
+pdm add nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>poetry</summary>
+
+```bash
+poetry add nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>conda</summary>
+
+```bash
+conda install nonebot-plugin-aidraw
+```
+
+</details>
+<details>
+<summary>uv</summary>
+
+```bash
+uv install nonebot-plugin-aidraw
+```
+
+</details>
+</details>
+
+### env最简化配置
+
+draw_api_url = "http://localhost:8080"      # API 地址
+draw_api_key = ""                           # API 密钥
+draw_model = "gpt-image-2"                  # 模型名称
+
+## 功能
+
+- `绘图` 调用绘图 API 生成图片
+
+### 支持的绘图后端
+
+- 任意 OpenAI 兼容的 `/v1/images/generations` 接口（如 OpenAI、Gemini、本地自部署等）
+- 返回 `url` 时直接转发图片链接
+- 自定义 API 地址、密钥、模型、尺寸、超时
+
+### NSFW 过滤
+
+- 群聊场景下可选启用 NSFW 关键词过滤（`DRAW_NSFW_ENABLED=true` + `DRAW_NSFW_KEYWORDS=[...]`）
+- 私聊自动跳过检测
+
+### 输出容错
+
+- 一旦拿到图片 URL，即使后续发送环节出现小问题，也不会再向用户推送错误消息，避免误扰
+
+
+## 主要功能
+
+- [x] 调用 OpenAI 兼容绘图接口生成图片
+- [x] 支持自定义 API 地址、密钥、模型
+- [x] 支持 base64 / URL 两种返回格式
+- [x] 群聊 NSFW 关键词过滤（可关闭）
+- [x] 黑白名单访问控制（支持群组/用户 ID）
+
+## env 设置
+
+```bash
+draw_api_url = "http://localhost:8080"      # API 地址
+draw_api_key = ""                           # API 密钥
+draw_model = "flux"                         # 模型名称
+draw_backend = "openai"                     # 后端类型: openai/gemini/sd
+draw_default_size = "1024x1024"            # 图片尺寸
+draw_timeout = 120                          # 超时时间(秒)
+draw_nsfw_enabled = false                  # NSFW 检测（仅群聊）
+draw_nsfw_keywords = []                    # NSFW 关键词
+draw_whitelist_mode = false                # 白名单模式
+draw_whitelist = []                        # 白名单
+draw_blacklist = []                         # 黑名单
+```
+
+> `draw_api_url` 留空时使用后端默认地址；填写时作为后缀拼接到 base URL 后面
+> ID 格式：`group_123456`（群组）或 `123456`（私聊）
+
+## 其他
+
+- 如果您有发现 BUG 或者更好的建议，欢迎提 Issue & Pr
+- 如果本插件对你有帮助，不要忘了点个 Star~
+- 本项目仅供学习使用，请勿用于商业用途
+- [MIT License](https://github.com/Agnes4m/nonebot_plugin_aidraw/blob/main/LICENSE) ©[@Agnes4m](https://github.com/Agnes4m)
+
+## 🌐 感谢
+
+- [nonebot2](https://github.com/nonebot/nonebot2) - 聊天机器人的基础框架
+- [nonebot-plugin-alconna](https://github.com/nonebot-plugin-alconna/nonebot-plugin-alconna) - NoneBot2 的 Alconna 适配

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+LICENSE
+README.md
+pyproject.toml
+nonebot_plugin_easy_aidraw/__init__.py
+nonebot_plugin_easy_aidraw/api.py
+nonebot_plugin_easy_aidraw/config.py
+nonebot_plugin_easy_aidraw/handler.py
+nonebot_plugin_easy_aidraw.egg-info/PKG-INFO
+nonebot_plugin_easy_aidraw.egg-info/SOURCES.txt
+nonebot_plugin_easy_aidraw.egg-info/dependency_links.txt
+nonebot_plugin_easy_aidraw.egg-info/requires.txt
+nonebot_plugin_easy_aidraw.egg-info/top_level.txt

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw.egg-info/requires.txt

@@ -0,0 +1,5 @@
+nonebot2>=2.5.0
+httpx>=0.25.0
+pillow>=10.0.0
+pydantic>=2.0.0
+nonebot-plugin-alconna>=0.62.0

nonebot_plugin_easy_aidraw-0.1.0/nonebot_plugin_easy_aidraw.egg-info/top_level.txt

@@ -0,0 +1 @@
+nonebot_plugin_easy_aidraw

nonebot_plugin_easy_aidraw-0.1.0/pyproject.toml

@@ -0,0 +1,76 @@
+[project]
+name = "nonebot-plugin-easy-aidraw"
+version = "0.1.0"
+description = "AI绘图插件 for NoneBot2，支持多种绘图API"
+authors = [
+    {name = "Agnes4m", email = "Z735803792@163.com"},
+]
+dependencies = [
+    "nonebot2>=2.5.0",
+    "httpx>=0.25.0",
+    "pillow>=10.0.0",
+    "pydantic>=2.0.0",
+    "nonebot-plugin-alconna>=0.62.0",
+]
+requires-python = ">=3.10"
+readme = "README.md"
+license = "MIT"
+
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["nonebot_plugin_easy_aidraw*"]
+exclude = ["img*"]
+
+[tool.black]
+line-length = 120
+target-version = ["py310", "py311", "py312", "py313", "py314"]
+include = '\.pyi?$'
+extend-exclude = ''''''
+
+[tool.isort]
+profile = "black"
+line_length = 120
+length_sort = false
+skip_gitignore = true
+force_sort_within_sections = true
+extra_standard_library = ["typing_extensions"]
+
+[tool.ruff]
+line-length = 120
+target-version = "py310"
+respect-gitignore = true
+
+[tool.ruff.lint]
+select = ["E", "W", "F", "UP", "C", "T", "PYI", "PT", "Q", "I"]
+ignore = ["C901", "T201", "E731", "E402"]
+
+[tool.ruff.lint.isort]
+length-sort = false
+force-sort-within-sections = true
+extra-standard-library = ["typing_extensions"]
+
+[tool.pyright]
+pythonVersion = "3.10"
+pythonPlatform = "All"
+typeCheckingMode = "basic"
+reportShadowedImports = false
+
+[tool.nonebot]
+plugin_dirs = []
+builtin_plugins = []
+
+[tool.nonebot.plugins]
+"@local" = ["nonebot_plugin_easy_aidraw"]
+
+[tool.nonebot.adapters]
+"@local" = []
+
+[dependency-groups]
+dev = [
+    "black>=26.5.0",
+    "ruff>=0.15.13",
+]

nonebot_plugin_easy_aidraw-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+