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

from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight

if __name__ == '__main__':
    rd = BatchGenerateProductImageModel(
        tasks_id="test1-89",
        image_strength=0.7,
        prompt=" The best quality, masterpiece, real image.Outwear,high quality clothing details,8K realistic,HDR",
        image_url="aida-results/result_40b1a2fe-e220-11ef-9bfa-0242ac150003.png",
        product_type="single",
        batch_size=2
    )
    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())