今晚看了个电影，回得有点迟。上周末写了一篇构建自己的智能聊天机器人。在微博中被一位技术大V转载，而后博客访问量陡增，之后又被微信一个机器学习的公众号转载，大家似乎很感兴趣，为了对得起增长的访问流量，趁热再写些东西

今天分享如何把聊天机器人接入微信，代替我们和好友聊天

源码我已经托管到github：wechat_bot

机器人的api接口

首先我们决定把聊天机器人封装为api服务，这样它的使用场合就不受限制了，能服务于任何http client

这样做的另一个原因是，我们的微信接入脚本是python2的，而中文聊天机器人脚本基于python3，我不想处理pyhton2/3的编码，所以把机器人封装为服务

阅读下文前，假设你读过我的上篇文章构建自己的智能聊天机器人

我们直接开始构建聊天服务.

创建bot_api.py:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

#!/usr/bin/env python
# coding: utf-8

from chatterbot import ChatBot
from chatterbot.trainers import ChatterBotCorpusTrainer
import hug


deepThought = ChatBot("deepThought")
deepThought.set_trainer(ChatterBotCorpusTrainer)
# 使用中文语料库训练它
deepThought.train("chatterbot.corpus.chinese")  # 语料库


@hug.get()
def get_response(user_input):
    response = deepThought.get_response(user_input).text
    return {"response":response}

源码十分简单，不必多做解释

在此我选择hug作为我的api框架，原因是它的效率很高（比flask还高不少）。

安装依赖：pip3 install hug ，chatterbot的安装方法参考上篇文章

跑起服务：hug -f bot_api.py

测试服务

在浏览器中测试

接入微信

首先感谢@liuwons的wxBot项目，它让我们能用代码与微信交互，这样一来使聊天过程（input/output）可编程了

于是我们可以让聊天机器人接管我们的聊天。

操作过程

首先下载wxBot脚本到本地：wget https://raw.githubusercontent.com/liuwons/wxBot/master/wxbot.py

接着创建wechat_bot.py:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

#!/usr/bin/env python
# coding: utf-8

from wxbot import WXBot
import requests
bot_api="http://127.0.0.1:8000/get_response"

class MyWXBot(WXBot):
    def handle_msg_all(self, msg):
        if msg['msg_type_id'] == 4 and msg['content']['type'] == 0:
            user_input = msg["content"]["data"]
            payload={"user_input":user_input}
            response = requests.get(bot_api,params=payload).json()["response"]
            #print(type(response)) # unicode
            self.send_msg_by_uid(response, msg['user']['id'])

def main():
    bot = MyWXBot()
    bot.DEBUG = True
    bot.conf['qr'] = 'png'
    bot.run()

if __name__ == '__main__':
    main()

安装依赖：pip install requests pyqrcode pypng Pillow

开始运行(使用python2)：python wechat_bot.py

之后扫码登录即可

开始聊天

这是程序收到的消息

这是被机器人接管的聊天界面

感谢小伙伴半夜出出镜助攻，哈哈码农都是夜猫子

have a good time

啊 写完都半夜了，赶紧睡觉