MeUtils 2024.3.4.13.4.45__py3-none-any.whl → 2025.1.16.17.15.52__py3-none-any.whl

meutils/apis/kling/api.py

@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : api
+# @Time         : 2024/10/10 10:05
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : https://github.com/nodetool-ai/nodetool/blob/bfd32cf40dcf1fe2711803f9c8730c0e816067e4/src/nodetool/nodes/kling/api.py#L99
+
+import jwt
+
+from meutils.pipe import *
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+
+from meutils.schemas.kling_types import API_BASE_URL, API_FEISHU_URL, STATUSES, send_message
+from meutils.schemas.kling_types import ImageRequest, VideoRequest
+
+
+def encode_jwt_token(ak, sk):
+    headers = {
+        "alg": "HS256",
+        "typ": "JWT"
+    }
+    payload = {
+        "iss": ak,
+        "exp": int(time.time()) + 1800,  # 有效时间，此处示例代表当前时间+1800s(30min)
+        "nbf": int(time.time()) - 5  # 开始生效的时间，此处示例代表当前时间-5秒
+    }
+    token = jwt.encode(payload, sk, headers=headers)
+    logger.debug(token)
+    return token
+
+
+async def create_task(request: VideoRequest, token: Optional[str] = None):
+    token = token or await get_next_token_for_polling(API_FEISHU_URL)
+    token = "ee6503baf0e447d59a3812fb2a344570|da8322347a7b4d5a9c04a088427883f7"
+    ak, sk = token.split("|")
+
+    headers = {
+        'content-type': 'application/json;charset=utf-8',
+        "Authorization": f"Bearer {encode_jwt_token(ak, sk)}"
+    }
+    payload = request.model_dump(exclude_none=True)
+    async with httpx.AsyncClient(base_url=API_BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post("/v1/videos/text2video", json=payload)
+        response.raise_for_status()
+        return response.json()
+        #
+        # if response.is_success:
+        #     data = response.json()
+        #     return data
+        # else:
+        #     raise Exception(f"请求失败，状态码：{response.status_code}，响应内容：{response.text}")
+
+
+if __name__ == '__main__':
+    # api_token = encode_jwt_token(ak, sk)
+    # print(api_token)  # 打印生成的API_TOKEN
+    arun(create_task(VideoRequest(prompt="可爱的小姐姐")))

meutils/apis/kling/images.py

@@ -0,0 +1,174 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : images
+# @Time         : 2024/9/24 10:14
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from meutils.io.files_utils import to_bytes
+
+from meutils.apis.kuaishou import klingai
+
+from meutils.schemas.kling_types import STATUSES, send_message
+from meutils.schemas.kling_types import ImageRequest, Task, TaskResponse
+from meutils.schemas.kuaishou_types import BASE_URL, KlingaiImageRequest
+from meutils.schemas.image_types import KlingImageRequest
+
+
+async def create_task(request: ImageRequest, vip: bool = False):
+    url = None
+    if request.image:
+        file = await to_bytes(request.image)
+        file_task = await klingai.upload(file)
+        url = file_task.url
+        # return url
+
+    logger.debug(request.model_dump_json(exclude_none=True, indent=4))
+
+    request = KlingaiImageRequest(url=url, **request.model_dump())
+    task_response = await klingai.create_task_plus(request, vip=vip)
+
+    return task_response
+
+    # {
+    #     "code": 0,
+    #     "message": "string",
+    #     "request_id": "string",
+    #     "data": {
+    #         "task_id": "string",
+    #         "task_status": "string",
+    #         "created_at": 0,
+    #         "updated_at": 0
+    #     }
+    # }
+
+
+async def get_task(task_id: str, token: str, oss: Optional[str] = None):
+    vip = "mini" not in task_id
+    task_type = "images" if 'image' in task_id else "videos"
+
+    result = await klingai.get_task_plus(task_id, token)
+
+    data = result['data']
+    message = result['message']
+    status_code = result['status']
+
+    task = None
+    try:
+        works = data['works']
+        urls = [work['resource']['resource'] for work in works]
+
+        items = [
+            {
+                "id": f"{uuid.uuid4()}",  # work['workId'],
+                "url": url,
+                "duration": f"{work['resource']['duration'] / 1000:.1f}",
+            } for work, url in zip(works, urls)
+        ]
+
+        task_result = {task_type: items}
+        task = Task(
+            task_id=str(data['task']['id']),
+            task_status=STATUSES.get(data['status'], data['status']),
+            task_status_msg=data['message'],
+            task_result=task_result
+        )
+
+        if data['status'] not in STATUSES:
+            send_message(f"未知状态：{data['status']}")
+            send_message(data)
+
+    except Exception as e:
+        logger.error(e)
+        send_message(f"task_id:{task_id} 失败\n{e}")
+        send_message(data)
+
+    response = TaskResponse(
+        code=status_code,
+        message=message,
+        data=task,
+    )
+    return response
+
+
+# {
+#     "code": 0,
+#     "message": "string",
+#     "request_id": "string",
+#     "data": {
+#         "task_id": "string",
+#         "task_status": "string",
+#         "task_status_msg": "string",
+#         "created_at": 1722769557708,
+#         "updated_at": 1722769557708,
+#         "task_result": {
+#             "videos": [
+#                 {
+#                     "id": "string",
+#                     "url": "string",
+#                     "duration": "string"
+#                 }
+#             ]
+#         }
+#     }
+# }
+
+
+async def generate(request: KlingImageRequest):
+    request = ImageRequest(**request.model_dump(exclude_none=True))
+    task = await create_task(request, vip=True)
+
+    for i in range(1, 15):
+        await asyncio.sleep(15 / i)
+        try:
+            data = await get_task(task.data.task_id, task.system_fingerprint)
+            images = data.data.task_result.get('images', [])
+            return {"data": images}
+
+        except Exception as e:
+            logger.debug(e)
+            continue
+
+
+if __name__ == '__main__':
+    request = ImageRequest(
+        prompt="多肉植物，带着水珠，潮玩盲盒风格，皮克斯，3D质感，温馨的环境，丰富的场景，最佳画质，超精细，Octane渲染",
+        aspect_ratio="1:1",
+        image_fidelity=0.5,
+        n=2,
+        negative_prompt="",
+    )
+
+
+    # task = arun(create_task(request, vip=True))
+    # task = arun(create_task(request, vip=False))
+
+    # image = "https://oss.ffire.cc/files/old.jpg"
+    # request = ImageRequest(
+    #     prompt="互相拥抱",
+    #     image=image
+    # )
+    # task = arun(create_task(request, vip=False))
+
+    # print(task.model_dump_json(exclude_none=True, indent=4))
+
+
+    request = KlingImageRequest(
+        prompt="多肉植物，带着水珠，潮玩盲盒风格，皮克斯，3D质感，温馨的环境，丰富的场景，最佳画质，超精细，Octane渲染",
+        aspect_ratio="1:1",
+        image_fidelity=0.5,
+        n=2,
+        negative_prompt="",
+    )
+    arun(generate(request))
+
+    # arun(get_task(task.data.task_id, task.system_fingerprint, oss='kling'))
+    # arun(get_task(task.data.task_id, task.system_fingerprint))
+
+    # arun(get_task("kling-image-pro-56994904", task.system_fingerprint, oss='kling'))
+
+    # arun(get_task(task.data.task_id, task.system_fingerprint, oss='kling'))

meutils/apis/kling/kolors_virtual_try_on.py

@@ -0,0 +1,111 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : virtual_try_on
+# @Time         : 2024/11/25 09:48
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+import jwt
+
+from meutils.pipe import *
+from meutils.schemas.task_types import TaskResponse
+from meutils.schemas.image_types import ImageRequest, ImagesResponse
+
+from meutils.schemas.kling_types import API_BASE_URL, TryOnRequest, TaskResponse
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.decorators.retry import retrying
+from meutils.str_utils.regular_expression import parse_url
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=cjTepf"
+
+
+def encode_jwt_token(token):
+    ak, sk = token.split('|')
+    headers = {
+        "alg": "HS256",
+        "typ": "JWT"
+    }
+    payload = {
+        "iss": ak,
+        "exp": int(time.time()) + 24 * 3600,  # 有效时间，此处示例代表当前时间+1800s(30min)
+        "nbf": int(time.time()) - 5  # 开始生效的时间，此处示例代表当前时间-5秒
+    }
+    token = jwt.encode(payload, sk, headers=headers)
+    logger.debug(token)
+    return token
+
+
+###################################################################### 重构
+
+######################################################################
+@alru_cache(ttl=3600)
+@retrying(title=__name__)
+async def create_task(request: TryOnRequest, token: Optional[str] = None):
+    token = token or await get_next_token_for_polling(FEISHU_URL)
+    token = encode_jwt_token(token)
+
+    headers = {
+        "Authorization": f"Bearer {token}"
+    }
+    payload = request.model_dump()
+    async with httpx.AsyncClient(base_url=API_BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post("/v1/images/kolors-virtual-try-on", json=payload)
+        response.raise_for_status()
+        data = response.json()
+        logger.debug(bjson(data))
+
+        return TaskResponse(**data, system_fingerprint=token)
+
+
+@retrying(title=__name__)  # 触发重试
+async def get_task(task_id, token):
+    headers = {
+        "Authorization": f"Bearer {token}"
+    }
+    async with httpx.AsyncClient(base_url=API_BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.get(f"/v1/images/kolors-virtual-try-on/{task_id}")
+        response.raise_for_status()
+        data = response.json()
+        return TaskResponse(**data)
+
+
+async def generate(request: ImageRequest):
+    human_image, cloth_image = parse_url(request.prompt)  # 两张图片
+
+    request = TryOnRequest(human_image=human_image, cloth_image=cloth_image)
+    task_response = await create_task(request)
+
+    for i in range(1, 10):
+        await asyncio.sleep(max(10 / i, 1))
+
+        response = await get_task(task_response.data.task_id, task_response.system_fingerprint)
+        if task_result := response.data.task_result:
+            images = task_result.get('images', [])
+            return ImagesResponse(data=images)
+
+
+if __name__ == '__main__':
+    # request = TryOnRequest()
+    # arun(create_task(request))
+    # {
+    #     "code": 0,
+    #     "message": "SUCCEED",
+    #     "request_id": "CjMVomdAMUcAAAAAAHcmvg",
+    #     "data": {
+    #         "task_id": "CjMVomdAMUcAAAAAAHcmvg",
+    #         "task_status": "submitted",
+    #         "created_at": 1732500869112,
+    #         "updated_at": 1732500869112
+    #     }
+    # }
+
+    # token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJkMmU1NmZkY2ZiODY0MmZmYWEyMjZlZWIxNzY4MzMyMiIsImV4cCI6MTczMzE4OTI4OSwibmJmIjoxNzMzMTAyODg0fQ.YjB792REbuFaUoyUXvd5bpI76_0PBWawDHDjPHOBA5Q'
+    # arun(get_task("CjiL9mdJhswAAAAAAG2C6g", token))
+
+    human_image: str = "https://oss.ffire.cc/files/try-on.png"
+    cloth_image: str = "https://oss.ffire.cc/files/x.png"
+
+    request = ImageRequest(prompt=f"{human_image}\n{cloth_image}")
+    arun(generate(request))

meutils/apis/kling/kolors_virtual_try_on_web.py

@@ -0,0 +1,126 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : virtual_try_on
+# @Time         : 2024/11/25 09:48
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+"""
+dressInput
+upperInput
+lowerInput
+{
+    "type": "mmu_img2img_aitryon",
+    "arguments": [
+        {
+            "name": "personType",
+            "value": "WOMAN"
+        },
+        {
+            "name": "__modelImageType",
+            "value": "OFFICIAL_MODEL_女模特4.png"
+        },
+        {
+            "name": "imageCount",
+            "value": "2"
+        },
+        {
+            "name": "biz",
+            "value": "klingai"
+        }
+    ],
+    "inputs": [
+        {
+            "name": "lowerInput",
+            "inputType": "URL",
+            "url": "https://h2.inkwai.com/bs2/upload-ylab-stunt/kling/tryon/model/v3/下装4.png"
+        },
+        {
+            "name": "humanImage",
+            "inputType": "URL",
+            "url": "https://p2.a.kwimgs.com/bs2/upload-ylab-stunt/kling/tryon/model/v3/女模特4.png"
+        }
+    ]
+}
+
+"""
+import jwt
+
+from meutils.pipe import *
+from meutils.schemas.task_types import TaskResponse  ############# 根据这个重构
+from meutils.schemas.kling_types import API_BASE_URL, TryOnRequest, TaskResponse
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.decorators.retry import retrying
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=cjTepf"
+
+
+def encode_jwt_token(token):
+    ak, sk = token.split('|')
+    headers = {
+        "alg": "HS256",
+        "typ": "JWT"
+    }
+    payload = {
+        "iss": ak,
+        "exp": int(time.time()) + 24 * 3600,  # 有效时间，此处示例代表当前时间+1800s(30min)
+        "nbf": int(time.time()) - 5  # 开始生效的时间，此处示例代表当前时间-5秒
+    }
+    token = jwt.encode(payload, sk, headers=headers)
+    logger.debug(token)
+    return token
+
+
+###################################################################### 重构
+
+######################################################################
+@alru_cache(ttl=3600)
+@retrying(title=__name__)
+async def create_task(request: TryOnRequest, token: Optional[str] = None):  ###### 所有的 TaskResponse标准化
+    token = token or await get_next_token_for_polling(FEISHU_URL)
+    token = encode_jwt_token(token)
+
+    headers = {
+        "Authorization": f"Bearer {token}"
+    }
+    payload = request.model_dump()
+    async with httpx.AsyncClient(base_url=API_BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post("/v1/images/kolors-virtual-try-on", json=payload)
+        response.raise_for_status()
+        data = response.json()
+        logger.debug(bjson(data))
+
+        return TaskResponse(**data, system_fingerprint=token)
+
+
+@retrying(title=__name__)  # 触发重试
+async def get_task(task_id, token):
+    headers = {
+        "Authorization": f"Bearer {token}"
+    }
+    async with httpx.AsyncClient(base_url=API_BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.get(f"/v1/images/kolors-virtual-try-on/{task_id}")
+        response.raise_for_status()
+        data = response.json()
+        return TaskResponse(**data)
+
+
+if __name__ == '__main__':
+    request = TryOnRequest()
+    # arun(create_task(request))
+    # {
+    #     "code": 0,
+    #     "message": "SUCCEED",
+    #     "request_id": "CjMVomdAMUcAAAAAAHcmvg",
+    #     "data": {
+    #         "task_id": "CjMVomdAMUcAAAAAAHcmvg",
+    #         "task_status": "submitted",
+    #         "created_at": 1732500869112,
+    #         "updated_at": 1732500869112
+    #     }
+    # }
+
+    token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJkMmU1NmZkY2ZiODY0MmZmYWEyMjZlZWIxNzY4MzMyMiIsImV4cCI6MTczMzE4OTI4OSwibmJmIjoxNzMzMTAyODg0fQ.YjB792REbuFaUoyUXvd5bpI76_0PBWawDHDjPHOBA5Q'
+    arun(get_task("CjiL9mdJhswAAAAAAG2C6g", token))

meutils/apis/kling/videos.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : videos
+# @Time         : 2024/9/24 10:14
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+
+from meutils.pipe import *
+from meutils.io.files_utils import to_bytes
+from meutils.apis.kuaishou import klingai, klingai_video
+from meutils.apis.kling.images import get_task
+
+from meutils.schemas.kling_types import STATUSES, send_message
+from meutils.schemas.kling_types import VideoRequest, Task, TaskResponse
+from meutils.schemas.kuaishou_types import KlingaiVideoRequest, Camera
+
+
+async def create_task(request: VideoRequest, vip: bool = False):
+    url = tail_image_url = None
+
+    if request.image:
+        file = await to_bytes(request.image)
+        file_task = await klingai.upload(file)
+        url = file_task.url
+
+    if request.image_tail:
+        file = await to_bytes(request.image)
+        file_task = await klingai.upload(file)
+        tail_image_url = file_task.url
+
+    camera_control = Camera(type=request.camera_control.type, **request.camera_control.config.model_dump())
+
+    request = KlingaiVideoRequest(
+        url=url,
+        tail_image_url=tail_image_url,
+        camera=camera_control,
+        **request.model_dump(exclude_none=True)
+    )
+
+    logger.debug(request.model_dump_json(exclude_none=True, indent=4))
+
+    task_response = await klingai_video.create_task_plus(request, vip=vip)
+    return task_response
+
+
+if __name__ == '__main__':
+    # 文生图
+    request = VideoRequest(
+        model="kling-v1",
+        prompt="多肉植物，带着水珠，潮玩盲盒风格，皮克斯，3D质感，温馨的环境，丰富的场景，最佳画质，超精细，Octane渲染",
+    )
+    # task = arun(create_task(request, vip=True))
+
+    image = "https://oss.ffire.cc/files/old.jpg"
+    request = VideoRequest(
+        prompt="互相拥抱",
+        image=image
+    )
+    task = arun(create_task(request, vip=True))
+
+    print(task.model_dump_json(exclude_none=True, indent=4))
+
+    arun(get_task(task.data.task_id, task.system_fingerprint))

meutils/apis/kling//351/211/264/346/235/203.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : 鉴权
+# @Time         : 2024/11/25 10:07
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+import time
+import jwt
+
+ak = ""  # 填写access key
+sk = ""  # 填写secret key
+
+
+def encode_jwt_token(ak, sk):
+    headers = {
+        "alg": "HS256",
+        "typ": "JWT"
+    }
+    payload = {
+        "iss": ak,
+        "exp": int(time.time()) + 1800,  # 有效时间，此处示例代表当前时间+1800s(30min)
+        "nbf": int(time.time()) - 5  # 开始生效的时间，此处示例代表当前时间-5秒
+    }
+    token = jwt.encode(payload, sk, headers=headers)
+    return token
+
+
+api_token = encode_jwt_token(ak, sk)
+print(api_token)  # 打印生成的API_TOKEN

meutils/apis/kuaidi.py

@@ -0,0 +1,32 @@
+# #!/usr/bin/env python
+# # -*- coding: utf-8 -*-
+# # @Project      : AI.  @by PyCharm
+# # @File         : kuaidi
+# # @Time         : 2024/4/24 16:58
+# # @Author       : betterme
+# # @WeChat       : meutils
+# # @Software     : PyCharm
+# # @Description  :
+#
+
+from meutils.pipe import *
+from meutils.decorators.retry import retrying
+from meutils.api.ali_apis import express_query
+
+
+@retrying
+def query(express_no):
+    appcode = '0ccd86184de94ca19c37cbb215b1f372'
+    return express_query(express_no, appcode=appcode).get('result')
+
+
+path = './快递登记系统.xlsx'
+sheet_name = '快递信息'
+df = pd.read_excel(path, sheet_name=sheet_name)
+express_numbers = df['单号'] + ':' + df['收件人电话'].astype(str).str.strip().str[-4:]
+df_ = pd.DataFrame(express_numbers.tolist() | xThreadPoolExecutor(query))
+# df = pd.concat([df, df_[["updateTime", "takeTime"]]], axis=1)
+df = pd.concat([df, df_], axis=1)
+
+with pd.ExcelWriter('单号信息.xlsm') as writer:
+    df.to_excel(writer, sheet_name)

meutils/apis/kuaishou/__init__.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/7/9 13:23
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+