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

meutils/apis/audio/fish.py

@@ -0,0 +1,248 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2024/7/5 12:05
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : todo: # 雷军 "738d0cc1a3e9430a9de2b544a466a7fc" 增加模型映射表
+# https://docs.fish.audio/api-reference/endpoint/openapi-v1/speech-to-text
+# https://docs.fish.audio/text-to-speech/create-model
+import shortuuid
+# https://api.fish.audio/文档
+
+from fish_audio_sdk import Session, TTSRequest, ASRRequest
+
+from meutils.pipe import *
+from meutils.io.files_utils import to_url
+
+from meutils.config_utils.lark_utils import get_next_token_for_polling
+from meutils.notice.feishu import send_message as _send_message
+
+BASE_URL = "https://api.fish.audio"
+FEISHU_URL = "https://xchatllm.feishu.cn/sheets/ZUyLsVyjMhtexRt3IB9ci9fUnWc?sheet=7ce4e3"
+FEISHU_URL_VIP = "https://xchatllm.feishu.cn/sheets/ZUyLsVyjMhtexRt3IB9ci9fUnWc?sheet=QLTif4"
+FEISHU_URL_API = "https://xchatllm.feishu.cn/sheets/ZUyLsVyjMhtexRt3IB9ci9fUnWc?sheet=v5C5xa"
+
+send_message = partial(
+    _send_message,
+    title=__name__,
+    url="https://open.feishu.cn/open-apis/bot/v2/hook/f0475882-ad39-49d9-ad77-523c0e768e96"
+)
+
+COVER_IMAGE = get_resolve_path('../../data/cowboy-hat-face.webp', __file__).read_bytes()
+
+
+async def check_token(token: str, for_api: bool = False):
+    path = "api-credit" if for_api else "package"
+    try:
+        headers = {
+            "Authorization": f"Bearer {token}"
+        }
+        async with httpx.AsyncClient(base_url=BASE_URL, headers=headers) as client:
+            response = await client.get(f"/wallet/self/{path}")
+            response.raise_for_status()
+            data = response.json()
+
+            # logger.debug(data)
+
+            credit = data.get('credit') or data.get('balance', 0)
+            logger.info(credit)
+
+            return float(credit) > 0
+    except Exception as e:
+        logger.error(e)
+        return False
+
+
+async def get_model_list(token, **kwargs):  # 上传的音色
+    params = {
+        'self': True,
+        'title': None,
+
+        'page_size': 10,
+        'page_number': 1,
+        **kwargs
+    }
+    headers = {
+        "authorization": f"Bearer {token}"
+    }
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers) as client:
+        response = await client.get("/model", params=params)
+        return response.is_success and response.json()
+
+
+async def create_tts_model(title, voices: List[bytes], texts: Optional[List[str]] = None):
+    session = Session(await get_next_token_for_polling(feishu_url=FEISHU_URL_API))  # 逆向token也兼容
+
+    model = session.create_model(
+        title=title,
+        visibility="public",
+
+        texts=texts,
+        voices=voices,
+
+        cover_image=COVER_IMAGE,
+    )
+    return model
+
+
+async def create_tts(
+        request: TTSRequest,
+        token: Optional[str] = None,
+        response_format: Optional[Literal["url",]] = None
+):
+    if len(request.text.encode()) < 500:
+        token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL)
+    else:
+        token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL_VIP)
+
+    headers = {
+        "Authorization": f"Bearer {token}",
+    }
+
+    payload = {
+        "type": "tts",
+        "channel": "free",
+        "stream": True,
+        "model": request.reference_id or "",  # 雷军 "738d0cc1a3e9430a9de2b544a466a7fc", # todo: 支持下 btyes
+        "parameters": {
+            "text": request.text
+        },
+        "format": request.format,
+    }
+    try:
+        async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=300) as client:
+            response = await client.post("/task", json=payload)
+            response.raise_for_status()
+
+            if response_format == "url":
+                # logger.debug(response.headers)
+
+                task_id = response.headers.get("task-id")
+                task_response = await client.get(f"/task/{task_id}")
+                data = task_response.json()
+                data["result"] = await to_url(data["result"], filename=f'{shortuuid.random()}.mp3')  # 转存 翻墙的
+                return data
+            return response.content  # bytes
+
+    except Exception as e:
+        logger.error(e)
+        session = Session(await get_next_token_for_polling(feishu_url=FEISHU_URL_API))
+        files = session.tts(TTSRequest(**request.model_dump()))
+
+        if response_format == "url":
+            url = await to_url(b"".join(files), filename=f'{shortuuid.random()}.mp3')
+
+            return {
+                "_id": shortuuid.random(),
+                "state": "finished",
+                "type": "tts",
+                "model": request.reference_id,
+                "parameters": {
+                    "text": request.text
+                },
+                "created_at": datetime.datetime.now().isoformat(),
+                "deleted": False,
+                "channel": "premium",
+                "finished_at": datetime.datetime.now().isoformat(),
+                "result": url,
+                "generate_error": False
+            }
+
+
+# async def create_tts_model(token, voices: list, **kwargs):
+#     """
+#     {'_id': '9d10cdbea3954aa9b8fd992fd24b92a7',
+#      'author': {'_id': 'd71d7c63c52e4d70be72e3afdb7952ab',
+#                 'avatar': '',
+#                 'nickname': '313303303'},
+#      'cover_image': 'coverimage/9d10cdbea3954aa9b8fd992fd24b92a7',
+#      'created_at': '2024-07-05T05:30:32.020433Z',
+#      'description': '',
+#      'languages': ['zh'],
+#      'like_count': 0,
+#      'liked': False,
+#      'mark_count': 0,
+#      'marked': False,
+#      'samples_text': [],
+#      'shared_count': 0,
+#      'state': 'trained',
+#      'tags': [],
+#      'task_count': 0,
+#      'title': 'chatfire',
+#      'train_mode': 'fast',
+#      'type': 'tts',
+#      'updated_at': '2024-07-05T05:30:32.020407Z',
+#      'visibility': 'public'}
+#     """
+#     payload = {
+#         **kwargs
+#     }
+#
+#     files = [
+#         ("cover_image", get_resolve_path('cowboy-hat-face.webp', __file__).read_bytes())
+#     ]
+#
+#     for voice in voices:
+#         files.append(("voices", voice))  # bytes
+#
+#     data = [
+#         ("visibility", "private"),
+#         ("type", "tts"),
+#         ("title", "Demo"),
+#         ("train_mode", "fast"),
+#         # Enhance audio quality will remove background noise
+#         ("enhance_audio_quality", "true"),
+#         # Texts are optional, but if you provide them, they must match the number of audio samples
+#         ("texts", "text1"),
+#         ("texts", "text2"),
+#     ],
+#
+#     files = {
+#         'title': (None, 'chatfire-tts'),
+#         'description': (None, ''),
+#         'type': (None, 'tts'),
+#         'train_mode': (None, 'fast'),
+#         'visibility': (None, 'public'),  # private
+#
+#         'voices': file,  # ('audio_name.mp3', file)
+#         'cover_image': open(get_resolve_path('cowboy-hat-face.webp', __file__), 'rb')
+#
+#     }
+#
+#     headers = {
+#         "authorization": f"Bearer {token}"
+#     }
+#     async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=100) as client:
+#         response = await client.post("/model", json=payload, files=files)
+#         # logger.debug(response.text)
+#         if response.is_success:
+#             _ = response.json()
+#             _['model_id'] = _['_id']
+#             return _
+#         else:
+#             return response.text
+
+
+if __name__ == '__main__':
+    token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoiZDcxZDdjNjNjNTJlNGQ3MGJlNzJlM2FmZGI3OTUyYWIiLCJleHAiOjE3MzAyODYwNjguNTMyMTA5NX0.GNeW2WLlBqyP6evNQFkXpqa0UYOg50_IwNO48DuJc04"
+    # print(bjson(arun(get_model_list(token))))
+    # file = open('/Users/betterme/Downloads/whisper-1719913495729-54f08dde5.wav.mp3.mp3', 'rb')
+
+    # arun(create_tts_model(title="chatfire-tts", voices=[Path("x.mp3").read_bytes()]))
+
+    # model="cf8ea9c2c9e947a6b9d0fccec68c9dbd"
+
+    # token = "5cf445234491429c9855d3fa666d49f6"
+
+    # arun(check_token(token, for_api=True))
+
+    request = TTSRequest(text="语音克隆" * 20, reference_id="cf8ea9c2c9e947a6b9d0fccec68c9dbd")
+    arun(create_tts(request, response_format="url"))
+    # text = arun(create_tts(request))
+    #
+    # Path('x.mp3').write_bytes(text)
+
+    # print(get_resolve_path('../../data/cowboy-hat-face.webp', __file__))

meutils/apis/baidu/__init__.py

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

meutils/apis/baidu/bdaitpzs.py

@@ -0,0 +1,229 @@
+#!/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_spreadsheet_values, get_next_token_for_polling
+from meutils.schemas.openai_types import ImageRequest, ImagesResponse
+from meutils.schemas.image_types import BAIDU_TASKS, BaiduImageProcessRequest
+
+from meutils.decorators.retry import retrying
+from meutils.schemas.image_types import ASPECT_RATIOS
+from meutils.oss.minio_oss import Minio
+from meutils.decorators.contextmanagers import try_catcher
+from meutils.schemas.baidu_types import BDAITPZSRequest
+from meutils.schemas.task_types import Task, TaskType
+from meutils.io.image import image_to_base64, crop_polygon
+from meutils.io.files_utils import to_url, to_bytes
+
+from meutils.notice.feishu import send_message as _send_message
+
+BASE_URL = "https://image.baidu.com"
+
+url = "https://image.baidu.com/aigc/pccreate"
+
+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"
+)
+
+
+@retrying(max_retries=3, title=__name__)
+async def create_task(
+        request: BDAITPZSRequest,
+        token: Optional[str] = None,
+        is_async: bool = True,
+        response_format: Optional[Literal["url", "b64_json"]] = None
+):
+    token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL, from_redis=True)
+
+    payload = request.model_dump()
+
+    headers = {
+        # 'X-Requested-With': 'XMLHttpRequest',
+        'Cookie': token,
+        # 'User-Agent': 'Apifox/1.0.0 (https://apifox.com)',
+        'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8'
+    }
+
+    async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=100) as client:
+
+        response = await client.post('/aigc/pccreate', data=payload)
+
+        logger.debug(response.status_code)
+        # logger.debug(response.text)
+        # {"status":5,"message":"网络异常，请稍候再试~","isGenerate":true} 服务端url访问不了
+        # {'status': 0, 'resType': 2, 'timestamp': '1727233935', 'token': '5a5fc23b052b31d2120abab9c1d02178'} 没有水印
+
+        if response.is_success:
+            data = response.json()
+            if data.get('status') == 5:
+                logger.debug(f"original_url => picInfo: {data}")
+                request.picInfo = image_to_base64(request.original_url)  # 优先级最高
+                return await create_task(request, token=token, is_async=is_async, response_format=response_format)
+
+            if "pcEditTaskid" not in data:  # data.get('status') == 0
+                send_message(f"照片失败\n{request.original_url}\n{request.model_dump_json(indent=4)}")
+                raise Exception(f"无法处理该照片，可联系管理员\n{data}")
+
+            if is_async:
+                # {
+                #     "status": 0,
+                #     "pcEditTaskid": "cr7sbuue1pnfd61bjgu0",
+                #     "resType": 0,
+                #     "timestamp": "1724892667",
+                #     "token": "e668cb4457494351f65300d9f388bb2b"
+                # }
+
+                task_id = f"{TaskType.pcedit}-{data['pcEditTaskid']}"
+                return Task(id=task_id, data=data, system_fingerprint=token)
+
+            else:  # {'algoprocess': 0, 'isGenerate': False, 'progress': 4}
+                logger.debug(data)
+                task_id = data['pcEditTaskid']
+                # url = f"https://image.baidu.com/aigc/pcquery?taskId={task_id}"
+                # for i in range(10):
+                #     await asyncio.sleep(3)
+                #     with try_catcher():
+                #         response = await client.get(url)
+                #         data = response.json()
+                #         logger.debug(f"progress: {data['progress']}")
+                #
+                #         if data['isGenerate']:
+                #             return data
+                for i in range(1, 30):
+                    await asyncio.sleep(2 / i)
+                    with try_catcher():
+                        data = await get_task(task_id, response_format=response_format)
+                        logger.debug(f"progress: {data['progress']}")
+                        if data['isGenerate']:
+                            return data
+
+        response.raise_for_status()
+
+
+@retrying(max_retries=3, title=__name__)
+async def get_task(
+        task_id,
+        token: Optional[str] = None,
+        response_format: Optional[Literal["url", "b64_json"]] = None
+):
+    task_id = task_id.split("-", 1)[-1]
+
+    params = {
+        "taskId": task_id
+    }
+    async with httpx.AsyncClient(base_url=BASE_URL, timeout=30) as client:
+        response = await client.get('/aigc/pcquery', params=params)
+        data = response.json()
+
+        if response_format == "url":
+            for arr in data.get('picArr', []):
+                if base64_string := arr.pop('src', None):
+                    data['url'] = await to_url(base64_string)
+
+        return data
+
+
+TASKS = {
+    "matting": '9',
+    "removewatermark": '1'
+}
+
+
+async def image_process(request: BaiduImageProcessRequest):
+    _request = BDAITPZSRequest(type=TASKS.get(request.task), original_url=request.image, thumb_url=request.image)
+    data = await create_task(_request, is_async=False, response_format="url")
+
+    if request.task == "matting":
+        outline_points = []
+        inline_points = []
+
+        if "foreground" not in data:  # 未识别到主体： todo: mask
+            return {"data": [{"url": data['url']}]}
+
+        for outlines in data['foreground'][0].get('outlines', []):
+            _ = list(map(lambda point: (point['x'], point['y']), outlines['line']))
+            outline_points.append(_)
+
+        for inlines in data['foreground'][0].get('inlines', []):
+            _ = list(map(lambda point: (point['x'], point['y']), inlines['line']))
+            inline_points.append(_)
+
+        image = await to_bytes(request.image)
+        result = crop_polygon(image, outline_points, inline_points)
+
+        # 将 PIL Image 转换为字节
+        buffer = io.BytesIO()
+        result.save(buffer, format="PNG")
+        byte_data = buffer.getvalue()
+        url = await to_url(byte_data)
+
+        return {"data": [{"url": url}]}
+    return {"data": [{"url": data['url']}]}
+
+
+if __name__ == '__main__':
+    url = "https://oss.ffire.cc/files/kling_watermark.png"
+    # url = "https://env-00jxgna201cb.normal.cloudstatic.cn/ori/tmp_dc12fc648ab10c4b8d310f3e8645781278e556a0264836d7fdb806c6bb83c493.jpeg"
+
+    # 涂抹消除
+    # url = "https://env-00jxgna201cb.normal.cloudstatic.cn/ori/tmp_dc12fc648ab10c4b8d310f3e8645781278e556a0264836d7fdb806c6bb83c493.jpeg"
+    # url_ = "https://env-00jxgna201cb.normal.cloudstatic.cn/water/tmp_2954a8d96011173a7f2b6baad8cc28317278e368d1393ca6.jpg"
+    # picInfo2 = image_to_base64(url_)
+    #
+    url = "https://cdn.meimeiqushuiyin.cn/imgImprove/tmp_f5648a3c7cc8973b483ee3bb54d23aea.jpg"  # 网络异常
+    url = "https://s22-def.ap4r.com/bs2/upload-ylab-stunt-sgp/se/ai_portal_sgp_queue_mmu_txt2img_aiweb/d08b9526-d641-4e04-83d2-27d48e29fa8c/1.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 = BDAITPZSRequest(
+        # picInfo=image_to_base64(url),
+        original_url=url,
+        thumb_url=url
+    )
+
+    # from meutils.io.image import image2nowatermark_image
+    #
+    # api_keys = get_spreadsheet_values(feishu_url=FEISHU_URL, to_dataframe=True)[0]
+    # for token in api_keys:
+    #     if token:
+    #         # request = BDAITPZSRequest(original_url=url, thumb_url=url)
+    #         # task = arun(create_task(request, token=token, is_async=False))
+    #         # print(task)
+    #         arun(image2nowatermark_image(url))
+    #         # arun(image2nowatermark_image(url, token=token))
+
+    #
+    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"
+    # # token='BAIDUID=F2185337B2F3F85DAB5D2E661BC00C8D:FG=1; BDUSS=hNQkdpbjBOT05EVjRCSFp3TUF-TXFKWHktcFppUGNyd0xZZGp3Mkp6TmVnbFptSVFBQUFBJCQAAAAAAAAAAAEAAACsbZVAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAF71LmZe9S5mOX; BDUSS_BFESS=hNQkdpbjBOT05EVjRCSFp3TUF-TXFKWHktcFppUGNyd0xZZGp3Mkp6TmVnbFptSVFBQUFBJCQAAAAAAAAAAAEAAACsbZVAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAF71LmZe9S5mOX; MAWEBCUID=web_axHhqlIWargVQsXuskqOnMzVXDWQHqQXOpFPkPVSvXFIwTNYaC; PSTM=1716434076; BIDUPSID=5FEB509ECAE0F7843030FD4ECC0800F8; newlogin=1; MCITY=-315%3A; H_WISE_SIDS_BFESS=60360_60674; indexPageSugList=%5B%22%E8%AF%81%E4%BB%B6%E7%85%A7%20%E7%94%B7%22%2C%22%E6%85%88%E7%A6%A7%E5%92%8C%E6%9D%8E%E8%8E%B2%E8%8B%B1%22%2C%22%E7%AE%80%E9%A1%BF%E7%81%AF%E5%85%B7%20%E9%81%A5%E6%8E%A7%E5%99%A8%22%2C%22%E5%88%B0%E5%BD%93%E9%81%93%E5%A3%AB%E9%82%A3%E4%BA%9B%E5%B9%B4%20%E8%B1%86%E7%93%A3%22%2C%22%E9%81%93%E5%A3%AB%20%E5%8D%A1%E9%80%9A%22%2C%22%E9%81%93%E5%A3%AB%22%2C%22%E6%88%91%E5%BD%93%E9%81%93%E5%A3%AB%E9%82%A3%E4%BA%9B%E5%B9%B4%22%2C%22%E8%AF%BB%E6%9E%B6%22%2C%22%E6%9D%B0%E8%AF%BA%E6%AF%94%E5%88%A9%20%E5%AF%B9%E6%AF%94%E7%85%A7%20%E5%8D%95%E8%86%9D%E8%B7%AA%E5%9C%B0%22%5D; BAIDUID_BFESS=F2185337B2F3F85DAB5D2E661BC00C8D:FG=1; BDRCVFR[C0p6oIjvx-c]=I67x6TjHwwYf0; delPer=0; PSINO=5; BDRCVFR[C0sZzZJZb70]=mk3SLVN4HKm; BA_HECTOR=alal0125ah0h0h0l8k8l0l01blu4191jd049m1u; ZFY=C8vEYu7ZbY0zTEITG3A0aWCevri4:BT:B0H1NpNZdQWKM:C; BDORZ=B490B5EBF6F3CD402E515D22BCDA1598; H_PS_PSSID=60360_60674_60683_60694_60572; BCLID=11757829139010941756; BCLID_BFESS=11757829139010941756; BDSFRCVID=pZCOJexroG3bGRntFj9auQWaJgKKv3JTDYLE8yu9T4VNR5DVYtg8EG0Pt_ZYMak-XolpogKKL2OTHm_F_2uxOjjg8UtVJeC6EG0Ptf8g0M5; BDSFRCVID_BFESS=pZCOJexroG3bGRntFj9auQWaJgKKv3JTDYLE8yu9T4VNR5DVYtg8EG0Pt_ZYMak-XolpogKKL2OTHm_F_2uxOjjg8UtVJeC6EG0Ptf8g0M5; H_BDCLCKID_SF=tJkH_KthJIP3e4515Pr_q4tJbq58etJXfaO4Vp7F54nKDp0R0JbI0J0h5p6vaP3m066MaM-ELDOxsMT4QfJJhxuH-qoKel5tQH5MXprN3KJmf-Op2hO0ytDNbHOE2-biW2tH2Mbdax7P_IoG2Mn8M4bb3qOpBtQmJeTxoUJ25DnJhhCGe6KbD6bQDH8Oq-jeHDrKBRbaHJOoDDv8QxRcy4LbKxnxJhoe0H6JXt3LBJnfEpjlbURxqqjy3-OkbfQ9babTQ-tbBp3k8MQXbn7FQfbQ0hOe5RvnMe5a2JK2MJ7JOpvsDxnxy-umQRPH-Rv92DQMVU52QqcqEIQHQT3m5-5bbN3ut6IHJbufoID2fIvDqTrP-trf5DCShUFs5J3dB2Q-XPoO3KOGqpoh5b6b2xKJb-bCb4biQPjNoMbgy4op8P3y0bb2DUA1y4vpX4bOBeTxoUJ2-bjWS-cV-xTqjbIebPRiWTj9QgbLMhQ7tt5W8ncFbT7l5hKpbt-q0x-jLTnhVn0MBCK0MD8xDT8Bj6Pq5fcj2nQXK4_tVTrJabC3qMcJXU6q2bDeQNba0hjXbgcBVRjaLx3HOR3oynD5yl0vWtv4qnctfGPJ-qr4Jlb4otbTeMonDh83Bn6kW4oWHGbyWPJO5hvv8KoO3M7VBUKmDloOW-TB5bbPLUQF5l8-sq0x0bOte-bQXH_E5bj2qRAHoK-h3j; H_BDCLCKID_SF_BFESS=tJkH_KthJIP3e4515Pr_q4tJbq58etJXfaO4Vp7F54nKDp0R0JbI0J0h5p6vaP3m066MaM-ELDOxsMT4QfJJhxuH-qoKel5tQH5MXprN3KJmf-Op2hO0ytDNbHOE2-biW2tH2Mbdax7P_IoG2Mn8M4bb3qOpBtQmJeTxoUJ25DnJhhCGe6KbD6bQDH8Oq-jeHDrKBRbaHJOoDDv8QxRcy4LbKxnxJhoe0H6JXt3LBJnfEpjlbURxqqjy3-OkbfQ9babTQ-tbBp3k8MQXbn7FQfbQ0hOe5RvnMe5a2JK2MJ7JOpvsDxnxy-umQRPH-Rv92DQMVU52QqcqEIQHQT3m5-5bbN3ut6IHJbufoID2fIvDqTrP-trf5DCShUFs5J3dB2Q-XPoO3KOGqpoh5b6b2xKJb-bCb4biQPjNoMbgy4op8P3y0bb2DUA1y4vpX4bOBeTxoUJ2-bjWS-cV-xTqjbIebPRiWTj9QgbLMhQ7tt5W8ncFbT7l5hKpbt-q0x-jLTnhVn0MBCK0MD8xDT8Bj6Pq5fcj2nQXK4_tVTrJabC3qMcJXU6q2bDeQNba0hjXbgcBVRjaLx3HOR3oynD5yl0vWtv4qnctfGPJ-qr4Jlb4otbTeMonDh83Bn6kW4oWHGbyWPJO5hvv8KoO3M7VBUKmDloOW-TB5bbPLUQF5l8-sq0x0bOte-bQXH_E5bj2qRAHoK-h3j; BDRCVFR[dG2JNJb_ajR]=mk3SLVN4HKm; userFrom=null; BDRCVFR[-pGxjrCMryR]=mk3SLVN4HKm; ab_sr=1.0.1_NTY4Y2RjNTU5MWYzM2IyZjY4M2EzMWNjYWM1MWRhYjdlZDhmZTdkZGQ1NGE0MGFkZDU0NWQzNTU1OWJiZDllZmU4YjcyZDIyNTJiNzYxOWQ4OGM1MjQ2ODQzYTM2OGQ3YjdkMzQxZWI4Y2U5ZjQ3ZWZkYTQ2M2I3Yzc2MDkxMjBhZjczODNjMDNmNTAxMzhiZDg2ZmU0MTIwZTUxYWMxNQ==; H_WISE_SIDS=60360_60674_60683_60694_60572'
+    # request = BDAITPZSRequest(original_url=url, thumb_url=url)
+
+    url = "https://edit-upload-pic.cdn.bcebos.com/fff888d57a9b2ae0ffd802c8f77d80a0.jpeg?authorization=bce-auth-v1%2FALTAKh1mxHnNIyeO93hiasKJqq%2F2024-10-17T09%3A48%3A15Z%2F3600%2Fhost%2F4e7a06db2ea622123db01a2c81114f0377876ac2acb019f1f1594ec67a07af8d"
+
+    request = BDAITPZSRequest(
+        type="9",
+        # picInfo=image_to_base64(url),
+        original_url=url,
+        thumb_url=url
+    )
+    # arun(create_task(request, token=token, is_async=False, response_format='url'))
+
+    # arun(get_task('cr84hhue1pn8m151of3g', response_format='url'))
+
+    url = "https://oss.ffire.cc/files/kling_watermark.png"
+    url = "https://cdn.meimeiqushuiyin.cn/2024-10-20/removeBg/tmp_037244e2d6e1ebb8aa6ae3e98584f7b0482fed4ec5ecae98.jpg"
+
+    request = BaiduImageProcessRequest(
+        task="matting",
+        image=url,
+    )
+
+    arun(image_process(request))

meutils/apis/baidu/test.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : test
+# @Time         : 2024/10/17 18:45
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import io
+
+from meutils.pipe import *
+
+from PIL import Image, ImageDraw
+
+# data = json.loads(Path('p1.json').read_text())
+#
+data = json.loads(Path('p2.json').read_text())
+
+from PIL import Image, ImageDraw
+
+def crop_polygon(image_path, outline_points, inline_points):
+    if isinstance(image_path, bytes):
+        image_path = io.BytesIO(image_path)
+
+    # 打开图像
+    img = Image.open(image_path)
+    print(img.size)
+
+    # 创建一个与原图大小相同的黑色遮罩
+    mask = Image.new('L', img.size, 0)
+
+    # 在遮罩上绘制白色多边形
+    for points in outline_points:
+        ImageDraw.Draw(mask).polygon(points, outline=1, fill=255)
+
+    for points in inline_points:
+        ImageDraw.Draw(mask).polygon(points, outline=1, fill=0)
+
+    # 将遮罩应用到原图
+    output = Image.new('RGBA', img.size, (0, 0, 0, 0))
+    output.paste(img, (0, 0), mask)
+
+    # # 将 PIL Image 转换为字节
+    # buffer = io.BytesIO()
+    # output.save(buffer, format="PNG")
+    # byte_data = buffer.getvalue()
+    #
+    # return byte_data
+    return output
+
+
+if __name__ == '__main__':
+    # 使用示例
+    # image_path = 'img.png'
+    image_path = 'test.jpg'
+    image_path = open('test.jpg', 'rb').read()
+
+    # points = list(np.array(points))
+    n = 3567 / 1280
+    n = 2.75
+    n = 1
+
+    outline_points = []
+    inline_points = []
+
+    for outlines in data['foreground'][0].get('outlines', []):
+        _ = list(map(lambda point: (point['x'], point['y']), outlines['line']))
+        outline_points.append(_)
+
+    for inlines in data['foreground'][0].get('inlines', []):
+        _ = list(map(lambda point: (point['x'], point['y']), inlines['line']))
+        inline_points.append(_)
+
+    result = crop_polygon(image_path, outline_points, inline_points)
+    result.save('output1111.png')
+
+    type(result)

meutils/apis/chatglm/__init__.py

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