CYDXDianXian / AI_image_gen

NovelAI绘图HoshinoBot插件版
GNU General Public License v3.0
95 stars 9 forks source link
hoshinobot hoshinobot-v2 novelai

说明

.

.

.

.

.

.

以下为归档前的说明

AI_image_gen

AI绘图HoshinoBot插件版

仓库地址:https://github.com/CYDXDianXian/AI_image_gen

感谢 sans 老师、[Cath]() 老师、 兰鹿 以及各群友上传的代码,这里主要对群友上传的各个版本代码进行了缝合

注意事项

特点

配置方法

  1. ...HoshinoBot\hoshino\modules目录下克隆该仓库:

    git clone https://github.com/CYDXDianXian/AI_image_gen.git
  2. 将本插件目录下的配置文件模板 config.template.json 复制并重命名为 config.json ,并进行如下设置:

    Warning \ 只有config.json中更改配置才会生效,请不要修改default_config.py的默认配置信息!

    更改完配置文件需重启bot才能生效!

    • api中填写IP地址
    • token中填写你的token
    • 【可选】在baidu_appid中填写自己的百度翻译APP ID,不填使用内置百度翻译
    • 【可选】在baidu_key填写自己的百度翻译密钥,不填使用内置百度翻译
    • 【可选】在app_id中填写自己的有道智云应用id,不填使用内置有道翻译
    • 【可选】在app_key中填写自己的有道智云应用秘钥,不填使用内置有道翻译

    百度翻译与有道翻译二选一即可,不用的翻译可以关掉。(建议使用百度翻译,对二次元词汇翻译效果较好,如何获取API请翻阅文档后半部分的API说明

    配置文件config.json中的选项都可以依据个人喜好进行更改,但请不要更改配置文件以外的任何文件,否则容易造成程序运行出错!

    {
       "base": {
           "daily_max": 20,  # 每日上限次数
           "freq_limit": 60,  # 频率限制
           "whitelistmode": False,  # 白名单模式开关
           "blacklistmode": True,  # 黑名单模式开关
           "ban_if_group_num_over": 1000,  # 屏蔽群人数超过1000人的群
           "enable_forward_msg": True,  # 是否开启转发消息模式
           "per_page_num": 28 # 用于查看图片时,每页最多有多少张图
       },
       "default": {
           "withdraw": 0,  # 撤回时间,单位秒。设置为0即为不撤回
           "arrange_tags": True,  # 是否开启tags整理
           "add_db": True,  # 是否开启XP数据录入
           "trans": True,  # 是否开启翻译
           "limit_word": True  # 是否开启违禁词过滤
       },
       "NovelAI": {
           "api": "https://lulu.uedbq.xyz",  # 设置api,默认格式:"https://lulu.uedbq.xyz",
           "token": "",  # 设置你的token,例如:"ADGdsvSFGsaA5S2D",(若你的api无需使用token,留空即可)
           "strength": "0.60" # Denoising strength 与原图的关联程度,越小关联越大
       },
       "baidu": {
           "baidu_trans": True,  # 百度翻译开关
           "baidu_api": "https://fanyi-api.baidu.com/api/trans/vip/translate",  # 百度api地址
           "baidu_appid": "",  # 【可选】自己的百度翻译APP ID,不填使用内置百度翻译
           "baidu_key": ""  # 【可选】自己的百度翻译密钥,不填使用内置百度翻译
       },
       "youdao": {
           "youdao_trans": False,  # 有道翻译开关
           "youdao_api": "https://openapi.youdao.com/api",  # 有道api地址
           "app_id": "",  # 【可选】自己的有道智云应用id,不填使用内置有道翻译
           "app_key": ""  # 【可选】自己的有道智云应用秘钥,不填使用内置有道翻译
       },
       "image4x": {
           "Real-CUGAN": True, # Real-CUGAN超分模型开关,可支持2、3、4倍超分,更锐利的线条,更好的纹理保留,虚化区域保留
           "Real-CUGAN-api": "https://dianxian-real-cugan.hf.space/api/predict", # Real-CUGAN的api地址
           "Real-ESRGAN": False, # 目前存在bug(图片尺寸过大生成的图会很小),故暂时不建议开启
           "Real-ESRGAN-api": "https://akhaliq-real-esrgan.hf.space/api/predict/" # Real-ESRGAN的api地址
       },
       "pic_tools":{
           "img2tag_api": "https://hysts-deepdanbooru.hf.space/api/predict", # DeepDanbooru的api地址
           "img2anime_api": "https://hylee-white-box-cartoonization.hf.space/api/predict/" # White-box-Cartoonization的api地址
       },
       "default_tags": {
           "tags": "miku"  # 如果没有指定tag的话,默认的tag
       },
       "ban_word": {
           "wordlist": [
               "r18",
               "naked",
               "vagina",
               "penis",
               "nsfw",
               "genital",
               "nude",
               "NSFW",
               "R18"
           ]
       },  # 屏蔽词列表
    }
  3. 安装依赖:

    pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
  4. hoshino/config/__bot__.py文件中,MODULES_ON里添加 "AI_image_gen"

  5. 运行Hoshinobot

  6. 更新插件:

    请在你的 hoshino/modules/AI_image_gen 文件夹里,打开powershell输入 git pull ,运行完重启hoshinobot即可

使用方法

注:+ 号不用输入

指令 说明
ai绘图帮助 获取本插件全部功能的使用说明
ai绘图/生成涩图+tag 关键词仅支持英文,用逗号隔开
清晰术/图片超分+图片 图片超分(默认2倍放大3级降噪)
清晰术+2倍/3倍/4倍放大+不/保守/强力降噪 图片放大倍率与降噪倍率选项
二次元化/动漫化+图片 照片二次元化
上传pic/上传图片 务必携带seed/scale/tags等参数
查看配方/查看tag+图片ID 查看已上传图片的配方
快捷绘图+图片ID 使用已上传图片的配方进行快捷绘图
查看个人pic/查看个人图片+页码 查看个人已上传的图片
查看本群pic/查看本群图片+页码 查看本群已上传的图片
查看全部pic/查看全部图片+页码 查看全部群已上传的图片
点赞pic/点赞图片+图片ID 对已上传图片进行点赞
删除pic/删除图片+图片ID 删除对应图片和配方(仅限维护组使用)
本群/个人XP排行 本群/个人的tag使用频率
本群/个人XP缝合 缝合tags进行绘图
图片鉴赏/生成tag+图片 根据上传的图片生成tags
回复消息+以图绘图/上传图片/图片鉴赏/清晰术/二次元化 回复消息使用上述功能
元素法典 xxx xxx可以是多种魔咒,空格分离
元素法典咏唱/吟唱 xxx 发动黑暗法典,多种魔咒用空格分离
以下为维护组使用(空格不能漏)
绘图 状态 [群号] 查看本群或指定群的模块开启状态
绘图 设置 撤回时间 0~999 [群号] 设置本群或指定群撤回时间(单位秒),0为不撤回
绘图 设置 tags整理/数据录入/中英翻译/违禁词过滤 开启/关闭 [群号] 启用或禁用本群或指定群的对应模块
绘图 黑/白名单 新增/添加/移除/删除 群号 修改黑白名单
黑名单列表/白名单列表 查询黑白名单列表
参数使用说明
{} 关键词上加{}代表增加权重,可以加很多个
[] 关键词上加[]代表减少权重,可以加很多个
&ntags=xxx 负面tags输入
&shape=Portrait/Landscape/Square 默认Portrait竖图。Landscape(横图),Square(方图)
&scale=11 默认11,赋予AI自由度的参数,越高表示越遵守tags,一般保持11左右不变
&seed=1111111 随机种子。在其他条件不变的情况下,相同的种子代表生成相同的图

参数用法示例:

image

image

API说明

目前可用的NovelAI-API:路路佬的API

如何使用翻译?(注:百度翻译二次元词汇比有道效果好一点)

使用效果预览

image

image

image

image

image

image

image

image

image

image image

鸣谢

go-cqhttp

HoshinoBot

setu_renew

Hugging Face

DeepDanbooru

Real-ESRGAN

White-box-Cartoonization

友情链接

ai_setu - sans

ai绘图安全版 - 姬野梦美

更新日志

2022-11-21:仓库归档,停止更新。若要使用后续新功能请使用sans佬的仓库ai_setu - sans

2022-11-19:更新api地址,解决图片超分和ai绘图不可用的问题

2022-10-27:再次优化图片超分、图片鉴赏、二次元化相关代码

2022-10-25:新增元素法典功能,优化图片超分、图片鉴赏、二次元化相关代码,解决功能使用不稳定问题

2022-10-23:优化图片处理逻辑,提高bot收发图片的速度

2022-10-22:清晰术功能更新,接入Real_CUGAN图片超分api,更新分群管理配置功能

2022-10-21:初步优化代码结构,重构消息发送模块,解决:以图绘图bug、图片上传重复问题、查看已上传图片时图片发送失败问题

2022-10-20:新增不用申请APIKEY的内置翻译,新增清晰术(图片超分)和图片二次元化

2022-10-20:新增转发消息模式;新增自动撤回消息功能;图片鉴赏直接生成文字版tags,方便复制;修复回复上传、回复以图绘图、回复图片鉴赏的bug

2022-10-16:新增回复消息以图绘图、上传图片、生成tags功能

2022-10-15:新增图片鉴赏功能,将帮助说明转为图片发送

2022-10-14:新增快捷绘图,查看已上传的图片配方,查看个人/本群/全部图片,删除上传的图片,接入百度翻译API

2022-10-12:新增自动将中文tag翻译为英文功能,新增XP缝合,新增上传、查看和点赞本群图片,新增tags整理/数据录入/中英翻译/违禁词过滤 开关控制,屏蔽词列表更新

2022-10-11:新增XP查询,修改API接口格式

2022-10-10:初次提交