Update app.py

app.py

@@ -7,7 +7,7 @@ from tqdm import tqdm
 import datetime
 import os
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 from translate import Translator
 from gradio_client import Client
 import json
@@ -97,6 +97,23 @@ def inference(original_prompt: str, image: Image) -> Image:
   res_img = Image.open(result_path)
   print('作品：' + result_path)
 
+  # 加载字体，设置字体大小
+  font_path = 'ttf/WawaSC-Regular.otf'
+  font_size = 50
+  font = ImageFont.truetype(font_path, font_size)
+  text = 'by 宁侠'
+
+  x0, y0, x1, y1 = font.getbbox(text)
+  text_width = x1 - x0
+  text_height = (y1 - y0)*2
+
+  watermark = Image.new('RGBA', (text_width, text_height))
+  draw = ImageDraw.Draw(watermark)
+  draw.text((0,0), text, font=font, fill=(255,255,255)) #阿根廷蓝：112,171,221
+
+  w, h = res_img.size
+  res_img.paste(watermark, (w - text_width - 10, h - text_height), watermark)
+
   return res_img