diff --git a/core.py b/core.py index cb6ea6d..5f811f9 100644 --- a/core.py +++ b/core.py @@ -528,10 +528,13 @@ def add_mark_thread(pic_path, cn_sub, leak, uncensored, hack, _4k, iso): count = (count + 1) % 4 if uncensored: add_to_pic(pic_path, img_pic, size, count, 3) + count = (count + 1) % 4 if hack: add_to_pic(pic_path, img_pic, size, count, 4) + count = (count + 1) % 4 if _4k: add_to_pic(pic_path, img_pic, size, count, 5) + count = (count + 1) % 4 if iso: add_to_pic(pic_path, img_pic, size, count, 6) img_pic.close() @@ -550,6 +553,8 @@ def add_to_pic(pic_path, img_pic, size, count, mode): pngpath = "Img/HACK.png" elif mode == 5: pngpath = "Img/4K.png" + elif mode == 6: + pngpath = "Img/ISO.png" else: print('[-]Error: watermark image param mode invalid!') return