Spaces:
Building
Building
| from PIL import Image, ImageDraw, ImageFont, ImageEnhance | |
| from pilmoji import Pilmoji | |
| # from flask import Flask, request, send_file | |
| 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://hisoka.net\nhttp://s.id/MaiSakurajima" | |
| # print(MPLUS_FONT.get_variation_names()) | |
| 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('*') # * for bold | |
| 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 italic | |
| 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 # Middle of the text | |
| 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() | |
| 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) |