XMWAI 0.1.5__tar.gz → 0.1.7__tar.gz

xmwai-0.1.7/MANIFEST.in

@@ -0,0 +1,2 @@
+recursive-include XMWAI/gif *
+recursive-include XMWAI/idiom.json

{xmwai-0.1.5/XMWAI.egg-info → xmwai-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.1.5
+Version: 0.1.7
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment

{xmwai-0.1.5 → xmwai-0.1.7}/XMWAI/__init__.py

@@ -1,5 +1,6 @@
 from .core import story, photo, reply, poem, get_access_token  # 从子模块导入函数到顶层
 from .magic_core import birthday  # 从子模块导入函数到顶层
 from .bomb_core import bomb  # 从子模块导入函数到顶层
+from .idiom_core import idiom,searchIdiom  # 从子模块导入函数到顶层
 
-__all__ = ["story", "photo", "reply", "poem", "get_access_token", 'birthday', 'bomb']  # 可选：明确导出的内容
+__all__ = ["story", "photo", "reply", "poem", "get_access_token", 'birthday', 'bomb', "idiom", "searchIdiom"]  # 可选：明确导出的内容

{xmwai-0.1.5 → xmwai-0.1.7}/XMWAI/bomb_core.py

@@ -24,7 +24,7 @@ def bomb(screen, x, y):
 
     # 第一次加载时：将所有 gif 加载为 turtle 形状
     if not hasattr(bomb, "_gif_loaded"):
-        gif_dir = files("XMAI.gif")  # 注意：XMAI/gif/__init__.py 必须存在
+        gif_dir = files("XMWAI.gif")  # 注意：XMWAI/gif/__init__.py 必须存在
 
         bomb._shapes = []  # 保存所有完整路径
         for i in range(86):

{xmwai-0.1.5 → xmwai-0.1.7}/XMWAI/core.py

@@ -184,24 +184,34 @@ def poem(title, key=""):
     elif key != "CaJQ":
         print("秘钥错误！请重新输入！")
         return "秘钥错误！请重新输入！"
-    key = "ZZC"
-    access_token = get_access_token(key)
-    if access_token:
-        url = f"https://aip.baidubce.com/rpc/2.0/nlp/v1/poem?access_token={access_token}"
-        payload = {
-            "text": title,  # 直接使用普通字符串
-        }
-        headers = {
-            'Content-Type': 'application/json',
-            'Accept': 'application/json'
-        }
+    messagesList = [
+            {
+                "role": "system",
+                "content": "唐代诗人"
+            },
+            {
+                "role": "user",
+                "content": f"请以《{title}》为题，创作一首七言绝句，每句7个字，一共4句，符合古诗韵律规范，内容积极乐观向上，适合中小学生阅读，不要解析，不要题目，不要标点符号，所有文字放在一行"
+            }
+            ]
         
-        # 直接发送JSON对象，不转换为字符串
-        response = requests.post(url, headers=headers, json=payload)
-        if response.status_code == 200:
-            return response.json()['poem'][0]['content']  # 使用response.json()直接解析JSON响应
-        else:
-            print("Failed with status code:", response.status_code)
-            print("Response:", response.text)
-    else:
-        print("Failed to get access token.")
+    url = "https://qianfan.baidubce.com/v2/chat/completions"
+    
+    payload = json.dumps({
+        "model": "ernie-4.5-turbo-32k",
+        "messages": messagesList
+    }, ensure_ascii=False)
+    headers = {
+        'Content-Type': 'application/json',
+        'appid': '',
+        'Authorization': 'Bearer bce-v3/ALTAK-cGbxpVA5AbSz6h8nbLaFh/b539762075d55c76d93dc78bcf0a91beeaf0490a'
+    }
+    
+    response = requests.request("POST", url, headers=headers, data=payload.encode("utf-8"))
+    
+    response_data = json.loads(response.text)
+    content = response_data["choices"][0]["message"]["content"]
+    content.replace(" ", "")
+    content.replace("\n", "")
+    content.replace("  ", "")
+    return content

{xmwai-0.1.5 → xmwai-0.1.7}/XMWAI/idiom_core.py

@@ -1,5 +1,4 @@
 import random
-
 import json
 
 
@@ -56,8 +55,11 @@ def searchIdiom(text, num=1):
     with open("idiom.json", "r", encoding="utf-8") as f:
         data = json.load(f)
         for i in data:
-            if text == i["word"][num-1]:
-                wordList.append(i["word"])
+            try:
+                if text == i["word"][num-1]:
+                    wordList.append(i["word"])
+            except:
+                pass
     if wordList:
         return random.choice(wordList)
     else:

{xmwai-0.1.5 → xmwai-0.1.7/XMWAI.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.1.5
+Version: 0.1.7
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment

{xmwai-0.1.5 → xmwai-0.1.7}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="XMWAI",  # 包名（pip install XMWAI）
-    version="0.1.5",  # 初始版本号
+    version="0.1.7",  # 初始版本号
     author="pydevelopment",  # 作者
     author_email="hekai@xiaoma.cn",  # 邮箱
     description="Small code King AI related library",  # 简短描述

xmwai-0.1.5/MANIFEST.in

@@ -1 +0,0 @@
-recursive-include XMWAI/gif *