refactor: move `captioned_strip` to library.

Different versions are used in several places.


Former-commit-id: e74b5ae17c035e953d2859a609d651464db0c64c

dalle_mini/helpers.py

@@ -0,0 +1,14 @@
+from PIL import Image, ImageDraw, ImageFont
+
+def captioned_strip(images, caption):
+    increased_h = 0 if caption is None else 48
+    w, h = images[0].size[0], images[0].size[1]
+    img = Image.new("RGB", (len(images)*w, h + increased_h))
+    for i, img_ in enumerate(images):
+        img.paste(img_, (i*w, increased_h))
+
+    if caption is not None:
+        draw = ImageDraw.Draw(img)
+        font = ImageFont.truetype("/usr/share/fonts/truetype/liberation2/LiberationMono-Bold.ttf", 40)
+        draw.text((20, 3), caption, (255,255,255), font=font)
+    return img

dev/predictions/wandb-examples-from-backend.py

@@ -6,6 +6,7 @@ import wandb
 import os
 
 from dalle_mini.backend import ServiceError, get_images_from_backend
+from dalle_mini.helpers import captioned_strip
 
 os.environ["WANDB_SILENT"] = "true"
 os.environ["WANDB_CONSOLE"] = "off"
@@ -19,16 +20,6 @@ run = wandb.init(id=id,
         resume="allow"
 )
 
-def captioned_strip(images, caption):
-    w, h = images[0].size[0], images[0].size[1]
-    img = Image.new("RGB", (len(images)*w, h + 48))
-    for i, img_ in enumerate(images):
-        img.paste(img_, (i*w, 48))
-    draw = ImageDraw.Draw(img)
-    font = ImageFont.truetype("/usr/share/fonts/truetype/liberation2/LiberationMono-Bold.ttf", 40)
-    draw.text((20, 3), caption, (255,255,255), font=font)
-    return img
-
 def log_to_wandb(prompts):
     try:
         backend_url = os.environ["BACKEND_SERVER"]

dev/predictions/wandb-examples.py

@@ -178,18 +178,8 @@ def clip_top_k(prompt, images, k=8):
 
 # ## Log to wandb
 
-from PIL import ImageDraw, ImageFont
+from dalle_mini.helpers import captioned_strip
 
-def captioned_strip(images, caption):
-    w, h = images[0].size[0], images[0].size[1]
-    img = Image.new("RGB", (len(images)*w, h + 48))
-    for i, img_ in enumerate(images):
-        img.paste(img_, (i*w, 48))
-    draw = ImageDraw.Draw(img)
-    font = ImageFont.truetype("/usr/share/fonts/truetype/liberation2/LiberationMono-Bold.ttf", 40)
-    draw.text((20, 3), caption, (255,255,255), font=font)
-    return img
-    
 def log_to_wandb(prompts):
     strips = []
     for prompt in prompts: