ChatGPT-DreamStudio WeChat Robot
🎨基于GO语言实现的微信聊天和图片生成机器人🎨
GitHub源代码地址
实现功能
- GPT机器人模型热度可配置
- 提问增加上下文&指令清空上下文
- DreamStudio图像生成模型参数可配置
- 可设定图像生成触发指令
- 机器人私聊回复&机器人群聊@回复
- 好友添加自动通过可配置
实现机制
-
利用微信A作为机器人扫码登录程序模拟的微信电脑端，程序后端调用API接口进行文本回复和图片生成。其他微信账号与微信A聊天实现微信个人机器人功能。基于openwechat开源仓库实现
-
基于openai官网提供的GPT API，实现文本交互功能，每个新账号前三个月有18美元免费额度。
-
基于stability官网提供的DreamStudio API,实现图像生成功能，每个账号注册送500张图像生成免费额度，玩玩基本够用了，不够的话10$可购买5000张。
GPT的官方文档和详细参数示例 。
DreamStudio的官方文档和详细参数示例 。
使用前提
- 有openai账号，并且创建好api_key，注册事项可以参考此文章 。
- 有dreamstudio.ai账号，并且创建好api_key
- 微信必须实名认证。最好用小号
注意事项
- 项目仅供娱乐，滥用可能有微信封禁的风险，请勿用于商业用途。
- 请注意收发敏感信息，本项目不做信息过滤。
- dreamstudio图像生成 仅对英文的支持比较好
结果展示
docker运行
使用docker快速运行本项目。
1.基于配置文件挂载运行(推荐)
# 1. 创建目录
$ mkdir -p /data/openai
$ cd /data/openai
# 2. 创建配置文件
$ touch config.json
# 3. 编辑配置文件 ... 配置内容粘贴下文 【配置说明】并按需修改
$ vim config.json
# 4. 拉取镜像
$ docker run -dti --name wechatbot -v /data/openai/config.json:/app/config.json yinqishuo/wechatbot:latest
# 5. 进入容器内部，打开日志文件扫码登陆
$ docker exec -it wechatbot bash
$ tail -f -n 50 /app/run.log # 操作出错后删除容器的操作
$ docker stop wechatbot
$ docker remove wechatbot# 退出容器
$ exit
其中配置文件参考下边的配置文件说明。
2.基于环境变量运行
# 运行项目，环境变量参考下方配置说明
$ docker run -itd --name wechatbot --restart=always \-e GPTAPIKEY=换成你的GPT key \-e AUTO_PASS=false \-e SESSION_TIMEOUT=60s \-e MODEL=text-davinci-003 \-e MAX_TOKENS=512 \-e TEMPREATURE=0.9 \-e REPLY_PREFIX=我是来自机器人回复: \-e SESSION_CLEAR_TOKEN=下个问题 \-e DREAMSTDIO_APIKEY=换成你的dreamstudio key \-e ENGINE_ID=stable-diffusion-v1-5 \-e PICTURE_WIDTH=512 \-e PICTURE_HEIGHT=512 \-e STEPS=30 \-e CFG_SCALE= 7 \-e PICTURE_TOKEN=生成图片 \yinqishuo/wechatbot:latest#进入容器内部，打开日志文件扫码登陆
$ docker exec -it wechatbot bash
$ tail -f -n 50 /app/run.log # 退出容器
$ exit
运行命令中映射的配置文件参考下边的配置文件说明。
3.配置说明
模板：
{"gpt_api_key": "你的gpt api key","auto_pass": true,"session_timeout": 60,"max_tokens": 1024,"model": "text-davinci-003","temperature": 1,"reply_prefix": "来自机器人回复：","session_clear_token": "我要问下一个问题了","dreamstdio_api_key":"你的dreamstdio账号api_key","engine_id":"stable-diffusion-v1-5","picture_width":512,"picture_height":512,"steps":30,"cfg_scale":7,"picture_token":"生成图片"
}
参数说明：
"gpt_api_key": # openai账号里设置的api_key
"auto_pass":# 是否自动通过好友添加"session_timeout": 60, # 会话超时时间，默认60秒，单位秒，在会话时间内所有发送给机器人的信息会作为上下文"max_tokens": 1024, # GPT响应字符数，最大2048，默认值512。会影响接口响应速度，字符越大响应越慢"model": "text-davinci-003", # GPT选用模型，默认text-davinci-003，具体选项参考官网训练场"temperature": 1, # GPT热度，0到1，默认0.9，数字越大创造力越强，但更偏离训练事实，越低越接近训练事实"reply_prefix": "来自机器人回复：", # 私聊回复前缀"session_clear_token": "清空会话" # 会话清空口令，默认`下个问题`"dreamstdio_api_key":"你的dreamstdio账号api_key", #dreamstdio账号的api_key"engine_id":"stable-diffusion-v1-5", #dreamstdio模型的名称"picture_width":512, #生成图片的宽度，长度，默认512*512"picture_height":512, #要求为64的倍数，且>=128,尺寸越大消耗的credits越多"steps":30, #代表模型的渲染步数，越高图片越精细，所需的渲染时间也越长，默认为30，数值越大消耗的credits越多；"cfg_scale":7, #表示生成图像与文本提示的相似度，越高越像"picture_token":"生成图片" #生成图像的触发口令
源码运行
适合了解go语言编程并想进行源码修改的同学，
# 获取项目
$ git clone https://github.com/yinqishuo/chatgpt-dreamstudio_wechat_robot# 进入项目目录
$ cd chatgpt_wechat_robot# 复制配置文件
$ cp config.dev.json config.json# 添加依赖
$ go mod tidy# 启动项目
$ go run main.go# 若想编译为可执行文件
$ go run main.go# 若想打包成docker镜像，需安装Docker ，建议在linux环境下打包，镜像名称为wechatbot
$ make docker# 执行镜像，步骤如上
常见问题
如无法登录
login error: write storage.json: bad file descriptor
删除掉storage.json文件重新登录。
如无法登录
login error: wechat network error: Get "https://wx.qq.com/cgi-bin/mmwebwx-bin/webwxnewloginpage": 301 response missing Location header
一般是微信登录权限问题，先确保PC端能否正常登录。
其他无法登录问题
尝试删除掉storage.json文件，结束进程(linux一般是kill -9 进程id)之后重启程序，重新扫码登录。
如果为docket部署，Supervisord进程管理工具会自动重启程序。
机器人一直答非所问
可能因为上下文累积过多。切换不同问题时，发送指令：启动时配置的session_clear_token字段。会清空上下文
https://link.zhihu.com/?target=https%3A//github.com/Maks-s/sd-akashic
图像生成技巧
想让 AI 图像生成器创作出精准高质的图像，填写准确合适的文本提示词十分重要。
- 不要只输入一个简单的词语（raw prompt），如 Panda（熊猫）、A warrior（战士）等，这样生成的图像会缺少美感和艺术性。
- 使用风格提示词能让图像更具艺术性。在提示词中加入艺术风格的关键词，如 Realistic（写实）、Oil painting（油画）、Pencil drawing（铅笔画）、Concept art（概念艺术）等；此外写实风格的提示词有多重表达形式，如[ a photo of + raw prompt ]、[ a photograph of + raw prompt ]、[ raw prompt,hyperrealistic ]、[ raw prompt,realistic ]。
- 使用艺术家名称让风格更具像或保持风格一致。比如想表现抽象艺术，可以使用[made by Pablo Picassoa]或者 [ raw prompt,Picassoa]。还可以同时输入多名艺术家，效果会更加有趣。
- 最终修饰词。在文本末尾加上的一个修饰词，使图像更符合你想要的效果。比如想要逼真的灯光，可以加上“Unreal Engine”，展现精密细节加上“4K”或“8K”，想要更有艺术性可以加上“trending on artstation”等。
① Stable Diffusion Artist Studies
网址： https://proximacentaurib.notion.site/e2537cbf42c34b7e9a9a4126f81dfd0d
一个由国外网友收集建立的艺术家风格概览表，找到你喜欢的风格后在自己的提示词中加上对应艺术家的名字，就能生成类似风格的图片。
② Stable Diffusion prompting cheatsheet
网址： https://moritz.pm/posts/parameters
一个简短的提示词列表，里面列举了如果你想要实现 3D、精致细节、光照、大环境等效果，应该使用哪些关键词。
③ Stable Diffusion Akashic Records
网址： https://github.com/Maks-s/sd-akashic
一个专业的研究资料库，收集了关于模型原理、艺术风格、提示词、使用技巧和其他有用的工具，适合想深入了解文本-图像扩散模型的人阅读。
本段内容复制于AI绘画神器：DreamStudio - 知乎 (zhihu.com)
友情提示
本项目是 fork 他人的项目来进行学习和使用，请勿商用，可以下载下来做自定义的功能。
项目基于eatmoreapple/openwechat 、ZYallers/chatgpt_wechat_robot 、qingconglaixueit/wechatbot开发。
