MeUtils 2024.3.18.15.56.48__py3-none-any.whl → 2025.1.17.17.55.21__py3-none-any.whl

meutils/apis/hf//350/257/201/344/273/266/347/205/247.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kolors
+# @Time         : 2024/7/25 08:42
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : https://huggingface.co/spaces/gokaygokay/KolorsPlusPlus
+# https://huggingface.co/spaces/Kwai-Kolors/Kolors-Virtual-Try-On
+
+from meutils.pipe import *
+from meutils.schemas.openai_types import ImageRequest, ImagesResponse
+from meutils.async_utils import sync_to_async
+from meutils.decorators.retry import retrying
+
+from gradio_client import Client as _Client, handle_file
+from meutils.io.image import image_to_base64
+
+from gradio_client import Client
+
+# client = Client("https://cunyue-person-modnet.demo.swanhub.co/")
+#
+#
+#
+# url = "https://cunyue-person-modnet.demo.swanhub.co/file=/app/image/image04.jpg"
+# result = client.predict(
+#     data=[image_to_base64(url)],
+#     # str (filepath or URL to image) in 'Image' Image component
+#     api_name="/predict"
+# )
+# print(result)
+
+from gradio_client import Client
+
+client = Client("https://s5k.cn/api/v1/studio/binzhouchn/try-on-cloth/gradio/", hf_token="a22224b9-b276-45b5-b8d1-10f3c81c8eec")
+result = client.predict(
+		name="Hello!!",
+		api_name="/predict"
+)
+print(result)

meutils/apis/hunyuan/__init__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/10/11 14:53
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *

meutils/apis/hunyuan/image_tools.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : image
+# @Time         : 2024/10/11 15:01
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import json_repair
+
+from meutils.pipe import *
+from meutils.decorators.retry import retrying, IgnoredRetryException
+from meutils.schemas.yuanbao_types import FEISHU_URL, YUANBAO_BASE_URL
+from meutils.schemas.image_types import HunyuanImageProcessRequest
+
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.io.files_utils import to_url
+from meutils.notice.feishu import send_message
+
+
+@retrying(min=3, ignored_exception_types=(IgnoredRetryException,))
+async def image_process(request: HunyuanImageProcessRequest, token: Optional[str] = None):
+    token = token or await get_next_token_for_polling(FEISHU_URL)
+
+    payload = {
+        "imageUrl": request.image if request.image.startswith('http') else await to_url(request.image),
+    }
+    if request.task == 'style':
+        payload.update({
+            "style": request.style,
+            "prompt": f"转换为{request.style}",
+        })
+
+    headers = {
+        'cookie': token
+    }
+    async with httpx.AsyncClient(base_url=YUANBAO_BASE_URL, headers=headers, timeout=100) as client:
+        response = await client.post(f"/api/image/{request.task}", json=payload)
+        response.raise_for_status()
+        logger.debug(response.text)
+
+        # 新版本的位置可能不一样 注意
+        # data = json_repair.repair_json(
+        #     response.text.replace(r'\u0026', '&').rsplit("data: [TRACEID", 1)[0],
+        #     return_objects=True
+        # )[-1]
+
+        skip_strings = ['DONE', 'TRACEID']
+        data = response.text.replace(r'\u0026', '&').splitlines() | xsse_parser(skip_strings=skip_strings)
+        data = data and data[-1]
+        logger.debug(data)
+
+        # todo: 错误处理
+        if isinstance(data, dict) and any(data["code"] == code for code in {"429"}):
+            Exception(f"重试: {response.text}")
+
+        elif isinstance(data, list) or any(i in response.text for i in {"当前图片没有检测到水印"}):  # 跳过重试并返回原始错误
+            raise IgnoredRetryException(f"忽略重试: \n{response.text}")
+
+        data = {
+            "data": [
+                {
+                    "url": data["imageUrl"],
+                    "imageUrl": data["imageUrl"],
+                    "thumbnailUrl": data["thumbnailUrl"],
+                }
+            ]
+        }
+
+        return data
+
+
+if __name__ == '__main__':
+    # request = ImageProcessRequest(image="https://oss.ffire.cc/files/kling_watermark.png", task='removewatermark')
+
+    with timer():
+        image = "https://sfile.chatglm.cn/chatglm4/3dcb1cc2-22ad-420b-9d16-dc71dffc02b2.png"
+        image = "https://oss.ffire.cc/files/kling_watermark.png"
+        image = "https://cdn.meimeiqushuiyin.cn/2024-12-05/ori/tmp_0511a4cb2066ffc309fa6f7a733ac1e93236709bf46c9430.jpg"
+        image = "https://cdn.meimeiqushuiyin.cn/2024-12-05/ori/tmp_de5e186878b079a87d22c561f17e6853.jpg"
+        request = HunyuanImageProcessRequest(image=image, task='removewatermark')
+
+        arun(image_process(request))

meutils/apis/images/__init__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/10/31 14:04
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *

meutils/apis/images/deepinfra.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : deepinfra
+# @Time         : 2024/10/31 14:04
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+from meutils.schemas.image_types import ImageRequest, ImagesResponse
+from meutils.notice.feishu import IMAGES, send_message as _send_message
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.apis.siliconflow.images import generate as siliconflow_generate
+
+BASE_URL = "https://api.deepinfra.com"
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=6lr4oi"
+
+DEFAULT_MODEL = "black-forest-labs/FLUX-1.1-pro"
+MODELS = {
+
+    "flux.1-schnell": "black-forest-labs/FLUX.1-schnell",
+    "flux.1-dev": "black-forest-labs/FLUX.1-dev",
+    "flux.1-pro": "black-forest-labs/FLUX.1-dev",
+    "flux.1-pro-max": "black-forest-labs/FLUX.1-dev",
+
+    "flux-schnell": "black-forest-labs/FLUX.1-schnell",
+    "flux-dev": "black-forest-labs/FLUX.1-dev",
+    "flux-pro": "black-forest-labs/FLUX.1-dev",
+    "flux-pro-max": "black-forest-labs/FLUX.1-dev",
+
+    "stable-diffusion-xl-base-1.0": "stabilityai/stable-diffusion-xl-base-1.0",  # 图生图
+    "stable-diffusion-2-1": "stabilityai/stable-diffusion-2-1",  # 图生图
+
+    "stable-diffusion": "stabilityai/stable-diffusion-3-medium",
+    "stable-diffusion-3-medium": "stabilityai/stable-diffusion-3-medium",
+    "stable-diffusion-3": "stabilityai/stable-diffusion-3-medium",
+
+    "stable-diffusion-3-5-large": "stabilityai/stable-diffusion-3-5-large"
+
+}
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url=IMAGES
+)
+
+
+async def generate(request: ImageRequest, token: Optional[str] = None):
+    try:
+        token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL)
+        headers = {"Authorization": f"Bearer {token}"}
+
+        request.model = DEFAULT_MODEL  # MODELS.get(request.model, DEFAULT_MODEL)
+
+        # Height of the generated image in pixels. Must be a multiple of 32 (Default: 1024, 256 ≤ height ≤ 1440)
+        w, h = map(int, request.size.split("x"))
+
+        max_size = max(w, h)
+        w, h = w * 1440 / max_size, h * 1440 / max_size
+        w, h = max(256, w), max(256, h)
+        request.size = f"{int(w)}x{int(h)}"
+
+        logger.debug(request)
+
+        payload = {
+            "prompt": request.prompt,
+            "prompt_upsampling": True,
+            "seed": request.seed,
+            "safety_tolerance": 6,
+            "width": w,
+            "height": h
+        }
+
+        async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+            response = await client.post(f"/v1/inference/{request.model}", json=payload)
+            response.raise_for_status()
+            data = response.json()
+
+            return ImagesResponse(image=data["image_url"], metadata=data)
+
+    except Exception as e:
+        logger.error(e)
+        send_message(f"生成失败：{e}, 使用siliconflow重试")
+        return await siliconflow_generate(request)
+
+
+if __name__ == '__main__':
+    arun(generate(ImageRequest(prompt="一只可爱的猫", size="1440x1440")))

meutils/apis/images/demo.py

@@ -0,0 +1,150 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : demo
+# @Time         : 2024/11/1 16:53
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+
+d1 = {
+    "type": "curated_styles",
+    "image_types": [
+        "any",
+        "digital_illustration",
+        "illustration_3d",
+        "digital_illustration_seamless",
+        "digital_illustration_pixel_art",
+        "digital_illustration_3d",
+        "digital_illustration_psychedelic",
+        "digital_illustration_hand_drawn",
+        "digital_illustration_grain",
+        "digital_illustration_glow",
+        "digital_illustration_80s",
+        "digital_illustration_watercolor",
+        "digital_illustration_voxel",
+        "digital_illustration_infantile_sketch",
+        "digital_illustration_2d_art_poster",
+        "digital_illustration_kawaii",
+        "digital_illustration_halloween_drawings",
+        "digital_illustration_2d_art_poster_2",
+        "digital_illustration_engraving_color",
+        "digital_illustration_flat_air_art",
+        "digital_illustration_hand_drawn_outline",
+        "digital_illustration_handmade_3d",
+        "digital_illustration_stickers_drawings"
+    ],
+    "transform_model": "recraftv3",
+    "offset": 0,
+    "limit": 30
+}
+
+d2 = {
+    "type": "curated_styles",
+    "image_types": [
+        "realistic_image",
+        "realistic_image_mockup",
+        "realistic_image_b_and_w",
+        "realistic_image_enterprise",
+        "realistic_image_hard_flash",
+        "realistic_image_hdr",
+        "realistic_image_natural_light",
+        "realistic_image_studio_portrait",
+        "realistic_image_motion_blur"
+    ],
+    "transform_model": "recraftv3",
+    "offset": 0,
+    "limit": 30
+}
+
+d3 = {
+    "type": "curated_styles",
+    "image_types": [
+        "any",
+        "digital_illustration",
+        "illustration_3d",
+        "digital_illustration_seamless",
+        "digital_illustration_pixel_art",
+        "digital_illustration_3d",
+        "digital_illustration_psychedelic",
+        "digital_illustration_hand_drawn",
+        "digital_illustration_grain",
+        "digital_illustration_glow",
+        "digital_illustration_80s",
+        "digital_illustration_watercolor",
+        "digital_illustration_voxel",
+        "digital_illustration_infantile_sketch",
+        "digital_illustration_2d_art_poster",
+        "digital_illustration_kawaii",
+        "digital_illustration_halloween_drawings",
+        "digital_illustration_2d_art_poster_2",
+        "digital_illustration_engraving_color",
+        "digital_illustration_flat_air_art",
+        "digital_illustration_hand_drawn_outline",
+        "digital_illustration_handmade_3d",
+        "digital_illustration_stickers_drawings"
+    ],
+    "transform_model": "recraftv3",
+    "offset": 0,
+    "limit": 30
+}
+
+d4 = {
+    "type": "curated_styles",
+    "image_types": [
+        "vector_illustration",
+        "vector_illustration_seamless",
+        "vector_illustration_line_art",
+        "vector_illustration_doodle_line_art",
+        "vector_illustration_flat_2",
+        "vector_illustration_70s",
+        "vector_illustration_cartoon",
+        "vector_illustration_kawaii",
+        "vector_illustration_linocut",
+        "vector_illustration_engraving",
+        "vector_illustration_halloween_stickers",
+        "vector_illustration_line_circuit"
+    ],
+    "transform_model": "recraftv3",
+    "offset": 0,
+    "limit": 30
+}
+
+
+# d1['image_types'] + d2['image_types']+ d3['image_types']+ d4['image_types']
+
+def min_max_normalize(data, feature_range=(0, 1)):
+    """
+    最大最小归一化函数,保持数据比例不变
+
+    参数:
+    data: 输入数据,可以是列表或numpy数组
+    feature_range: 目标范围元组,默认为(0,1)
+
+    返回:
+    归一化后的数据
+    """
+    import numpy as np
+
+    data = np.asarray(data)
+    min_val = np.min(data)
+    max_val = np.max(data)
+
+    # 防止除零
+    if max_val == min_val:
+        return np.zeros_like(data)
+
+    min_range, max_range = feature_range
+    range_scale = max_range - min_range
+    data_scale = max_val - min_val
+
+    # 保持比例的归一化
+    scale_factor = range_scale / data_scale
+    scaled_data = data * scale_factor + (min_range - min_val * scale_factor)
+
+    return scaled_data
+
+min_max_normalize([1000, 3000], feature_range=(1024, 1024))

meutils/apis/images/edits.py

@@ -0,0 +1,146 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : image_tools
+# @Time         : 2024/8/28 13:17
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.schemas.image_types import ImageRequest, ImagesResponse
+from meutils.schemas.image_types import ImageProcess
+
+from meutils.io.files_utils import to_bytes, to_base64, to_url_fal
+
+from meutils.notice.feishu import send_message as _send_message
+
+BASE_URL = "https://image.baidu.com"
+
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/GYCHsvI4qhnDPNtI4VPcdw2knEd?sheet=jrWhAS"
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url="https://open.feishu.cn/open-apis/bot/v2/hook/dc1eda96-348e-4cb5-9c7c-2d87d584ca18"
+)
+
+
+async def make_request_for_gitee(payload, token: Optional[str] = None, response_format: str = "url"):
+    s = time.time()
+    feishu_url = "https://xchatllm.feishu.cn/sheets/MekfsfVuohfUf1tsWV0cCvTmn3c?sheet=PDnO6X"
+    token = token or await get_next_token_for_polling(feishu_url)
+
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "X-Failover-Enabled": "true",
+        "X-Package": "1910"
+    }
+
+    files = {
+        "image": ("_.png", payload.pop('image'))
+    }
+    base_url = "https://ai.gitee.com/v1"
+    async with httpx.AsyncClient(base_url=base_url, headers=headers, timeout=60) as client:
+        response = await client.post("/images/mattings", data=payload, files=files)
+        response.raise_for_status()
+        response = ImagesResponse(**response.json())
+        if response_format == "url":
+            url = await to_url_fal(response.data[0].b64_json, content_type="image/png")
+            response.data[0].url = url
+            response.data[0].b64_json = None
+            response.timings = {"inference": time.time() - s}
+
+        return response
+
+
+async def make_request_for_baidu(payload, token: Optional[str] = None, response_format: str = "url"):
+    s = time.time()
+    token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL, from_redis=True)
+    headers = {
+        'Cookie': token,
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36",
+        'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8'
+    }
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=60) as client:
+        response = await client.post("/aigc/pccreate", data=payload)  # pcEditTaskid
+        response.raise_for_status()
+        data = response.json()
+
+        logger.debug(data)
+
+        image_base64 = None
+        if task_id := data.get("pcEditTaskid"):
+            for i in range(30):
+                await asyncio.sleep(3)
+                try:
+                    response = await client.get(f'/aigc/pcquery?taskId={task_id}')
+                    if data := response.json().get("picArr", []):
+                        image_base64 = data[0].get("src")
+                        break
+                except Exception as e:
+                    logger.error(e)
+                    if i > 3: break
+
+        if response_format == "url":
+            url = await to_url_fal(image_base64, content_type="image/png")
+            return ImagesResponse(data=[{"url": url}], timings={"inference": time.time() - s})
+        else:
+            return ImagesResponse(data=[{"b64_json": image_base64}], timings={"inference": time.time() - s})
+
+
+async def edit_image(request: ImageProcess):
+    image, mask = await asyncio.gather(to_base64(request.image), to_base64(request.mask))
+    payload = {
+        "type": "1",  # 去水印
+
+        "picInfo": image,
+        "picInfo2": mask,
+        # "original_url": "", # 更快但是会有错误
+    }
+
+    if request.model == "remove-watermark":
+        if mask:
+            payload['type'] = "2"
+        return await make_request_for_baidu(payload)
+
+    elif request.model == "clarity":
+        payload['type'] = "3"
+        return await make_request_for_baidu(payload)
+
+    elif request.model == "expand":
+        payload['type'] = "4"
+        payload['ext_ratio'] = request.aspect_ratio
+        return await make_request_for_baidu(payload)
+    ################################################################################################
+
+    elif request.model == "rmbg-2.0":
+        payload = {
+            "model": request.model,
+            "image": await to_bytes(image),
+        }
+        return await make_request_for_gitee(payload, response_format=request.response_format)
+
+
+if __name__ == '__main__':
+    token = "BAIDUID=FF8BB4BF861992E2BF4A585A37366236:FG=1; BAIDUID_BFESS=FF8BB4BF861992E2BF4A585A37366236:FG=1; BIDUPSID=FF8BB4BF861992E2BF4A585A37366236; BDRCVFR[dG2JNJb_ajR]=mk3SLVN4HKm; userFrom=null; ab_sr=1.0.1_NjY5OWZiZDg5YTJmYTQzNWUyNzU1YjBmN2FlMDFiNjMyOTVhMDE3ZWVlYWY5N2Y2MTg4NGI1MzRmMmVjMjQyZjlhZTU2MmM1NDRlMmU4YzgwMzRiMjUyYTc4ZjY1OTcxZTE4OTA4YTlmMWIwZWUzNTdiMzlhZTRiM2IzYTQ0MjgyMzc2MjQwMGRlYzZlZDhjOTg5Yzg4NWVjMTNiZmVmZQ==; BDRCVFR[-pGxjrCMryR]=mk3SLVN4HKm; H_WISE_SIDS=60273_60360_60623_60664_60678_60684_60700"
+
+    url = "https://oss.ffire.cc/files/kling_watermark.png"
+    # url = "https://s22-def.ap4r.com/bs2/upload-ylab-stunt-sgp/se/ai_portal_sgp_queue_mmu_txt2img_aiweb/9c520b80-efc2-4321-8f0e-f1d34d483ddd/1.png"
+
+    request = ImageProcess(
+        model="remove-watermark",
+        # model="clarity",
+        # model="expand",
+        # model="rmbg-2.0",
+
+        image=url,
+        # mask=url,
+
+        # response_format="b64_json"
+    )
+    arun(edit_image(request))
+
+    # arun(image_edit(request))

meutils/apis/images/eidt.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : eidt
+# @Time         : 2025/1/7 13:27
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+
+
+from meutils.pipe import *
+
+from openai import OpenAI
+
+client = OpenAI(
+    base_url="https://ai.gitee.com/v1",
+    api_key="WPCSA3ZYD8KBQQ2ZKTAPVUA059J2Q47TLWGB2ZMQ",
+    default_headers={"X-Failover-Enabled": "true", "X-Package": "1910"},
+)
+
+response = client.images.edit(
+    model="Kolors",
+    prompt="笑起来",
+    size="1024x1024",
+    extra_body={
+        "steps": 25,
+        "guidance_scale": 6,
+    },
+
+    mask=open("test.png", "rb"),
+
+    image=open("test.png", "rb"),
+)

meutils/apis/images/flux/__init__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/8/21 13:29
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *