"""
# You can add any available edit to the list.

strength can be range or list:
    dict( method='ganspace', edit='eye_closeness', strength=[15,-15])
    dict( method='interfacegan', edit='pose', strength=range(-3,3,1) )    


Interpolate option can be used to edit an image towards to another one. Use --target_images
flag to specify the path of the target images.
"""

edit_configs = [
    # Inversion
    dict( method='inversion'),

    # Interpolation
    dict( method='interpolate', edit='random', strength=5),
    dict( method='interpolate', edit='target', strength=5),


    # Interfacegan
    dict( method='interfacegan', edit='smile', strength=2),
    dict( method='interfacegan', edit='age',   strength=-3),
    dict( method='interfacegan', edit='pose',  strength=1),

    # GANSpace. Find more edits in editings/ganspace_pca/ganspace_configs.csv
    dict( method='ganspace', edit='eye_openness',  strength=20),
    dict( method='ganspace', edit='lipstick',      strength=20),
    dict( method='ganspace', edit='trimmed_beard', strength=-20),
    dict( method='ganspace', edit='overexposed',   strength=5),

    # StyleClip. We support both mapper and global directions for StyleClip
    # Mapper. See all possible edits editings/styleclip/styleclip_directions/mapper. 
    dict( method='styleclip', type='mapper', edit='afro', strength=0.1),
    # Global 
    # Text prompt: face with hair -> face with x hair
    dict( method='styleclip', edit='curly_hair',        strength=3,    disentanglement=0.29),
    dict( method='styleclip', edit='bobcut_hair',       strength=3,    disentanglement= 0.16),
    dict( method='styleclip', edit='long_hair',         strength=3,    disentanglement= 0.18),
    dict( method='styleclip', edit='straight_hair',     strength=3,     disentanglement=0.17),
    dict( method='styleclip', edit='black_hair',        strength=3,        disentanglement=0.25),
    dict( method='styleclip', edit='grey_hair',         strength=4,         disentanglement=0.133),
    dict( method='styleclip', edit='fringe_hair',       strength=3,       disentanglement=0.2),
    dict( method='styleclip', edit='hightop_fade_hair', strength=3, disentanglement=0.2),


    #Text prompt: face -> face with x
    dict( method='styleclip', edit='makeup', strength=3, disentanglement=0.11),
    dict( method='styleclip', edit='wrinkle', strength=4, disentanglement=0.14),  

    #Text prompt: face -> x face 
    dict( method='styleclip', edit='sad', strength=4, disentanglement=0.12),
    dict( method='styleclip', edit='suprised', strength=4, disentanglement=0.12),
    dict( method='styleclip', edit='angry', strength=4, disentanglement=0.12),
    dict( method='styleclip', edit='female', strength=3, disentanglement=0.15), 

    dict( method='styleclip', edit='eyeglasses', strength=2),
    dict( method='styleclip', edit='bangs', strength=2),
    dict( method='styleclip', edit='pale', strength=2),
    dict( method='styleclip', edit='tanned', strength=2),

    # GradCtrl
    dict( method='gradctrl', edit='age'),
    dict( method='gradctrl', edit='smile'),
    dict( method='gradctrl', edit='eyeglasses'),


]