今晚看了个电影，回得有点迟。上周末写了一篇构建自己的智能聊天机器人。在微博中被一位技术大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
啊 写完都半夜了，赶紧睡觉
