update task_listener.py

AeonOrg · Nov 26, 2024 · c0f8aa8 · c0f8aa8
1 parent 606c41a
commit c0f8aa8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/bot/helper/listeners/task_listener.py b/bot/helper/listeners/task_listener.py
@@ -153,6 +153,11 @@ async def on_download_complete(self):
             if self.isCancelled:
                 return
 
+        if self.watermark:
+            up_path = await self.proceedWatermark(up_path, gid)
+            if self.isCancelled:
+                return
+
         if self.screenShots:
             up_path = await self.generateScreenshots(up_path)
             if self.isCancelled: