【Python】uiautomation+openai构建基于chatGPT的微信聊天机器人
目录
前言
一、uiautomation是什么？
二、openai怎么使用？
三、使用步骤
1.前期准备
2.引入库
3.主程序
4.演示效果
总结
前言
人工ai对话技术最近火热，相信很多人会想到做一款微信聊天机器人，让自己那些对程序不是很了解的亲朋好友使用体验一下。
目前大多数的微信聊天机器人，都是基于itchat或者wxautoapi 来操作微信，原理是监听和操作微信的网页版端口，来达到短信收发的目的。但腾讯官方对于这种办法卡得很严，很多微信账号（包括我的）已经不能使用了，就算使用也会面临封禁的风险。
也有用pyautogui来实现的，不过这种基于图像识别的技术的可移植性不是太好，也很难去实现某些功能。
在本文中，利用uiautomation去操作PC端微信，利用官方的openai库获得AI自动回复，实现微信聊天机器人的功能。
此微信聊天机器人的具有以下功能：
1.可以回复多名好友，能自动检测未读消息并回复。
2.配置上下文联系数量和每天问题字数限制，并且每天凌晨可以刷新字数限制。
3.可以给好友配置VIP权限，具有更多的上下文联系数量和问题限制。
4.可以回复群消息，只有当群友@聊天机器人的时候，才会回复。
一、uiautomation是什么？
uiautomation是python的一个模块，封装了微软UIAutomation API，可以对windows的桌面进行控件的搜索、点击等一些列操作，因为完全是桌面级操作，不涉及API接口问题，所以不容易被针对封禁——至少在微信操作上是这样。
uiautomation的原github地址链接
中文说明链接
二、openai怎么使用？
openai是官方提供的python库，通过API接口就可以去获取ChatGPT的回答，值得注意的是如何去实现上下文功能，下面是一段演示代码：
# Note: you need to be using OpenAI Python v0.27.0 for the code below to work
import openai
openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Who won the world series in 2020?"},
{"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
{"role": "user", "content": "Where was it played?"}
]
)在messages中，{"role": "system", "content": "You are a helpful assistant."}算是表头，系统消息有助于设置助手的行为。在上面的例子中，助手被指示“你是一个有帮助的助手”。通常，对话首先使用系统消息格式化，然后是交替的用户和助理消息。
{"role": "user", "content": "Who won the world series in 2020?"}就是第一次对话，"role": "user"表示是用户发起的，问的问题是"Who won the world series in 2020?"。
而{"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."}就是第一次的回答，"role": "assistant"表示是AI的，回答内容是"The Los Angeles Dodgers won the World Series in 2020."
{"role": "user", "content": "Where was it played?"}就是第二次回答了，这样往复就能实现上下文的联系。
需要注意的是，message的发送也占用token数量，openai的chat-3.5模型收费价格是0.002美元/K token，虽然每个账号有18美元的免费额度，但是不限制上下文数量，估计没多久你的额度就爆了。
三、使用步骤
1.前期准备
首先推荐一个pytharm的插件，可以使用中文变量，像我这种英文不好的，最讨厌的就是去想变量名了，有了这个插件会方便好多，还减少了注释。直接在插件搜索Chinese就行：
其次，需要你把urllib3的版本回退到1.25.11，因为使用openai需要挂代理，但是1.26版以上会报错，而1.25.11版本就没有这个问题了。打开cmd，输入以下命令：
pip uninstall urllib3
pip install urllib3==1.25.112.引入库
import time
from datetime import datetime
import uiautomation as ui # 导入控件控制库
import pyperclip # 导入剪切板相关库
import pyautogui # 导入自动操作库
import openai3.主程序
openai.api_key = "" # 这里填入你自己的OpenAi的API
ui.SetGlobalSearchTimeout(0) # 设置全局搜索超时
info = {}
'''
用户配置,范例：
{'张'：{
'更新时间': '2023-04-01',
'是否VIP': '是”,
'最大Token数': 5000,
'已使用Token数': 0,
'上下文联系数量': 5,
'messages'：[{'role': 'system', 'content': '张的对话'}]
}
}
'''
VipList = ['赵某', '王某', '智障'] # vip名单
# 对用户创建一个专门的配置文件，用于存放权限以及对话上下文
def 用户配置(name, group=False):
global info
# 第一步：检索配置中是否存在该人
if name in info: # 如果在配置列表中
if info[name]['更新时间'] != datetime.now().strftime('%y-%m-%d'): # 如果当前日期和更新时间不同,更新频率为1天
info[name]['更新时间'] = datetime.now().strftime('%y-%m-%d') # 刷新更新时间
info[name]['已使用Token数'] = 0 # 重置已使用Token数
info[name]['messages'] = [{'role': 'system', 'content': name + '的对话'}] # 重置messages
else: # 不在配置列表中
info[name] = {} # 以用户名创建配置列表
info[name]['更新时间'] = datetime.now().strftime('%y-%m-%d')
info[name]['messages'] = [{'role': 'system', 'content': name + '的对话'}]
info[name]['已使用Token数'] = 0
# 第二步：判断一下用户是否为vip
if name in VipList: # 如果用户为vip用户
info[name]['是否VIP'] = '是'
info[name]['最大Token数'] = 5000
info[name]['上下文联系数量'] = 10
else: # 如果用户非vip用户
info[name]['是否VIP'] = '否'
info[name]['最大Token数'] = 1000
info[name]['上下文联系数量'] = 1
# 第三步：判断一下用户是否为群聊用户
if group:
info[name]['是否VIP'] = '否'
info[name]['最大Token数'] = 100000 # 因为是在群聊中所有人一起使用，可以设适当设置大一些，谨慎使用！
info[name]['上下文联系数量'] = 1 # 因为是在群聊中所有人一起使用，所以最好就设置成不联系上下文了
def getOpenAiMsg(name, question):
"""
此函数这是向OpenAi发送对话，并返回AI的回答
Args:
name:用户的名称，用于读写对应的用户配置，str格式
question:对话的具体内容，str格式
Returns:
answer:AI的回答,str格式
"""
global info
if question == "权限": # 如果用户输入“权限”，
answer = name + "\n" \
+ "更新时间: " + info[name]['更新时间'] + "\n" \
+ "是否VIP: " + info[name]['是否VIP'] + "\n" \
+ "上下文联系数量: " + str(info[name]['上下文联系数量']) + "\n" \
+ "最大Token数: " + str(info[name]['最大Token数']) + "\n" \
+ "已使用Token数: " + str(info[name]['已使用Token数'])
return answer # 则返回用户配置信息，不会触发后续的OpenAi自动回复
if info[name]['已使用Token数'] >= info[name]['最大Token数']: # 查询是否超出配额
answer = '很抱歉，你今天使用量超出最大配额，发送“权限”查看您的账号情况'
return answer
else:
if len(info[name]['messages']) >= 2 * info[name]['上下文联系数量'] + 1: # 如果上下文数量超出预设
# 删除第一组对话
info[name]['messages'].pop(1)
info[name]['messages'].pop(1)
if info[name]['上下文联系数量'] <= 1: # 如果上下文数量设置为1
info[name]['messages'] = [{'role': 'user', 'content': question}] # 直接设置一条即可
else:
info[name]['messages'].append({'role': 'user', 'content': question}) # 增添对话信息
try:
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo',
messages=info[name]['messages'],
temperature=0.2
)
answer = response.choices[0].message.content # 从返回值中提取回答
# 把回答添加到用户配置中的上下文,为了防止tokens爆炸，限制了字数
info[name]['messages'].append({'role': 'assistant', 'content': answer[:100]})
info[name]['已使用Token数'] = info[name]['已使用Token数'] + response.usage.total_tokens
return answer
except:
answer = '很抱歉，服务器出错，请重试'
return answer
def 普通用户回复():
"""普通用户的回复"""
wx = ui.WindowControl(Name="微信", searchDepth=1) # 绑定名为微信的主窗口控件
wx.SwitchToThisWindow() # 主窗口置前显示
hw = wx.ListControl(Name="会话") # 绑定对话列表
we = hw.EditControl(searchDepth=3) # 查找新对话
if we.Exists(0):
好友 = we.GetParentControl().GetParentControl()
姓名 = 好友.Name
print("聊天对象:" + 姓名)
we.Click(simulateMove=False)
last_msg = wx.ListControl(Name="消息").GetChildren()[-1].Name # 获取最新消息
print("获得消息:" + last_msg)
用户配置(姓名)
response = getOpenAiMsg(姓名, last_msg) # 将收到的信息转发给OpenAi
print("你的回复:\n" + response)
# print(info[姓名]) # 调试用，查看下用户配置
print(
'已消耗标记：' + str(info[姓名]['已使用Token数']) + "\n--------------------------------------------------\n")
# 输入回复信息
pyperclip.copy(response) # 复制信息
pyautogui.hotkey('ctrl', 'v') # 粘贴
pyautogui.press('enter') # 回车
# 隐藏对话
好友.RightClick()
右键菜单 = ui.MenuControl(ClassName="CMenuWnd")
if 右键菜单.Exists(3, 0.1):
右键菜单.TextControl(Name="不显示聊天").Click()
def 群聊用户回复():
"""群聊用户的回复，请将群消息接收设置为消息免打扰"""
wx = ui.WindowControl(Name="微信", searchDepth=1) # 绑定名为微信的主窗口控件
wx.SwitchToThisWindow() # 主窗口置前显示
hw = wx.ListControl(Name="会话") # 绑定名为对话列表
we = hw.EditControl(SubName="有人@我", searchDepth=5) # 查找新对话
if we.Exists(0):
群对象 = we.GetParentControl().GetParentControl().GetParentControl().GetParentControl()
群名 = 群对象.Name
print("聊天群对象:" + 群名)
we.Click(simulateMove=False)
last_msg = wx.ListControl(Name="消息").GetChildren()[-1].Name # 获取最新消息
last_msg = last_msg[7:len(last_msg)] # 将@信息删除，注意这里要和你自身的微信名相匹配,尤其注意微信被@后会多一个‘\u2005’的字符
# print(wx.ListControl(Name="消息").GetChildren()[-1]) # 感兴趣的可以用这行命令去看下原始消息文本
# print(last_msg == '权限') # 建议先在群里发送”权限“两个字，用这行命令比对一下
print("获得群消息:" + last_msg)
用户配置(群名, True)
response = getOpenAiMsg(群名, last_msg) # 将收到的信息转发给OpenAi
print("你的回复:\n" + response)
print(
'已消耗标记：' + str(info[群名]['已使用Token数']) + "\n--------------------------------------------------\n")
# 输入回复信息
pyperclip.copy(response) # 复制信息
pyautogui.hotkey('ctrl', 'v') # 粘贴
pyautogui.press('enter') # 回车
# 隐藏对话
群对象.RightClick() # 隐藏权限
右键菜单 = ui.MenuControl(ClassName="CMenuWnd")
if 右键菜单.Exists(3, 0.1):
右键菜单.TextControl(Name="不显示聊天").Click()
print("======================================\n开始\n======================================\n")
while True:
普通用户回复()
群聊用户回复() # 记得把群消息接收设置为消息免打扰
time.sleep(0.2)
4.演示效果
好友对话，发送”权限“时不会触发chatgpt，而是查看用户对应权限
好友对话，可以看到能正确触发上下文联系
群聊，从成本考虑，群聊一般不设置上下文联系。
总结
本文仅仅简单介绍了uiautomation和openai的一些相关知识，以及相应的实现代码，后续考虑把这个程序直接部署在服务器上长期运行，可惜在阿里云、腾讯云等国内厂家产品中，可部署的境外云服务器并不便宜，境内服务器又需要挂代理，实际体验下来还是挺受网络波动影响，快的时候可以秒回，慢的时候要等十几秒钟。
最后，编程不易，文章也写的幸苦，为了踩坑颇花了一些钱，这里求个捐助，随便大家给多少~