feat: picture auto-generate

README.md

@@ -5,10 +5,11 @@
  
 本项目是基于ChatGPT的微信聊天机器人，通过 [OpenAI](https://github.com/openai/openai-quickstart-python) 接口生成对话内容，使用 [itchat](https://github.com/littlecodersh/ItChat) 实现微信消息的接收和自动回复。已实现的特性如下：
 
-- [x] **基础功能：** 接收私聊及群组中的微信消息，使用ChatGPT生成回复内容，完成自动回复
+- [x] **w文本对话：** 接收私聊及群组中的微信消息，使用ChatGPT生成回复内容，完成自动回复
 - [x] **规则定制化：** 支持私聊中按指定规则触发自动回复，支持对群组设置自动回复白名单
 - [x] **多账号：** 支持多微信账号同时运行
-- [ ] **会话上下文：** 支持用户维度的上下文记忆
+-  [x]  **图片生成：** 支持根据描述生成图片，并自动发送至个人聊天或群聊
+
 
 # 更新
 > **2022.12.17：**  原来的方案是从 [ChatGPT页面](https://chat.openai.com/chat) 获取session_token，使用 [revChatGPT](https://github.com/acheong08/ChatGPT) 直接访问web接口，但随着ChatGPT接入Cloudflare人机验证，这一方案难以在服务器顺利运行。 所以目前使用的方案是调用 OpenAI 官方提供的 [API](https://beta.openai.com/docs/api-reference/introduction)，回复质量上基本接近于ChatGPT的内容，劣势是暂不支持有上下文记忆的对话，优势是稳定性和响应速度较好。
@@ -23,6 +24,11 @@
 
 ![group-chat-sample.jpg](docs/images/group-chat-sample.jpg)
 
+### 图片生成
+
+![group-chat-sample.jpg](docs/images/image-create-sample.jpg)
+
+
 # 快速开始
 
 ## 准备
@@ -34,12 +40,12 @@
 
 前往 [OpenAI注册页面](https://beta.openai.com/signup) 创建账号，参考这篇 [博客](https://www.cnblogs.com/damugua/p/16969508.html) 可以通过虚拟手机号来接收验证码。创建完账号则前往 [API管理页面](https://beta.openai.com/account/api-keys) 创建一个 API Key 并保存下来，后面需要在项目中配置这个key。 
 
-> 项目中使用的对话模型是 davinci，计费方式是每1k字 (包含请求和回复) 消耗 $0.02，账号创建有免费的 $18 额度，使用完可以更换邮箱重新注册。
+> 项目中使用的对话模型是 davinci，计费方式是每1k字 (包含请求和回复) 消耗 $0.02，图片生成是每张消耗 $0.016，账号创建有免费的 $18 额度，使用完可以更换邮箱重新注册。
 
 
 ### 3.运行环境
 
-支持运行在 Linux、MacOS、Windows 系统上，需安装有 `Python`(版本在3.7.1 ~ 3.8.16 之间)，推荐使用Linux服务器，可以托管在后台长期运行。
+支持运行在 Linux、MacOS、Windows 系统上，且安装有 `Python`(版本需在 3.7.1~3.8.10 之间)，推荐使用Linux服务器，可托管于后台长期运行。
 
 克隆项目代码：
 
@@ -50,28 +56,38 @@ https://github.com/zhayujie/chatgpt-on-wechat
 安装所需核心依赖：
 
 ```bash
-pip3 install itchat
-pip3 install openai
+pip3 install itchat==1.3.10      	
+pip3 install openai==0.25.0
 ```
+> 注: 图片生成功能依赖openai 0.25.0版本，要求Python版本在3.7.1以上，而itchat目前只能运行于Python3.9以下版本，故需要Python版本在 3.7.1~3.9 之间，推荐使用 3.7.X 版本。
 
 ## 配置
 
-配置文件在根目录的 `config.json` 中，示例文件及各配置项含义如下：
+配置文件的模板在根目录的`config-template.json`中，需复制该模板创建最终生效的 `config.json` 文件：
 
 ```bash
+cp config-template.json config.json
+```
+
+然后填入自定义配置，各配置项含义如下：
+
+```bash
+# config.json文件内容示例
 { 
-  "open_ai_api_key": "${YOUR API KEY}$"                      # 上面创建的 OpenAI API KEY
-  "single_chat_prefix": ["bot", "@bot"],                     # 私聊时文本需要包含该前缀才能触发机器人回复
-  "single_chat_reply_prefix": "[bot] ",                      # 私聊时自动回复的前缀，用于区分真人
-  "group_chat_prefix": ["@bot"],                             # 群聊时包含该前缀则会触发机器人回复
-  "group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"] # 开启自动回复的群名称列表
+  "open_ai_api_key": "${YOUR API KEY}$"                       # 必填，上面创建的 OpenAI API KEY
+  "single_chat_prefix": ["bot", "@bot"],                      # 私聊时文本需要包含该前缀才能触发机器人回复
+  "single_chat_reply_prefix": "[bot] ",                       # 私聊时自动回复的前缀，用于区分真人
+  "group_chat_prefix": ["@bot"],                              # 群聊时包含该前缀则会触发机器人回复
+  "group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"], # 开启自动回复的群名称列表
+  "image_create_prefix": ["画", "看", "找"]                                                       # 开启图片生成的前缀
 }
 ```
 **配置说明：**
 
 + 个人聊天中，会以 "bot" 或 "@bot" 为开头的内容触发机器人，对应配置中的 `single_chat_prefix`；机器人回复的内容会以 "[bot]" 作为前缀， 以区分真人，对应的配置为 `single_chat_reply_prefix`
 + 群组聊天中，群名称需配置在 `group_name_white_list ` 中才能开启群聊自动回复，默认只要被@就会触发机器人自动回复，另外群聊天中只要检测到以 "@bot" 开头的内容，同样会自动回复，这对应配置 `group_chat_prefix`
-+ 关于OpenAI对话接口的参数配置，可以参考 [接口文档](https://beta.openai.com/docs/api-reference/completions) 直接在代码 `bot\openai\open_ai_bot.py` 中进行调整
++ 对于图像生成，在满足个人或群组触发条件外，还需要额外的关键词，对应配置 `image_create_prefix `
++ 关于OpenAI对话及图片接口的参数配置（内容自由度、回复字数限制、图片大小等），可以参考 [对话接口](https://beta.openai.com/docs/api-reference/completions) 和 [图像接口](https://beta.openai.com/docs/api-reference/completions)  文档直接在 [代码](https://github.com/zhayujie/chatgpt-on-wechat/blob/master/bot/openai/open_ai_bot.py) `bot\openai\open_ai_bot.py` 中进行调整。
 
 
 ## 运行