59 files changed, 76 insertions, 11 deletions
diff --git a/ATRI/data/emoji/JZ.png b/ATRI/data/emoji/JZ.png
new file mode 100644
index 0000000..a287140
--- /dev/null
+++ b/ATRI/data/emoji/JZ.png
diff --git a/ATRI/data/emoji/JZ1.png b/ATRI/data/emoji/JZ1.png
new file mode 100644
index 0000000..5e508e6
--- /dev/null
+++ b/ATRI/data/emoji/JZ1.png
diff --git a/ATRI/data/emoji/SUKI2.png b/ATRI/data/emoji/SUKI2.png
new file mode 100644
index 0000000..b1c0849
--- /dev/null
+++ b/ATRI/data/emoji/SUKI2.png
diff --git a/ATRI/data/emoji/WQ1.jpg b/ATRI/data/emoji/WQ1.jpg
index 79be736..c557d29 100644
--- a/ATRI/data/emoji/WQ1.jpg
+++ b/ATRI/data/emoji/WQ1.jpg
diff --git a/ATRI/data/emoji/WQ2.jpg b/ATRI/data/emoji/WQ2.jpg
index c557d29..4d812f6 100644
--- a/ATRI/data/emoji/WQ2.jpg
+++ b/ATRI/data/emoji/WQ2.jpg
diff --git a/ATRI/data/img/kyaru/1.png b/ATRI/data/img/kyaru/1.png
new file mode 100644
index 0000000..abe8a85
--- /dev/null
+++ b/ATRI/data/img/kyaru/1.png
diff --git a/ATRI/data/img/kyaru/10.png b/ATRI/data/img/kyaru/10.png
new file mode 100644
index 0000000..6de1c7d
--- /dev/null
+++ b/ATRI/data/img/kyaru/10.png
diff --git a/ATRI/data/img/kyaru/11.png b/ATRI/data/img/kyaru/11.png
new file mode 100644
index 0000000..da6141f
--- /dev/null
+++ b/ATRI/data/img/kyaru/11.png
diff --git a/ATRI/data/img/kyaru/12.png b/ATRI/data/img/kyaru/12.png
new file mode 100644
index 0000000..de727a9
--- /dev/null
+++ b/ATRI/data/img/kyaru/12.png
diff --git a/ATRI/data/img/kyaru/13.png b/ATRI/data/img/kyaru/13.png
new file mode 100644
index 0000000..f20fd60
--- /dev/null
+++ b/ATRI/data/img/kyaru/13.png
diff --git a/ATRI/data/img/kyaru/14.png b/ATRI/data/img/kyaru/14.png
new file mode 100644
index 0000000..2fcc384
--- /dev/null
+++ b/ATRI/data/img/kyaru/14.png
diff --git a/ATRI/data/img/kyaru/15.png b/ATRI/data/img/kyaru/15.png
new file mode 100644
index 0000000..531c994
--- /dev/null
+++ b/ATRI/data/img/kyaru/15.png
diff --git a/ATRI/data/img/kyaru/16.png b/ATRI/data/img/kyaru/16.png
new file mode 100644
index 0000000..8df68bd
--- /dev/null
+++ b/ATRI/data/img/kyaru/16.png
diff --git a/ATRI/data/img/kyaru/17.png b/ATRI/data/img/kyaru/17.png
new file mode 100644
index 0000000..465c661
--- /dev/null
+++ b/ATRI/data/img/kyaru/17.png
diff --git a/ATRI/data/img/kyaru/18.png b/ATRI/data/img/kyaru/18.png
new file mode 100644
index 0000000..5484601
--- /dev/null
+++ b/ATRI/data/img/kyaru/18.png
diff --git a/ATRI/data/img/kyaru/19.png b/ATRI/data/img/kyaru/19.png
new file mode 100644
index 0000000..43e3fc7
--- /dev/null
+++ b/ATRI/data/img/kyaru/19.png
diff --git a/ATRI/data/img/kyaru/2.png b/ATRI/data/img/kyaru/2.png
new file mode 100644
index 0000000..117ed3b
--- /dev/null
+++ b/ATRI/data/img/kyaru/2.png
diff --git a/ATRI/data/img/kyaru/20.png b/ATRI/data/img/kyaru/20.png
new file mode 100644
index 0000000..0485974
--- /dev/null
+++ b/ATRI/data/img/kyaru/20.png
diff --git a/ATRI/data/img/kyaru/21.png b/ATRI/data/img/kyaru/21.png
new file mode 100644
index 0000000..a65b9c0
--- /dev/null
+++ b/ATRI/data/img/kyaru/21.png
diff --git a/ATRI/data/img/kyaru/22.png b/ATRI/data/img/kyaru/22.png
new file mode 100644
index 0000000..2c821a9
--- /dev/null
+++ b/ATRI/data/img/kyaru/22.png
diff --git a/ATRI/data/img/kyaru/23.png b/ATRI/data/img/kyaru/23.png
new file mode 100644
index 0000000..239f4f6
--- /dev/null
+++ b/ATRI/data/img/kyaru/23.png
diff --git a/ATRI/data/img/kyaru/24.png b/ATRI/data/img/kyaru/24.png
new file mode 100644
index 0000000..33b0774
--- /dev/null
+++ b/ATRI/data/img/kyaru/24.png
diff --git a/ATRI/data/img/kyaru/25.png b/ATRI/data/img/kyaru/25.png
new file mode 100644
index 0000000..1cf424a
--- /dev/null
+++ b/ATRI/data/img/kyaru/25.png
diff --git a/ATRI/data/img/kyaru/26.png b/ATRI/data/img/kyaru/26.png
new file mode 100644
index 0000000..70155b4
--- /dev/null
+++ b/ATRI/data/img/kyaru/26.png
diff --git a/ATRI/data/img/kyaru/27.png b/ATRI/data/img/kyaru/27.png
new file mode 100644
index 0000000..75bf8a5
--- /dev/null
+++ b/ATRI/data/img/kyaru/27.png
diff --git a/ATRI/data/img/kyaru/28.png b/ATRI/data/img/kyaru/28.png
new file mode 100644
index 0000000..771f4be
--- /dev/null
+++ b/ATRI/data/img/kyaru/28.png
diff --git a/ATRI/data/img/kyaru/29.png b/ATRI/data/img/kyaru/29.png
new file mode 100644
index 0000000..a7528af
--- /dev/null
+++ b/ATRI/data/img/kyaru/29.png
diff --git a/ATRI/data/img/kyaru/3.png b/ATRI/data/img/kyaru/3.png
new file mode 100644
index 0000000..0629642
--- /dev/null
+++ b/ATRI/data/img/kyaru/3.png
diff --git a/ATRI/data/img/kyaru/30.png b/ATRI/data/img/kyaru/30.png
new file mode 100644
index 0000000..b76853c
--- /dev/null
+++ b/ATRI/data/img/kyaru/30.png
diff --git a/ATRI/data/img/kyaru/31.png b/ATRI/data/img/kyaru/31.png
new file mode 100644
index 0000000..932a7fb
--- /dev/null
+++ b/ATRI/data/img/kyaru/31.png
diff --git a/ATRI/data/img/kyaru/32.png b/ATRI/data/img/kyaru/32.png
new file mode 100644
index 0000000..6e6b378
--- /dev/null
+++ b/ATRI/data/img/kyaru/32.png
diff --git a/ATRI/data/img/kyaru/33.png b/ATRI/data/img/kyaru/33.png
new file mode 100644
index 0000000..70822d6
--- /dev/null
+++ b/ATRI/data/img/kyaru/33.png
diff --git a/ATRI/data/img/kyaru/34.png b/ATRI/data/img/kyaru/34.png
new file mode 100644
index 0000000..83e4341
--- /dev/null
+++ b/ATRI/data/img/kyaru/34.png
diff --git a/ATRI/data/img/kyaru/35.png b/ATRI/data/img/kyaru/35.png
new file mode 100644
index 0000000..2e99fcf
--- /dev/null
+++ b/ATRI/data/img/kyaru/35.png
diff --git a/ATRI/data/img/kyaru/36.png b/ATRI/data/img/kyaru/36.png
new file mode 100644
index 0000000..f45a78e
--- /dev/null
+++ b/ATRI/data/img/kyaru/36.png
diff --git a/ATRI/data/img/kyaru/37.png b/ATRI/data/img/kyaru/37.png
new file mode 100644
index 0000000..b79fd3f
--- /dev/null
+++ b/ATRI/data/img/kyaru/37.png
diff --git a/ATRI/data/img/kyaru/38.png b/ATRI/data/img/kyaru/38.png
new file mode 100644
index 0000000..5591c12
--- /dev/null
+++ b/ATRI/data/img/kyaru/38.png
diff --git a/ATRI/data/img/kyaru/39.png b/ATRI/data/img/kyaru/39.png
new file mode 100644
index 0000000..a35a98a
--- /dev/null
+++ b/ATRI/data/img/kyaru/39.png
diff --git a/ATRI/data/img/kyaru/4.png b/ATRI/data/img/kyaru/4.png
new file mode 100644
index 0000000..db0301f
--- /dev/null
+++ b/ATRI/data/img/kyaru/4.png
diff --git a/ATRI/data/img/kyaru/5.png b/ATRI/data/img/kyaru/5.png
new file mode 100644
index 0000000..8212f94
--- /dev/null
+++ b/ATRI/data/img/kyaru/5.png
diff --git a/ATRI/data/img/kyaru/6.png b/ATRI/data/img/kyaru/6.png
new file mode 100644
index 0000000..c005aa8
--- /dev/null
+++ b/ATRI/data/img/kyaru/6.png
diff --git a/ATRI/data/img/kyaru/7.png b/ATRI/data/img/kyaru/7.png
new file mode 100644
index 0000000..3818ee9
--- /dev/null
+++ b/ATRI/data/img/kyaru/7.png
diff --git a/ATRI/data/img/kyaru/8.png b/ATRI/data/img/kyaru/8.png
new file mode 100644
index 0000000..d09733b
--- /dev/null
+++ b/ATRI/data/img/kyaru/8.png
diff --git a/ATRI/data/img/kyaru/9.png b/ATRI/data/img/kyaru/9.png
new file mode 100644
index 0000000..8978fba
--- /dev/null
+++ b/ATRI/data/img/kyaru/9.png
diff --git a/ATRI/data/img/kyaru/idk.png b/ATRI/data/img/kyaru/idk.png
new file mode 100644
index 0000000..e8df7d9
--- /dev/null
+++ b/ATRI/data/img/kyaru/idk.png
diff --git a/ATRI/data/sqlite/setu/nearR18.db b/ATRI/data/sqlite/setu/nearR18.db
index 7e8d3fd..cb6e128 100644
--- a/ATRI/data/sqlite/setu/nearR18.db
+++ b/ATRI/data/sqlite/setu/nearR18.db
diff --git a/ATRI/data/sqlite/setu/normal.db b/ATRI/data/sqlite/setu/normal.db
index 476e9de..13d61ab 100644
--- a/ATRI/data/sqlite/setu/normal.db
+++ b/ATRI/data/sqlite/setu/normal.db
diff --git a/ATRI/data/sqlite/setu/r18.db b/ATRI/data/sqlite/setu/r18.db
index 2319292..7bcd784 100644
--- a/ATRI/data/sqlite/setu/r18.db
+++ b/ATRI/data/sqlite/setu/r18.db
diff --git a/ATRI/plugins/AIchangeFace.py b/ATRI/plugins/AIchangeFace.py
index 1d82e12..3b72997 100644
--- a/ATRI/plugins/AIchangeFace.py
+++ b/ATRI/plugins/AIchangeFace.py
@@ -1,9 +1,11 @@
 import os
+import cv2
 import json
 import requests
 import base64
 import nonebot
 import time
+from pathlib import Path
 from nonebot import on_command, CommandSession
 
 import config # type: ignore
@@ -13,6 +15,7 @@ bot = nonebot.get_bot()
 master = config.MASTER()
 key = config.FACE_KEY()
 secret = config.FACE_SECRET()
+SCALE_FACTOR = 1 # 图像的放缩比
 
 
 #获取图片的人脸特征参数
@@ -27,7 +30,7 @@ def find_face(imgpath):
 
 
 #换脸,函数传参中number表示两张脸的相似度为99%
-def change_face(image_1, image_2, user, number=99):
+def change_face(image_1, image_2, user, kyaru, number=99):
     url = "https://api-cn.faceplusplus.com/imagepp/v1/mergeface"
     find_p1 = find_face(image_1)
     find_p2 = find_face(image_2)
@@ -47,7 +50,13 @@ def change_face(image_1, image_2, user, number=99):
     response = requests.post(url,data=data).json()
     results = response['result']
     image = base64.b64decode(results)
-    files = f'ATRI/data/temp/face/{user}'
+    files = 'test'
+    if int(kyaru) == 1:
+        files = f'ATRI/data/temp/face/{user}'
+    elif int(kyaru) == 2:
+        files = f'ATRI/data/temp/head/{user}'
+    print(files)
+
     if not os.path.exists(files):
         os.mkdir(files)
     with open(files + '/img3.jpg','wb') as file:
@@ -61,11 +70,11 @@ def change_face(image_1, image_2, user, number=99):
 @on_command('ai_ch_face', aliases = ['AI换脸', 'ai换脸'], only_to_me = False)
 async def _(session: CommandSession):
     user = session.event.user_id
-    with open("ATRI/plugins/switch/switch.json", 'r') as f:
+    with open(Path('.') / 'ATRI' / 'plugins' / 'switch' / 'switch.json', 'r') as f:
         data = json.load(f)
     
     if data["change_face"] == 0:
-        with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
+        with open(Path('.') / 'ATRI' / 'plugins' / 'noobList' / 'noobList.json', 'r') as f:
             data0 = json.load(f)
         
         if str(user) in data0.keys():
@@ -102,22 +111,78 @@ async def _(session: CommandSession):
             except:
                 session.finish('请求数据貌似失败了...')
             
-            img1File = f'ATRI/data/temp/face/{user}/img1.jpg'
-            img2File = f'ATRI/data/temp/face/{user}/img2.jpg'
+            img1File = Path('.') / 'ATRI' / 'data' / 'temp' / 'face' / f'{user}' / 'img1.jpg'
+            img2File = Path('.') / 'ATRI' / 'data' / 'temp' / 'face' / f'{user}' / 'img2.jpg'
 
             try:
-                change_face(img1File, img2File, user)
+                change_face(img1File, img2File, user, 1)
             except:
                 session.finish('emm...貌似失败了呢......')
             
             time.sleep(0.5)
-            doneIMG = f'ATRI/data/temp/face/{user}/img3.jpg'
+            doneIMG = Path('.') / 'ATRI' / 'data' / 'temp' / 'face' / f'{user}' / 'img3.jpg'
             img = os.path.abspath(doneIMG)
             await session.send(f'[CQ:image,file=file:///{img}]')
             files = f'ATRI/data/temp/face/{user}'
             os.remove(files)
 
 
-@on_command('change_u_head', aliases = ['接头霸王'], only_to_me = False)
-async def _(session: CommandSession):
-    pass # 明天做
-\ No newline at end of file
+# def f_1(x, A, B):
+#     return A*x + B
+
+# @on_command('change_u_head', aliases = ['接头霸王'], only_to_me = False)
+# async def _(session: CommandSession):
+#     user = session.event.user_id
+#     with open("ATRI/plugins/switch/switch.json", 'r') as f:
+#         data = json.load(f)
+    
+#     if data["change_face"] == 0:
+#         with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
+#             data0 = json.load(f)
+        
+#         if str(user) in data0.keys():
+#             pass
+#         else:
+#             img1 = session.get('img1', prompt = '请发送需要换头的图片')
+#             a = img1.split(',')
+#             a = a[2].replace(']', '')
+#             a = a.replace('url=', '')
+#             print(a)
+#             try:
+#                 imgres1 = requests.get(a)
+#                 file1 = f'ATRI/data/temp/head/{user}'
+#                 if not os.path.exists(file1):
+#                     os.mkdir(file1)
+#                 with open(file1 + '/img1.jpg', 'wb') as f:
+#                     f.write(imgres1.content)
+#             except:
+#                 session.finish('获取数据貌似失败了呢...')
+#             img1File = Path('.') / 'ATRI' / 'data' / 'temp' / 'head' / f'{user}' / 'img1.jpg'
+
+#             imgN = Path('.') / 'ATRI' / 'data' / 'img' / 'kyaru' / 'idk.png '
+#             img = os.path.abspath(imgN)
+#             await session.send(f'[CQ:image,file=:///{img}]')
+#             head = session.get('head', prompt = '请输入头的序号，例如选择：1，发送：1')
+#             if head.isdigit():
+#                 pass
+#             else:
+#                 await session.send('请输入阿拉伯数字！')
+#                 return
+#             headIMG = Path('.') / 'ATRI' / 'data' / 'img' / 'kyaru' / f'{int(head)}.png'
+
+#             try:
+#                 img = cv2.imread(img1File)
+#                 face_cascade = cv2.CascadeClassifier(cv2.haarcascades + r'haarcascade_frontalface_default.xml')
+#                 gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+#                 faces = face_cascade.detectMultiScale(gray, scaleFactor = 1.15, minNeighbors = 5, minSize = (5, 5))
+#                 await session.send(faces)
+
+#             except:
+#                 session.finish('emm...貌似焊接失败了呢......')
+            
+#             time.sleep(0.5)
+#             doneIMG = Path('.') / 'ATRI' / 'data' / 'temp' / 'head' / f'{user}' / 'img3.jpg'
+#             img = os.path.abspath(doneIMG)
+#             await session.send(f'[CQ:image,file=file:///{img}]')
+#             files = f'ATRI/data/temp/head/{user}'
+#             os.remove(files)
+\ No newline at end of file
diff --git a/ATRI/plugins/__pycache__/AIchangeFace.cpython-38.pyc b/ATRI/plugins/__pycache__/AIchangeFace.cpython-38.pyc
new file mode 100644
index 0000000..bb2e1bb
--- /dev/null
+++ b/ATRI/plugins/__pycache__/AIchangeFace.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/animeSearch.cpython-38.pyc b/ATRI/plugins/__pycache__/animeSearch.cpython-38.pyc
new file mode 100644
index 0000000..bdedd1e
--- /dev/null
+++ b/ATRI/plugins/__pycache__/animeSearch.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/chat.cpython-38.pyc b/ATRI/plugins/__pycache__/chat.cpython-38.pyc
index 6dd0d2a..c514c22 100644
--- a/ATRI/plugins/__pycache__/chat.cpython-38.pyc
+++ b/ATRI/plugins/__pycache__/chat.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/checkData.cpython-38.pyc b/ATRI/plugins/__pycache__/checkData.cpython-38.pyc
new file mode 100644
index 0000000..0bc957b
--- /dev/null
+++ b/ATRI/plugins/__pycache__/checkData.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/check_data.cpython-38.pyc b/ATRI/plugins/__pycache__/check_data.cpython-38.pyc
new file mode 100644
index 0000000..bc17da5
--- /dev/null
+++ b/ATRI/plugins/__pycache__/check_data.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/setu.cpython-38.pyc b/ATRI/plugins/__pycache__/setu.cpython-38.pyc
index 8da12f5..94c589b 100644
--- a/ATRI/plugins/__pycache__/setu.cpython-38.pyc
+++ b/ATRI/plugins/__pycache__/setu.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/switch.cpython-38.pyc b/ATRI/plugins/__pycache__/switch.cpython-38.pyc
index df8416c..3197411 100644
--- a/ATRI/plugins/__pycache__/switch.cpython-38.pyc
+++ b/ATRI/plugins/__pycache__/switch.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/upload_sqlite.cpython-38.pyc b/ATRI/plugins/__pycache__/upload_sqlite.cpython-38.pyc
index 6d1153c..291f33e 100644
--- a/ATRI/plugins/__pycache__/upload_sqlite.cpython-38.pyc
+++ b/ATRI/plugins/__pycache__/upload_sqlite.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/welcome.cpython-38.pyc b/ATRI/plugins/__pycache__/welcome.cpython-38.pyc
index d0736b6..25b8468 100644
--- a/ATRI/plugins/__pycache__/welcome.cpython-38.pyc
+++ b/ATRI/plugins/__pycache__/welcome.cpython-38.pyc
diff --git a/ATRI/plugins/__pycache__/wordCloud.cpython-38.pyc b/ATRI/plugins/__pycache__/wordCloud.cpython-38.pyc
new file mode 100644
index 0000000..49556bf
--- /dev/null
+++ b/ATRI/plugins/__pycache__/wordCloud.cpython-38.pyc