From 3a4fe159db43f6d29209022d8904baa7bdcef99e Mon Sep 17 00:00:00 2001
From: yoshiko2 <yoshiko2.dev@gmail.com>
Date: Thu, 29 Jun 2023 03:31:00 +0800
Subject: [PATCH] Fix multiple watermark

---
 core.py | 5 +++++
 1 file changed, 5 insertions(+)

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