Spaces:
Runtime error
Runtime error
File size: 2,795 Bytes
bc1c656 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
import cv2
from PIL import ImageFont, ImageDraw, Image
def process(img,font_text,font_fac,font_x,font_y,font_col,font_op):
img.save('tmp.png')
img = Image.open('tmp.png').convert("RGBA")
im=img
#im=Image.open(img)
txt = Image.new('RGBA', im.size, (255,255,255,0))
w, h = im.size
print (f'FONT COLOR: {font_col}')
#t_fill = ("#"+"E6"+f"{font_col.strip('#')}")
#t_fill = (font_col)
h1 = font_col.strip("#")
rgb_tup = tuple(int(h1[i:i+2], 16) for i in (0, 2, 4))
print (rgb_tup)
a,b,c = rgb_tup
t_fill = (a,b,c,font_op)
print (f'FONT COLOR: {t_fill}')
#x = int(w-font_x)
#y = int(h-font_y)
x = int(font_x)
y = int(font_y)
draw = ImageDraw.Draw(txt)
text = f'{font_text}'
font_size=font_fac
font = ImageFont.truetype("./fonts/SansitaOne.ttf", int(font_size))
size = font.getsize(text)
draw.text((x-size[0]/2, y),text, font = font, fill=t_fill)
#txt.putalpha(128)
combined = Image.alpha_composite(im, txt)
return combined
def textover(im,txt1="",txt2=""):
#im.save('tmp.png')
im = Image.open(im)
inp=1
hh=0
hhh=25
#cnt = len(inp)
cnt = inp
font_a = 30
font_b = 10
if cnt >0:
font_a = font_a + (cnt * 2)
font_b = font_b + (cnt * 2)
#hh = hh-int(cnt/2)
hhh = hhh+int(cnt/2)
w,h = im.size
print (w)
print (h)
font_x = (w/2)
font_y = h-hhh
out = process(im,txt1,font_fac=font_a,font_x=font_x,font_y=hh,font_col="#000000",font_op=255)
out = process(out,txt2,font_fac=font_b,font_x=font_x,font_y=font_y,font_col="#000000",font_op=255)
#out.save("out.png")
return out
def background_image():
img_b = np.zeros([512,512,3],dtype=np.uint8)
img_b.fill(255)
img_b.save("background.png")
def custom_overlay(txt=None):
background_image()
background = Image.open("background.png")
x,y = background.size
if x > y:
aa=y
if y > x:
aa=x
if x==y:
aa=x
out = f'{txt}'
print (f'txt {out}')
qrm = qr.QRCode(box_size=10,error_correction=qr.constants.ERROR_CORRECT_H)
qrm.add_data(out)
qrm.make(fit=True)
img1 = qrm.make_image(fill_color=fill, back_color="white")
img1 = img1.resize((aa,aa))
img1.save("im2.png")
image_bgr = cv2.imread('im2.png')
h, w, c = image_bgr.shape
image_bgra = np.concatenate([image_bgr, np.full((h, w, 1), 255, dtype=np.uint8)], axis=-1)
white = np.all(image_bgr == [255, 255, 255], axis=-1)
image_bgra[white, -1] = 0
cv2.imwrite('result.png', image_bgra)
over_im = Image.open("result.png")
background.paste(over_im, (0, 0), over_im)
background.save("out_im.png")
return "out_im.png" |