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

XMWAI/__init__.py

@@ -2,6 +2,8 @@ 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  # 从子模块导入函数到顶层
 
 __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"]  # 可选：明确导出的内容

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,171 @@
+import base64
+import urllib
+import requests
+import json
+from PIL import Image
+from PIL import ImageGrab
+from io import BytesIO
+import time
+import os
+import ctypes
+
+
+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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.2.7
+Version: 0.2.8
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment
@@ -12,6 +12,7 @@ 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
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

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

@@ -1,9 +1,10 @@
-XMWAI/__init__.py,sha256=91jaQmcGqp_B4i68bNfn_0m6JTFzz-itTiCmt__vIR8,487
+XMWAI/__init__.py,sha256=lkYAOe3hpF_pg8UMnTVqDCy4X63Ozu7VmsvNY_sC6e0,701
 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=RrpsRQ1Jtl1Qivht08nzGMgeTf-4I7jW8YGnDlBKRzk,6056
 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.8.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
+xmwai-0.2.8.dist-info/METADATA,sha256=sIk8AWvgMlf3RWaKNl4L4CmcSNEo3v_Af04vS69eZRI,1064
+xmwai-0.2.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xmwai-0.2.8.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
+xmwai-0.2.8.dist-info/RECORD,,