|
from PIL import Image, ImageDraw, ImageFont, ImageEnhance |
|
from pilmoji import Pilmoji |
|
|
|
from fastapi import FastAPI, Request, Query |
|
from fastapi.responses import StreamingResponse |
|
import uvicorn |
|
import requests |
|
import warnings |
|
import io |
|
from wrap import fw_wrap |
|
|
|
warnings.simplefilter("ignore") |
|
|
|
BASE_GD_IMAGE = Image.open("images/base-gd.png") |
|
BASE_RV_IMAGE = Image.open("images/base-gd-rv.png") |
|
|
|
BASE_GD_W_IMAGE = Image.open("images/base-gd-w.png") |
|
BASE_RV_W_IMAGE = Image.open("images/base-gd-w-rv.png") |
|
|
|
BASE_IMAGE = Image.open("images/base.png") |
|
if BASE_IMAGE.mode != "RGBA": |
|
BASE_IMAGE = BASE_IMAGE.convert("RGBA") |
|
MPLUS_FONT = ImageFont.truetype("fonts/MPLUSRounded1c-Regular.ttf", size=16) |
|
MPLUS_FONTBOLD = ImageFont.truetype("fonts/MPLUSRounded1c-Bold.ttf", size=16) |
|
|
|
|
|
BRAND = "http://s.id/MaiSakurajima" |
|
|
|
def getsize(font, text): |
|
left, top, right, bottom = font.getbbox(text) |
|
return right - left, bottom |
|
|
|
from PIL import ImageFont, ImageDraw |
|
from pilmoji import Pilmoji |
|
|
|
|
|
def drawText(im, ofs, string, font="fonts/MPLUSRounded1c-Regular.ttf", bold_font="fonts/MPLUSRounded1c-Bold.ttf", size=16, color=(0, 0, 0, 255), split_len=None, padding=4, disable_dot_wrap=False): |
|
v = ImageDraw.Draw(im) |
|
fontObj = ImageFont.truetype(font, size=size) |
|
boldFontObj = ImageFont.truetype(bold_font, size=size) |
|
|
|
pure_lines = [] |
|
pos = 0 |
|
l = "" |
|
|
|
if not disable_dot_wrap: |
|
for char in string: |
|
if char == "\n": |
|
pure_lines.append(l) |
|
l = "" |
|
pos += 1 |
|
elif char in ["、", ",", "。", "."]: |
|
pure_lines.append(l + char) |
|
l = "" |
|
pos += 1 |
|
else: |
|
l += char |
|
pos += 1 |
|
if l: |
|
pure_lines.append(l) |
|
else: |
|
pure_lines = string.split("\n") |
|
|
|
lines = [] |
|
for line in pure_lines: |
|
lines.extend(fw_wrap(line, width=split_len)) |
|
|
|
dy = 0 |
|
draw_lines = [] |
|
|
|
for line in lines: |
|
tsize = getsize(fontObj, line) |
|
ofs_y = ofs[1] + dy |
|
t_height = tsize[1] |
|
x = int(ofs[0] - (tsize[0] / 2)) |
|
draw_lines.append((x, ofs_y, line)) |
|
ofs_y += t_height + padding |
|
dy += t_height + padding |
|
|
|
adj_y = -30 * (len(draw_lines) - 1) |
|
|
|
for dl in draw_lines: |
|
with Pilmoji(im) as p: |
|
parts = dl[2].split('*') |
|
xPos = dl[0] |
|
yPos = adj_y + dl[1] |
|
|
|
for i, part in enumerate(parts): |
|
if i % 2 == 1: |
|
p.text((xPos, yPos), part, font=boldFontObj, fill=color) |
|
xPos += v.textlength(part, font=boldFontObj) |
|
else: |
|
subparts = part.split('_') |
|
for j, subpart in enumerate(subparts): |
|
if j % 2 == 1: |
|
text_width, text_height = getsize(fontObj, subpart) |
|
|
|
padding = 15 |
|
italic_text_image = Image.new("RGBA", (text_width + padding * 2, text_height + padding * 2), (0, 0, 0, 0)) |
|
italic_draw = ImageDraw.Draw(italic_text_image) |
|
italic_draw.text((padding, padding), subpart, font=fontObj, fill=color) |
|
|
|
|
|
slant_amount = 0.2 |
|
italic_text_image = italic_text_image.transform( |
|
italic_text_image.size, |
|
Image.AFFINE, |
|
(1, slant_amount, 0, 0, 1, 0), |
|
resample=Image.BICUBIC |
|
) |
|
|
|
im.alpha_composite(italic_text_image, dest=(int(xPos), int(yPos - padding))) |
|
|
|
xPos += text_width + int(slant_amount * size) |
|
else: |
|
strikethrough_parts = subpart.split('~') |
|
for k, s_part in enumerate(strikethrough_parts): |
|
if k % 2 == 1: |
|
p.text((xPos, yPos), s_part, font=fontObj, fill=color) |
|
text_width = v.textlength(s_part, font=fontObj) |
|
line_y = yPos + text_height // 2 |
|
v.line((xPos, line_y, xPos + text_width, line_y), fill=color, width=1) |
|
xPos += text_width |
|
else: |
|
p.text((xPos, yPos), s_part, font=fontObj, fill=color) |
|
xPos += v.textlength(s_part, font=fontObj) |
|
|
|
real_y = ofs[1] + adj_y + dy |
|
return (0, dy, real_y) |
|
|
|
def make(name, id, content, icon): |
|
img = BASE_IMAGE.copy() |
|
|
|
icon = Image.open(io.BytesIO(requests.get(icon).content)) |
|
icon = icon.resize((720, 720), Image.LANCZOS) |
|
icon = icon.convert("L") |
|
icon_filtered = ImageEnhance.Brightness(icon) |
|
|
|
img.paste(icon_filtered.enhance(0.7), (0, 0)) |
|
img.paste(BASE_GD_IMAGE, (0, 0), BASE_GD_IMAGE) |
|
|
|
tx = ImageDraw.Draw(img) |
|
|
|
tsize_t = drawText(img, (890, 270), content, size=55, color=(255, 255, 255, 255), split_len=20) |
|
|
|
name_y = tsize_t[2] + 40 |
|
tsize_name = drawText(img, (890, name_y), f"@{name}", size=28, color=(255, 255, 255, 255), split_len=25, disable_dot_wrap=True) |
|
|
|
id_y = name_y + tsize_name[1] + 4 |
|
drawText(img, (890, id_y), id, size=18, color=(180, 180, 180, 255), split_len=45, disable_dot_wrap=True) |
|
|
|
tx.text((1070, 694), BRAND, font=MPLUS_FONT, fill=(120, 120, 120, 255)) |
|
|
|
file = io.BytesIO() |
|
img.save(file, format="PNG", quality=95) |
|
file.seek(0) |
|
return file |
|
|
|
def colorMake(name, id, content, icon): |
|
img = BASE_IMAGE.copy() |
|
|
|
icon = Image.open(io.BytesIO(requests.get(icon).content)) |
|
icon = icon.resize((720, 720), Image.LANCZOS) |
|
|
|
img.paste(icon, (0, 0)) |
|
img.paste(BASE_GD_IMAGE, (0, 0), BASE_GD_IMAGE) |
|
|
|
tx = ImageDraw.Draw(img) |
|
|
|
tsize_t = drawText(img, (890, 270), content, size=55, color=(255, 255, 255, 255), split_len=20) |
|
|
|
name_y = tsize_t[2] + 40 |
|
tsize_name = drawText(img, (890, name_y), f"@{name}", size=28, color=(255, 255, 255, 255), split_len=25, disable_dot_wrap=True) |
|
|
|
id_y = name_y + tsize_name[1] + 4 |
|
drawText(img, (890, id_y), id, size=18, color=(180, 180, 180, 255), split_len=45, disable_dot_wrap=True) |
|
|
|
tx.text((1070, 694), BRAND, font=MPLUS_FONT, fill=(120, 120, 120, 255)) |
|
|
|
file = io.BytesIO() |
|
img.save(file, format="PNG", quality=95) |
|
file.seek(0) |
|
return file |
|
|
|
def reverseMake(name, id, content, icon): |
|
img = BASE_IMAGE.copy() |
|
|
|
icon = Image.open(io.BytesIO(requests.get(icon).content)) |
|
icon = icon.resize((720, 720), Image.LANCZOS) |
|
icon = icon.convert("L") |
|
icon_filtered = ImageEnhance.Brightness(icon) |
|
|
|
img.paste(icon_filtered.enhance(0.7), (570, 0)) |
|
img.paste(BASE_RV_IMAGE, (0, 0), BASE_RV_IMAGE) |
|
|
|
tx = ImageDraw.Draw(img) |
|
|
|
tsize_t = drawText(img, (390, 270), content, size=55, color=(255, 255, 255, 255), split_len=20) |
|
|
|
name_y = tsize_t[2] + 40 |
|
tsize_name = drawText(img, (390, name_y), f"@{name}", size=28, color=(255, 255, 255, 255), split_len=25, disable_dot_wrap=True) |
|
|
|
id_y = name_y + tsize_name[1] + 4 |
|
drawText(img, (390, id_y), id, size=18, color=(180, 180, 180, 255), split_len=45, disable_dot_wrap=True) |
|
|
|
tx.text((6, 694), BRAND, font=MPLUS_FONT, fill=(120, 120, 120, 255)) |
|
|
|
file = io.BytesIO() |
|
img.save(file, format="PNG", quality=95) |
|
file.seek(0) |
|
return file |
|
|
|
def reverseColorMake(name, id, content, icon): |
|
img = BASE_IMAGE.copy() |
|
|
|
icon = Image.open(io.BytesIO(requests.get(icon).content)) |
|
icon = icon.resize((720, 720), Image.LANCZOS) |
|
|
|
img.paste(icon, (570, 0)) |
|
img.paste(BASE_RV_IMAGE, (0, 0), BASE_RV_IMAGE) |
|
|
|
tx = ImageDraw.Draw(img) |
|
|
|
tsize_t = drawText(img, (390, 270), content, size=55, color=(255, 255, 255, 255), split_len=20) |
|
|
|
name_y = tsize_t[2] + 40 |
|
tsize_name = drawText(img, (390, name_y), f"@{name}", size=28, color=(255, 255, 255, 255), split_len=25, disable_dot_wrap=True) |
|
|
|
id_y = name_y + tsize_name[1] + 4 |
|
drawText(img, (390, id_y), id, size=18, color=(180, 180, 180, 255), split_len=45, disable_dot_wrap=True) |
|
|
|
tx.text((6, 694), BRAND, font=MPLUS_FONT, fill=(120, 120, 120, 255)) |
|
|
|
file = io.BytesIO() |
|
img.save(file, format="PNG", quality=95) |
|
file.seek(0) |
|
return file |
|
|
|
def whiteMake(name, id, content, icon): |
|
img = BASE_IMAGE.copy() |
|
|
|
icon = Image.open(io.BytesIO(requests.get(icon).content)).convert("RGBA") |
|
icon = icon.resize((720, 720), Image.LANCZOS) |
|
|
|
img.paste(icon, (0, 0), icon) |
|
img.paste(BASE_GD_W_IMAGE, (0, 0), BASE_GD_W_IMAGE) |
|
|
|
tx = ImageDraw.Draw(img) |
|
|
|
tsize_t = drawText(img, (890, 270), content, size=55, color=(0, 0, 0, 0), split_len=20) |
|
|
|
name_y = tsize_t[2] + 40 |
|
tsize_name = drawText(img, (890, name_y), f"@{name}", size=28, color=(0, 0, 0, 0), split_len=25, disable_dot_wrap=True) |
|
|
|
id_y = name_y + tsize_name[1] + 4 |
|
drawText(img, (890, id_y), id, size=18, color=(90, 90, 90, 255), split_len=45, disable_dot_wrap=True) |
|
|
|
tx.text((1070, 694), BRAND, font=MPLUS_FONT, fill=(120, 120, 120, 255)) |
|
|
|
file = io.BytesIO() |
|
img.save(file, format="PNG", quality=95) |
|
file.seek(0) |
|
return file |
|
|
|
def reverseWhiteMake(name, id, content, icon): |
|
img = BASE_IMAGE.copy() |
|
|
|
icon = Image.open(io.BytesIO(requests.get(icon).content)).convert("RGBA") |
|
icon = icon.resize((720, 720), Image.LANCZOS) |
|
|
|
img.paste(icon, (570, 0), icon) |
|
img.paste(BASE_RV_W_IMAGE, (0, 0), BASE_RV_W_IMAGE) |
|
|
|
tx = ImageDraw.Draw(img) |
|
|
|
tsize_t = drawText(img, (390, 270), content, size=55, color=(0, 0, 0, 0), split_len=20) |
|
|
|
name_y = tsize_t[2] + 40 |
|
tsize_name = drawText(img, (390, name_y), f"@{name}", size=28, color=(0, 0, 0, 0), split_len=25, disable_dot_wrap=True) |
|
|
|
id_y = name_y + tsize_name[1] + 4 |
|
drawText(img, (390, id_y), id, size=18, color=(90, 90, 90, 255), split_len=45, disable_dot_wrap=True) |
|
|
|
tx.text((6, 694), BRAND, font=MPLUS_FONT, fill=(110, 110, 110, 255)) |
|
|
|
file = io.BytesIO() |
|
img.save(file, format="PNG", quality=95) |
|
file.seek(0) |
|
return file |
|
|
|
app = FastAPI() |
|
@app.get("/") |
|
async def main( |
|
type: str = Query(None), |
|
name: str = Query("SAMPLE"), |
|
id: str = Query(""), |
|
content: str = Query("Make it a Quote"), |
|
icon: str = Query("https://cdn.discordapp.com/embed/avatars/0.png") |
|
): |
|
if type == "color": |
|
image_io = colorMake(name, id, content, icon) |
|
elif type == "reverse": |
|
image_io = reverseMake(name, id, content, icon) |
|
elif type == "reverseColor": |
|
image_io = reverseColorMake(name, id, content, icon) |
|
elif type == "white": |
|
image_io = whiteMake(name, id, content, icon) |
|
elif type == "reverseWhite": |
|
image_io = reverseWhiteMake(name, id, content, icon) |
|
else: |
|
image_io = make(name, id, content, icon) |
|
|
|
image_io.seek(0) |
|
|
|
return StreamingResponse(image_io, media_type="image/png") |
|
|
|
if __name__ == "__main__": |
|
uvicorn.run(app, host="0.0.0.0", port=3000) |
