把聊天机器人接入微信

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

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

源码我已经托管到github:wechat_bot

机器人的api接口

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

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

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

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

创建bot_api.py:

#!/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:

#!/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

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




Fork me on GitHub