101 |
102 |
103 |
104 |
105 |
179 |
180 |
--------------------------------------------------------------------------------
/extensions/mai_wechat/__init__.py:
--------------------------------------------------------------------------------
1 | from flask import Flask
2 | from flask_socketio import SocketIO
3 | from meowServer import MeowAI as MA
4 | from .server import main
5 |
6 | def init_app(meowAPP:Flask,meowSIO:SocketIO,meowAI:MA):
7 | main(meowAPP,meowSIO,meowAI)
--------------------------------------------------------------------------------
/extensions/mai_wechat/requirements.txt:
--------------------------------------------------------------------------------
1 | uiautomation
2 | Pillow
3 | pywin32
4 | psutil
5 | pyperclip
--------------------------------------------------------------------------------
/extensions/mai_wechat/server.py:
--------------------------------------------------------------------------------
1 | from .wxauto import WeChat,elements
2 | from flask import Flask,Blueprint,render_template,request
3 | from flask_socketio import SocketIO
4 | from threading import Thread, Event
5 | import json,time
6 | from meowServer import MeowAI as MA
7 |
8 | class WeChatServer():
9 | def __init__(self,meowAPP:Flask,meowSIO:SocketIO, meowAI: MA):
10 | self.meowAI=meowAI
11 | self.meowAPP=meowAPP
12 | self.meowSIO=meowSIO
13 | self.stop_event = Event()
14 | self.app = Blueprint('wechat', __name__,static_folder='static',template_folder='templates',url_prefix='/extension/wechat')
15 | self.wx=None
16 | self.prev_state=None
17 | self.chats=[]
18 | @self.app.route('/')
19 | def index():
20 | return render_template('mai_wechat/index.html')
21 | @self.app.route('/start',methods=['POST'])
22 | def start():
23 | try:
24 | self.wx = WeChat()
25 | self.stop_event.clear()
26 | data = request.get_json()
27 | thread = Thread(target=self.run, args=(data['name'],data['chatHistory']))
28 | thread.start()
29 | return json.dumps({'code': 0, 'msg': f'启动成功，获取到已登录窗口：{self.wx.nickname}'})
30 | except:
31 | return json.dumps({'code': 1, 'msg': '启动失败，请检查是否已经打开并登录微信'})
32 | @self.app.route('/stop',methods=['POST','GET'])
33 | def stop():
34 | self.stop_event.set()
35 | self.prev_state=None
36 | self.wx=None
37 | return json.dumps({'code': 0, 'msg': '微信自动对话停止成功'})
38 | @self.app.route('/status',methods=['POST','GET'])
39 | def status():
40 | if self.wx:
41 | return json.dumps({'code': 1, 'msg': '已开始'})
42 | else:
43 | return json.dumps({'code': 0, 'msg': '未开始'})
44 |
45 | self.meowAPP.register_blueprint(self.app)
46 | self.index=index
47 |
48 |
49 | def reply(self,msg:str,chat:elements.ChatWnd)->str:
50 | '''
51 | self.chats=[
52 | {
53 | "role": "User",
54 | "content": "What is the meaning of life?"
55 | },
56 | {
57 | "role": "Assistant",
58 | "content": "The meaning of life is to live a happy and fulfilling life."
59 | },
60 | {
61 | "role": "User",
62 | "content": "How do cats call?"
63 | },
64 | ]
65 | '''
66 | self.chats.append({"role": "User", "content": self.meowAI.purr(msg)})
67 | if self.prev_state:
68 | messages=self.chats[len(self.chats)-2:]
69 | else:
70 | messages=self.chats
71 | print(messages)
72 | reply,self.prev_state=self.meowAI.chat(messages,self.prev_state,lambda x:print(x[0],end=''))
73 | reply=reply.replace('\n\n','\n').replace('。','')
74 | self.chats.append({"role": "Assistant", "content": reply})
75 | return reply
76 |
77 | def run(self,name:str,chat_history:bool=False):
78 | try:
79 | wx=self.wx
80 | wx.ChatWith(who=name)
81 | if chat_history:
82 | wx.LoadMoreMessage()
83 | for chat in wx.GetAllMessage():
84 | if chat[0]=='SYS':pass
85 | elif chat[0]=='Self':
86 | self.chats.append({"role": "Assistant", "content": chat[1]})
87 | else:
88 | self.chats.append({"role": "User", "content": chat[1]})
89 | print(self.chats)
90 | wx.AddListenChat(who=name)
91 | while not self.stop_event.is_set():
92 | chats = wx.GetListenMessage()
93 | for chat in filter(lambda x:x.who==name,chats):
94 | msgs=chats.get(chat)
95 | for msg in filter(lambda x:x.type=='friend',msgs):
96 | reply=self.reply(msg.content,chat)
97 | chat.SendMsg(reply)
98 | time.sleep(0.1)
99 | except Exception as e:
100 | self.meowSIO.emit('emit',{'code':1,'msg':'微信监听失败:'+str(e)})
101 | self.stop_event.set()
102 | self.prev_state=None
103 | self.meowSIO.emit('wechat_stop','微信监听失败:'+str(e))
104 | print('错误：',e)
105 |
106 |
107 | def main(meowAPP:Flask,meowSIO:SocketIO,meowAI:MA):
108 | WeChatServer(meowAPP,meowSIO,meowAI)
--------------------------------------------------------------------------------
/extensions/mai_wechat/static/info.json:
--------------------------------------------------------------------------------
1 | {
2 | "name": "wechat",
3 | "version": "1.0.0",
4 | "description": "微信自动回复",
5 | "url": "/extension/wechat",
6 | "logo": "logo.png"
7 | }
--------------------------------------------------------------------------------
/extensions/mai_wechat/static/logo.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/bilibini/Meow-AI/468d8c668118277491859b082085343feb673866/extensions/mai_wechat/static/logo.png
--------------------------------------------------------------------------------
/extensions/mai_wechat/templates/mai_wechat/index.html:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
72 |
微信自动回复
73 |
74 |
75 |
76 |
77 |
78 |
79 |
80 |
81 |
温馨提示：
82 |
在“启动”前，请先登录电脑微信
83 |
84 |