from app.schemas.generate_image import BatchGenerateProductImageModel
from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product

if __name__ == '__main__':
    rd = BatchGenerateProductImageModel(
        batch_tasks_id="",
        batch_data_list="",
        user_id=""
    )
    x = batch_generate_product.delay(rd.dict())
    print(x)

    """relight"""
    # rd = BatchGenerateRelightImageModel(
    #     tasks_id="123-89",
    #     # prompt="beautiful woman, detailed face, sunshine, outdoor, warm atmosphere",
    #     prompt="Colorful black",
    #     image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
    #     direction="Right Light",
    #     product_type="single",
    #     batch_size=2
    # )
    # batch_generate_relight.delay(rd.dict())
    """pose transform"""
    # rd = BatchPoseTransformModel(
    #     tasks_id="123-89",
    #     image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
    #     pose_id="1",
    #     batch_size=10
    # )
    # batch_pose_transform_tasks.delay(rd.dict())