ChatGPT-DreamStudio WeChat Robot

🎨基于GO语言实现的微信聊天和图片生成机器人🎨

个人微信接入ChatGPT，实现和GPT机器人互动聊天，同时支持基于文本生成图像。支持私聊回复和群聊艾特回复。

实现功能

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 DREAMSTUDIO_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": "我要问下一个问题了",
  "dreamstudio_api_key": "你的dreamstudio账号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": "清空会话"  # 会话清空口令，默认`下个问题`
  "dreamstudio_api_key":"你的dreamstudio账号api_key",     #dreamstudio账号的api_key
  "engine_id":"stable-diffusion-v1-5",     			  #dreamstudio模型的名称
  "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**开发。

oubeichen/chatgpt-dreamstudio_wechat_robot