# uv pip install volcengine
from volcengine.visual.VisualService import VisualService
from JiMeng.Kit.JmCommon import JmCommon

if __name__ == '__main__':
    visual_service = VisualService()

    visual_service.set_ak(JmCommon.ak)
    visual_service.set_sk(JmCommon.sk)

    # https://www.volcengine.com/docs/6791/1330203
    form = {
        "req_key": "t2i_xl_sft",
        # "prompt": "少女，光影，瘦，白皙，干净，美丽",
        "prompt": "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的工人，背着一个小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。",
        "width": 1024,
        "height": 1024,
        "seed": -1,
        "ddim_steps": 20,
        "scale": 7.0,
        "return_url": True,
        "logo_info": {
            "add_logo": True,
            "position": 2,
            "language": 0,
            "opacity": 1
        }
    }
    resp = visual_service.text2img_xl_sft(form)
    print(resp)