sixsixliu / limusbot Goto Github PK

waifus = []


async def get_waifu():
    global waifus
    if not waifus:
        import json

        import requests

        URL = "https://api.waifulabs.com/generate"
        payload = {"step": 0}
        headers = {"Content-Type": "application/json"}

        response_str = requests.post(
            URL, data=json.dumps(payload), headers=headers
        ).text
        waifus.extend(
            [
                "base64://" + waifu_data["image"]
                for waifu_data in json.loads(response_str)["newGirls"]
            ]
        )
    current_waifu = waifus.pop()
    return Message(f"[CQ:image,file={current_waifu}]")