diff --git a/Readme.md b/Readme.md index f87125e..6336ba7 100644 --- a/Readme.md +++ b/Readme.md @@ -1 +1,48 @@ -# SXYWeChatBot \ No newline at end of file +# SXYWeChatBot + +一个接入了ChatGPT和NovelAI的微信聊天机器人,兼容windows、mac、linux平台,代码很短很简单很容易扩展别的功能~ + +## 安装配置方法 + +### 依赖 + +用到了两种编程语言:go和python3。使用go是因为本项目依赖于强力的使用go写成的openwechat实现对微信会话的获取以及发送消息的功能。调用ChatGPT以及Stable Diffusion模型则使用python3。 + +python3需要再安装这些库,使用pip安装就可以: + +``` +torch flask openai diffusers +``` + +``` +pip install torch flask openai diffusers +``` + +当然如果使用cuda加速建议按照pytorch官网提供的方法安装支持cuda加速的torch版本。 + +Apple Silicon的macbook上可以使用mps后端加速,我开发的时候使用的就是M1 Max芯片的Macbook Pro。 + +### 修改配置 + +你需要有一个OpenAI账号,然后将API Key写到config.json的OpenAI-API-Key字段后,然后保存 + +### 然后就可以运行了 + +Mac/Linux用户可以直接运行start.sh: + +``` +./start.sh config.json +``` + +或者分开运行bot和wechat_client: + +``` +go run wechat_client.go +python bot.py +``` + +### 注意 + +第一次运行需要下载Stable Diffusion模型,默认的stabilityai/stable-diffusion-2-1有将近10GB,并且从外网下载,需要有比较快速稳定的网络条件。 + +## 机器人使用方法 \ No newline at end of file diff --git a/bot.py b/bot.py index 7b79a8c..a4c32df 100644 --- a/bot.py +++ b/bot.py @@ -1,4 +1,3 @@ -from distutils.command.config import config import flask import requests import json