Update app.py

app.py

@@ -25,7 +25,7 @@ from diffusers.utils import export_to_gif, load_image
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file, save_file
 from diffusers import AnimateDiffPipeline, MotionAdapter, DDIMScheduler, StableDiffusionPipeline
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFont
 
 # logging
 
@@ -123,12 +123,6 @@ pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
 
 # functionality
 
-def textsize(text):
-    im = Image.new(mode="P", size=(0, 0))
-    draw = ImageDraw.Draw(im)
-    _, _, width, height = draw.textbbox((0, 0), text=text)
-    return width, height
-
 def run(cmd):
     return str(subprocess.run(cmd, shell=True, capture_output=True, env=None).stdout)
 
@@ -247,11 +241,12 @@ def handle_generate(*_inp):
     
     if inp[5] != "":
         draw = ImageDraw.Draw(pipe_out)
-        textwidth, textheight = textsize(inp[5])
-        margin = 10
-        x = width - textwidth - margin
-        y = height - textheight - margin
-        draw.text((x, y), inp[5])
+        textheight=32
+        font = ImageFont.truetype(r"OpenSans-Bold.ttf", textheight)
+        textwidth = draw.textlength(inp[5])
+        x = (width - textwidth) // 2
+        y = (height - textheight) // 2
+        draw.text((x, y), inp[5],font=font)
         
     name = generate_random_string(12) + ( ".png" if time == 0 else ".gif" )
     if inp[4] == 0.0: