app/examples.py

from PIL import Image

make_dict = lambda x : { 'background': Image.open(x).convert("RGBA"), 'layers': [Image.new("RGBA", Image.open(x).size, (255, 255, 255, 0))], 'composite': Image.open(x).convert("RGBA") }
    
null_dict = {
    'background': None,
    'composite': None,
    'layers': []
}

IMG_REF = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/img_ref.png",
    "assets/gradio/pos_aware/002/img_ref.png", 
    ## pos-aware: User-drawn mask mask
    "assets/gradio/pos_aware/003/img_ref.png",
    "assets/gradio/pos_aware/004/img_ref.png",
    "assets/gradio/pos_aware/005/img_ref.png",
    ## pos-free
    "assets/gradio/pos_free/001/img_ref.png",
    "assets/gradio/pos_free/002/img_ref.png",
    "assets/gradio/pos_free/003/img_ref.png",
    "assets/gradio/pos_free/004/img_ref.png",
    ]

IMG_TGT1 = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/img_target.png", 
    "assets/gradio/pos_aware/002/img_target.png", 
    ## pos-aware: User-drawn mask mask
    None,
    None,
    None,
    ## pos-free
    "assets/gradio/pos_free/001/img_target.png", 
    "assets/gradio/pos_free/002/img_target.png",
    "assets/gradio/pos_free/003/img_target.png",
    "assets/gradio/pos_free/004/img_target.png",
    ]

IMG_TGT2 = [
    ## pos-aware: precise mask
    null_dict, 
    null_dict, 
    ## pos-aware: User-drawn mask mask
    make_dict("assets/gradio/pos_aware/003/img_target.png"), 
    make_dict("assets/gradio/pos_aware/004/img_target.png"),
    make_dict("assets/gradio/pos_aware/005/img_target.png"),
    ## pos-free
    null_dict,
    null_dict,
    null_dict,
    null_dict,
    ]

MASK_TGT = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/mask_target.png", 
    "assets/gradio/pos_aware/002/mask_target.png", 
    ## pos-aware: User-drawn mask mask
    "assets/gradio/pos_aware/003/mask_target.png", 
    "assets/gradio/pos_aware/004/mask_target.png",
    "assets/gradio/pos_aware/005/mask_target.png",
    ## pos-free
    "assets/gradio/pos_free/001/mask_target.png",
    "assets/gradio/pos_free/002/mask_target.png", 
    "assets/gradio/pos_free/003/mask_target.png",
    "assets/gradio/pos_free/004/mask_target.png",
    ]

CUSTOM_MODE = [
    ## pos-aware
    "Position-aware",  
    "Position-aware", 
    "Position-aware", 
    "Position-aware",
    "Position-aware",
    ## pos-free
    "Position-free",
    "Position-free",
    "Position-free",
    "Position-free",
    ]

INPUT_MASK_MODE = [
    ## pos-aware: precise mask
    "Precise mask", 
    "Precise mask", 
    ## pos-aware: User-drawn mask mask
    "User-drawn mask", 
    "User-drawn mask",
    "User-drawn mask",
    ## pos-free
    "Precise mask",
    "Precise mask",
    "Precise mask",
    "Precise mask",
    ]

SEG_REF_MODE = [
    ## pos-aware
    "Full Ref", 
    "Full Ref", 
    "Full Ref", 
    "Full Ref", 
    "Full Ref",
    ## pos-free
    "Full Ref",
    "Full Ref",
    "Full Ref",
    "Full Ref",
    ]

PROMPTS = [
    ## pos-aware: precise mask
    "", 
    "", 
    ## pos-aware: User-drawn mask mask
    "A delicate necklace with a mother-of-pearl clover pendant hangs gracefully around the neck of a woman dressed in a black pinstripe blazer.",
    "",
    "",
    ## pos-free
    "TThe charming, soft plush toy is joyfully wandering through a lush, dense jungle, surrounded by vibrant green foliage and towering trees.",
    "A bright yellow alarm clock sits on a wooden desk next to a stack of books in a cozy, sunlit room.",
    "A Lego figure dressed in a vibrant chicken costume, leaning against a wooden chair, surrounded by lush green grass and blooming flowers.",
    "The crocheted gingerbread man is perched on a tree branch in a dense forest, with sunlight filtering through the leaves, casting dappled shadows around him."
    ]

IMG_GEN = [
    ## pos-aware: precise mask
    "assets/gradio/pos_aware/001/img_gen.png", 
    "assets/gradio/pos_aware/002/img_gen.png", 
    ## pos-aware: User-drawn mask mask
    "assets/gradio/pos_aware/003/img_gen.png",
    "assets/gradio/pos_aware/004/img_gen.png",
    "assets/gradio/pos_aware/005/img_gen.png",
    ## pos-free
    "assets/gradio/pos_free/001/img_gen.png",
    "assets/gradio/pos_free/002/img_gen.png",
    "assets/gradio/pos_free/003/img_gen.png",
    "assets/gradio/pos_free/004/img_gen.png",
    ]

SEED = [
    ## pos-aware
    97175498,
    2126677963, 
    346969695,
    1172525388,
    268683460,
    ## pos-free
    2126677963,
    418898253,
    2126677963,
    2126677963
    ]

TRUE_GS = [
    # pos-aware
    1,
    1,
    1,
    1,
    1,
    # pos-free
    3,
    3,
    3,
    3,
]

NUM_STEPS = [
    ## pos-aware
    32,
    32,
    32,
    32,
    32,
    ## pos-free
    20,
    20,
    20,
    20,
]

GUIDANCE = [
    ## pos-aware
    40,
    48,
    40,
    48,
    48,
    ## pos-free
    40,
    40,
    40,
    40,
]

GRADIO_EXAMPLES = [
    [IMG_REF[0], IMG_TGT1[0], IMG_TGT2[0], CUSTOM_MODE[0], INPUT_MASK_MODE[0], SEG_REF_MODE[0], PROMPTS[0], SEED[0], TRUE_GS[0], '0', NUM_STEPS[0], GUIDANCE[0]],
    [IMG_REF[1], IMG_TGT1[1], IMG_TGT2[1], CUSTOM_MODE[1], INPUT_MASK_MODE[1], SEG_REF_MODE[1], PROMPTS[1], SEED[1], TRUE_GS[1], '1', NUM_STEPS[1], GUIDANCE[1]],
    [IMG_REF[2], IMG_TGT1[2], IMG_TGT2[2], CUSTOM_MODE[2], INPUT_MASK_MODE[2], SEG_REF_MODE[2], PROMPTS[2], SEED[2], TRUE_GS[2], '2', NUM_STEPS[2], GUIDANCE[2]],
    [IMG_REF[3], IMG_TGT1[3], IMG_TGT2[3], CUSTOM_MODE[3], INPUT_MASK_MODE[3], SEG_REF_MODE[3], PROMPTS[3], SEED[3], TRUE_GS[3], '3', NUM_STEPS[3], GUIDANCE[3]],
    [IMG_REF[4], IMG_TGT1[4], IMG_TGT2[4], CUSTOM_MODE[4], INPUT_MASK_MODE[4], SEG_REF_MODE[4], PROMPTS[4], SEED[4], TRUE_GS[4], '4', NUM_STEPS[4], GUIDANCE[4]],
    [IMG_REF[5], IMG_TGT1[5], IMG_TGT2[5], CUSTOM_MODE[5], INPUT_MASK_MODE[5], SEG_REF_MODE[5], PROMPTS[5], SEED[5], TRUE_GS[5], '5', NUM_STEPS[5], GUIDANCE[5]],
    [IMG_REF[6], IMG_TGT1[6], IMG_TGT2[6], CUSTOM_MODE[6], INPUT_MASK_MODE[6], SEG_REF_MODE[6], PROMPTS[6], SEED[6], TRUE_GS[6], '6', NUM_STEPS[6], GUIDANCE[6]],
    [IMG_REF[7], IMG_TGT1[7], IMG_TGT2[7], CUSTOM_MODE[7], INPUT_MASK_MODE[7], SEG_REF_MODE[7], PROMPTS[7], SEED[7], TRUE_GS[7], '7', NUM_STEPS[7], GUIDANCE[7]],
    [IMG_REF[8], IMG_TGT1[8], IMG_TGT2[8], CUSTOM_MODE[8], INPUT_MASK_MODE[8], SEG_REF_MODE[8], PROMPTS[8], SEED[8], TRUE_GS[8], '8', NUM_STEPS[8], GUIDANCE[8]],
]