微信机器人接入Midjourney API接口

获取机器人

# 启动程序
安装依赖
```bash
pip install -r requirements.txt
```


启动程序
```bash
export WECHATY_PUPPET_SERVICE_TOKEN=xxx
export ZHISHUYUN_CHATGPT_35_TOKEN=xxx
export ZHISHUYUN_MIDJOURNEY_TOKEN=xxx
python3 main.py
```


然后日志会打出二维码，可以扫码登录


> tip  
> 可以通过命令行参数（小写）覆盖环境变量的配置，如  
> ```bash
> python3 main.py -zhishuyun_midjourney_token xxx2
> ```


## 使用Docker
创建文件Dockerfile（放在项目目录）
```Dockerfile
FROM python:3.9-buster


COPY requirements.txt requirements.txt
RUN pip install --trusted-host https://repo.huaweicloud.com -i https://repo.huaweicloud.com/repository/pypi/simple -r requirements.txt &&\
    rm -rf root/.cache &&\
    rm -f requirements.txt \
COPY src app/src
COPY main.py app/main.py
WORKDIR app
CMD ["python3", "main.py"]
```


构建镜像
```bash
cd <项目目录>
# we-mid:0.1 为tag，可以自行设置
docker build -t we-mid:0.1 . 
```


启动服务
```bash
docker run --name we-mid \
       -e WECHATY_PUPPET_SERVICE_TOKEN=xxx \
       -e ZHISHUYUN_CHATGPT_35_TOKEN=xxx \
       -e ZHISHUYUN_MIDJOURNEY_TOKEN=xxx \
       --restart unless-stopped -d \
       we-mid:0.1
```


# 依赖服务
## GPT及Midjourney
- GPT使用[ChatGPT 3.5 API](https://data.zhishuyun.com/documents/59fb1199-6694-4afb-a222-3554d7f7d05a)
- Midjourney使用[Midjourney Imagine API](https://data.zhishuyun.com/documents/58ea7cc1-c685-40c3-a619-f29f9ac5d8f4)


## Wechaty
[WeChaty](https://github.com/wechaty/wechaty)是一个聊天机器人框架，支持多种IM系统，如微信、Lark等。目前有一些成熟的付费Puppet Service提供:
- [WorkPro](https://wechaty.js.org/docs/puppet-services/workpro)，完美支持企业微信
- [PadLocal](https://wechaty.js.org/docs/puppet-services/padlocal)，以ipad副设备登录微信，需要一个容器做代理
- [Paimon](https://wechaty.js.org/docs/puppet-services/paimon)，以ipad副设备登录微信，无需额外容器，但功能有一些限制


可以在[wechaty/puppet-padlocal](https://github.com/wechaty/puppet-padlocal)看下功能区别。


# 逻辑简述
- 在`midjourney_bot.py`中实现WechatyBot用于监听WeChaty的各种事件
  - 主要逻辑都写在了`on_message`中
  - 原理是对字符串的处理，解析出用户发送的命令，判断执行哪些逻辑或是回复哪些消息
- `id_generator.py`中简单实现了一个字符串编解码的过程，用于生成任务ID
- `img_compress.py`中简单实现了一个类似微信朋友圈图片压缩的功能
- `translate.py`和`zsy_midjoureny.py`中分别调用相应Api，发送http请求，取回结果


# 其它
WeChaty新版本已经支持了插件式开发（还没有研究），代码可能更简洁清楚

技术架构

技术交流