udpate script

script/image_mark.py

@@ -4,18 +4,22 @@ import sys
 import random
 from PIL import Image,ImageDraw,ImageFont
 def add_text_to_image(image, text):
+    mode = 'RGB'
+    if image.endswith(".png"):
+        mode = "RGBA"
+
     #水印字体随机大小
     font_size = random.randint(30,60)
     font = ImageFont.truetype('Arial.ttf', font_size)
     
     # 添加背景
-    new_img = Image.new('RGBA', (image.size[0] * 3, image.size[1] * 3), (0, 0, 0, 0))
+    new_img = Image.new(mode, (image.size[0] * 3, image.size[1] * 3), (0, 0, 0, 0))
     new_img.paste(image, image.size)
     
     # 添加水印
     font_len = len(text)
-    rgba_image = new_img.convert('RGBA')
-    text_overlay = Image.new('RGBA', rgba_image.size, (255, 255, 255, 0))
+    rgba_image = new_img.convert(mode)
+    text_overlay = Image.new(mode, rgba_image.size, (255, 255, 255, 0))
     image_draw = ImageDraw.Draw(text_overlay)
     
     #随机位置