Update app.py

app.py

@@ -25,6 +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
 
 # logging
 
@@ -239,7 +240,12 @@ def handle_generate(*_inp):
     pipe_out = pipe_generate(*inp)
     
     if inp[5] != "":
-        
+        draw = ImageDraw.Draw(pipe_out)
+        textwidth, textheight = draw.textsize(inp[5])
+        margin = 10
+        x = width - textwidth - margin
+        y = height - textheight - margin
+        draw.text((x, y), inp[5])
         
     name = generate_random_string(12) + ( ".png" if time == 0 else ".gif" )
     if inp[4] == 0.0: