XMWAI 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

XMWAI/__init__.py

@@ -2,6 +2,9 @@ from .core import story, photo, reply, poem, crack  # 从子模块导入函数
 from .magic_core import birthday  # 从子模块导入函数到顶层
 from .bomb_core import bomb  # 从子模块导入函数到顶层
 from .idiom_core import idiom, searchIdiom, get_json_path  # 从子模块导入函数到顶层
+from .trial_class import make, get_file_content_as_base64, save_pic, detect_windows_scale, get_file_content_as_base64_2, cartoon  # 从子模块导入函数到顶层
 
 __all__ = ["story", "photo", "reply", "poem", 'birthday', 'bomb',
-           "idiom", "searchIdiom", "get_json_path", "crack"]  # 可选：明确导出的内容
+           "idiom", "searchIdiom", "get_json_path", "crack",
+           "make", "get_file_content_as_base64", "save_pic", "detect_windows_scale",
+           "get_file_content_as_base64_2", "cartoon"]  # 可选：明确导出的内容

XMWAI/core.py

@@ -5,9 +5,6 @@ import os
 import zipfile
 
 
-# AI绘图大师函数
-
-
 # U1-02 Story GPT----------------------------------
 def story(role, time, address, event, key=""):
     content = role+time+address+event

XMWAI/trial_class.py

@@ -0,0 +1,283 @@
+import base64
+import tkinter as tk
+import urllib
+import requests
+import json
+from PIL import Image
+from PIL import ImageGrab
+from io import BytesIO
+import time
+import os
+import ctypes
+import cv2
+import numpy as np
+
+'''-----------体验课1-----------'''
+
+
+def make(screen):
+    save_pic(screen)
+    while not os.path.exists("pic.png"):
+        pass
+
+    """
+    使用 AK，SK 生成鉴权签名（Access Token）
+    :return: access_token，或是None(如果错误)
+    """
+    API_KEY = "YleMT041wl8zkXhk1Y4AdEuk"
+    SECRET_KEY = "xQousjKEqphGwVMKHJlUSGDXp7PiUVpk"
+    url = "https://aip.baidubce.com/oauth/2.0/token"
+    params = {"grant_type": "client_credentials",
+              "client_id": API_KEY, "client_secret": SECRET_KEY}
+
+    # 获取访问令牌并构建API请求URL
+    access_token = str(requests.post(
+        url, params=params).json().get("access_token"))
+    create_url = f"https://aip.baidubce.com/rpc/2.0/ernievilg/v1/txt2imgv2?access_token={access_token}"
+    query_url = f"https://aip.baidubce.com/rpc/2.0/ernievilg/v1/getImgv2?access_token={access_token}"
+
+    # 读取图像并转换为Base64编码
+    with open("pic.png", "rb") as image_file:
+        base64_string = base64.b64encode(image_file.read()).decode('utf-8')
+
+    # 构建创建任务的请求参数
+    create_payload = json.dumps({
+        "prompt": "参考当前图，希望画面卡通一些，画面可以丰富一些，内容积极向上，参考宫崎骏画风",
+        "width": 1024,
+        "height": 1024,
+        "image": base64_string,
+        "change_degree": 1
+    }, ensure_ascii=False)
+
+    headers = {
+        'Content-Type': 'application/json',
+        'Accept': 'application/json'
+    }
+
+    # 发送创建任务请求
+    response = requests.post(create_url, headers=headers,
+                             data=create_payload.encode("utf-8"))
+    response.raise_for_status()  # 检查请求是否成功
+    task_id = response.json()["data"]["task_id"]
+    # print(f"任务已创建，ID: {task_id}")
+
+    # 轮询检查任务状态
+    query_payload = json.dumps({"task_id": task_id}, ensure_ascii=False)
+    task_status = "RUNNING"
+    print("AI图片生成中.......")
+    while task_status == "RUNNING":
+        time.sleep(30)
+        response = requests.post(
+            query_url, headers=headers, data=query_payload.encode("utf-8"))
+        response.raise_for_status()
+        task_status = response.json()["data"]["task_status"]
+        # print(f"任务状态: {task_status}")
+
+    # 处理任务结果
+    if task_status == "SUCCESS":
+        picture = requests.get(response.json()[
+                               "data"]["sub_task_result_list"][0]["final_image_list"][0]["img_url"])
+        image_data = BytesIO(picture.content)
+        image = Image.open(image_data)
+        image.save('image.gif')
+        os.system("image.gif")
+    else:
+        print(f"任务失败，状态: {task_status}")
+
+
+def get_file_content_as_base64(path, urlencoded=False):
+    """
+    获取文件base64编码
+    :param path: 文件路径
+    :param urlencoded: 是否对结果进行urlencoded 
+    :return: base64编码信息
+    """
+    with open(path, "rb") as f:
+        content = base64.b64encode(f.read()).decode("utf8")
+        if urlencoded:
+            content = urllib.parse.quote_plus(content)
+    return content
+
+
+def save_pic(screen, output_file="pic.png"):
+    """
+    精准截取turtle绘图区域，不包含窗口边框和标题栏
+    专为Windows系统优化，支持高DPI缩放
+
+    参数:
+        screen: turtle的Screen对象
+        output_file: 输出文件名
+    """
+    canvas = screen.getcanvas()
+    screen.update()
+
+    try:
+        # 获取画布位置和大小
+        x = canvas.winfo_rootx()
+        y = canvas.winfo_rooty()
+        width = canvas.winfo_width()
+        height = canvas.winfo_height()
+
+        # 检测Windows屏幕缩放比例
+        scale_factor = detect_windows_scale()
+        # print(f"检测到屏幕缩放比例: {scale_factor}x")
+
+        # 调整截图区域，排除窗口边框
+        # 典型Windows窗口边框宽度约为8像素，标题栏约为30像素
+        border_width = int(8 * scale_factor)
+        title_height = int(30 * scale_factor)
+
+        # 计算实际绘图区域
+        img = ImageGrab.grab(
+            bbox=(
+                x + border_width,  # 左边界，排除左侧边框
+                y + title_height,  # 上边界，排除标题栏和上边框
+                x + width - border_width,  # 右边界，排除右侧边框
+                y + height - border_width  # 下边界，排除底部边框
+            )
+        )
+
+        # 保存为PNG
+        img.save(output_file)
+        # print(f"已保存图形到 {output_file} (尺寸: {img.size})")
+    except Exception as e:
+        print(f"截图时出错: {e}")
+        print("提示: 尝试使用save_turtle_canvas_fallback函数或手动截图")
+
+
+def detect_windows_scale():
+    """
+    检测Windows系统的屏幕缩放比例
+    优先使用高精度方法，失败则降级使用兼容性方法
+    """
+    try:
+        # 高精度方法：获取系统DPI
+        user32 = ctypes.windll.user32
+        user32.SetProcessDPIAware()
+
+        # 尝试获取系统DPI (Windows 8.1及以上)
+        if hasattr(user32, 'GetDpiForSystem'):
+            dpi = user32.GetDpiForSystem()
+            return dpi / 96.0  # Windows标准DPI是96
+
+        # 尝试获取显示器DPI (Windows 10及以上)
+        if hasattr(user32, 'GetDpiForWindow'):
+            root = tk.Tk()
+            dpi = user32.GetDpiForWindow(root.winfo_id())
+            root.destroy()
+            return dpi / 96.0
+
+        # 兼容方法：获取屏幕尺寸并与标准尺寸比较
+        screen_width = user32.GetSystemMetrics(0)
+        return screen_width / 1920.0  # 假设标准分辨率为1920x1080
+    except Exception as e:
+        print(f"检测缩放比例时出错: {e}")
+        print("使用默认缩放比例1.0")
+        return 1.0
+
+
+'''-----------体验课2-----------'''
+
+
+def get_file_content_as_base64_2(path, urlencoded=False):
+    """
+    获取文件base64编码
+    :param path: 文件路径
+    :param urlencoded: 是否对结果进行urlencoded 
+    :return: base64编码信息
+    """
+    with open(path, "rb") as f:
+        content = base64.b64encode(f.read()).decode("utf8")
+        if urlencoded:
+            content = urllib.parse.quote_plus(content)
+    return content
+
+
+def cartoon(name):
+    """
+    使用 AK，SK 生成鉴权签名（Access Token）
+    :return: access_token，或是None(如果错误)
+    """
+    API_KEY = "LjdoA5Ar7MGrwynZfTFcB7K3"
+    SECRET_KEY = "3htSVp4IhW8LIyetP5Yo8NdWvF0yNH0W"
+    url = "https://aip.baidubce.com/oauth/2.0/token"
+    params = {"grant_type": "client_credentials",
+              "client_id": API_KEY, "client_secret": SECRET_KEY}
+    s = str(requests.post(url, params=params).json().get("access_token"))
+
+    url = "https://aip.baidubce.com/rest/2.0/image-process/v1/selfie_anime?access_token=" + s
+    base64_string = get_file_content_as_base64_2("pic.jpg", True)
+    payload = "image={}".format(base64_string)
+    headers = {
+        'Content-Type': 'application/x-www-form-urlencoded',
+        'Accept': 'application/json'
+    }
+    response = requests.request(
+        "POST", url, headers=headers, data=payload.encode("utf-8"))
+    base64_data = response.json()["image"]
+    image_data = base64.b64decode(base64_data)
+    with open("img.png", 'wb') as f:
+        f.write(image_data)
+
+    photo = cv2.imread('img.png')
+    original = cv2.imread('pic.jpg')
+    height, width = photo.shape[:2]
+
+    # 扫描效果参数
+    scan_bar_height = 60  # 扫描光带高度
+    glow_width = 15       # 光晕宽度
+    glow_intensity = 0.3  # 光晕强度
+    scan_speed = 5        # 扫描速度（毫秒）
+
+    # 创建一个与图像大小相同的透明层用于绘制扫描效果
+    overlay = np.zeros_like(original, dtype=np.uint8)
+
+    for i in range(-scan_bar_height, height + scan_bar_height):
+        # 创建一个空白图像
+        display = original.copy()
+        # 只显示扫描到的部分
+        if i > 0:
+            display[:min(i, height), :] = photo[:min(i, height), :]
+
+        # 清除上一帧的扫描效果
+        overlay.fill(0)
+
+        # 计算扫描光带位置
+        scan_start = max(0, i - scan_bar_height // 2)
+        scan_end = min(height, i + scan_bar_height // 2)
+
+        if scan_start < scan_end:
+            # 创建渐变扫描光带
+            for y in range(scan_start, scan_end):
+                # 计算渐变因子（0-1-0）
+                distance_from_center = abs(y - i)
+                gradient_factor = 1.0 - \
+                    (distance_from_center / (scan_bar_height / 2))
+
+                # 绿色扫描线主体
+                green_intensity = int(255 * gradient_factor)
+                overlay[y, :, 1] = green_intensity
+
+                # 添加轻微的蓝色调，使扫描线更真实
+                overlay[y, :, 2] = int(60 * gradient_factor)
+
+            # 添加光晕效果
+            for y in range(max(0, scan_start - glow_width), min(height, scan_end + glow_width)):
+                if y < scan_start or y >= scan_end:
+                    # 计算光晕强度
+                    distance_from_edge = min(
+                        abs(y - scan_start), abs(y - scan_end))
+                    glow_factor = max(
+                        0, 1.0 - (distance_from_edge / glow_width)) * glow_intensity
+
+                    # 添加绿色光晕
+                    overlay[y, :, 1] = np.clip(
+                        overlay[y, :, 1] + (255 * glow_factor), 0, 255)
+
+            # 将半透明扫描效果叠加到图像上
+            alpha = 0.5  # 透明度
+            cv2.addWeighted(overlay, alpha, display, 1 - alpha, 0, display)
+
+        # 显示稳定的扫描效果
+        cv2.imshow(name, display)
+        cv2.waitKey(scan_speed)  # 控制扫描速度

{xmwai-0.2.7.dist-info → xmwai-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.2.7
+Version: 0.2.9
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment
@@ -12,6 +12,9 @@ Requires-Python: >=3.7.0
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: requests>=2.32.3
+Requires-Dist: Pillow>=11.1.0
+Requires-Dist: opencv-python>=3.4.18.65
+Requires-Dist: numpy>=2.2.3
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{xmwai-0.2.7.dist-info → xmwai-0.2.9.dist-info}/RECORD

@@ -1,9 +1,10 @@
-XMWAI/__init__.py,sha256=91jaQmcGqp_B4i68bNfn_0m6JTFzz-itTiCmt__vIR8,487
+XMWAI/__init__.py,sha256=Kz93zgrnLiGqlqCp6y8Y4McEBJOznvQrS-IPhwsvCTw,795
 XMWAI/bomb_core.py,sha256=h2ZPH3SuoG2L_XOf1dcK8p3lhw5QzhneWl2yMLj1RiU,1819
 XMWAI/cookbook_core.py,sha256=mo3IYgh_bYD_3vH2Stkjwo9okFOb4abooZeJ-Djx6VU,3675
-XMWAI/core.py,sha256=H4EpXmXXakVNleUKyUMVnl3AmR8XjjVLsVfGvsLRUqg,7933
+XMWAI/core.py,sha256=SxzfbOW7WZALuA5fQ4dGieJR2-roxufjXQ_mqwGDg3E,7905
 XMWAI/idiom_core.py,sha256=yU-VHhqqoutVm6GVikcjL3m9yuB1hUsFBpPYvwY4n5g,1689
 XMWAI/magic_core.py,sha256=Ms4b12PJ8rjsmceg1VUqWCWx2ebvdhLp4sIF6K_Vaok,3491
+XMWAI/trial_class.py,sha256=GRDEe24a-DYMRyoTLgJbE5dCSQY0CbYqB5ALsBfYmtA,10226
 XMWAI/file/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 XMWAI/file/idiom.json,sha256=HUtPRUzhxBbWoasjadbmbA_5ngQ5AXLu9weQSZ4hzhk,10319857
 XMWAI/gif/0.gif,sha256=LGpAReVTyZEb1J-bWYTpxxHbIxmLJ-3wA0lw4cBqdsY,303
@@ -93,8 +94,8 @@ XMWAI/gif/84.gif,sha256=6Ip_uQmvrr2fagLXu1YqWyI_DL2PVVtKCPtmNtzt3P4,38767
 XMWAI/gif/85.gif,sha256=6Ip_uQmvrr2fagLXu1YqWyI_DL2PVVtKCPtmNtzt3P4,38767
 XMWAI/gif/9.gif,sha256=cPouth-xwc3QPcg2m6HMP2OD1ZCeRBD_-RPImvvKAA0,9485
 XMWAI/gif/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xmwai-0.2.7.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
-xmwai-0.2.7.dist-info/METADATA,sha256=-pGe6LcF2Fy7GeNWgQxhP8dEOcKyRpVZ_HeEQ2FnQDY,1033
-xmwai-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xmwai-0.2.7.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
-xmwai-0.2.7.dist-info/RECORD,,
+xmwai-0.2.9.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
+xmwai-0.2.9.dist-info/METADATA,sha256=73264zV2DE4eGCdSDdI-Ba7G6RvZhd1cyWc4cWX3hFU,1134
+xmwai-0.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xmwai-0.2.9.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
+xmwai-0.2.9.dist-info/RECORD,,