import os import time import uuid import numpy as np import gradio as gr import fn import edimg import edgif from PIL import Image, ImageDraw, ImageFont, ImageColor, ImageFilter, ImageSequence, PngImagePlugin from copy import deepcopy import maker import pre import pandas as pd from edimg import qr_icon img_blank = Image.open('resource/blank.png') side = Image.open("resources/side.png") side_special = Image.open("resources/side_special.png") side_oma1 = Image.open("resources/side_oma1.png") side_oma2 = Image.open("resources/side_oma2.png") def sorting(img_type, img_static, img_dynamic_file, obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, member_name_dropdown, group_name_radio, group_name_txt, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial, horizontal, preset): match img_type: case "Image": if img_static == None: gr.Warning("Upload the image first") return "resource/upload_first.png", "resource/upload_first.png", None, None img_static = Image.open(img_static) if not img_static.size == (1164, 1800): img_static = edimg.resize_img(img_static, 1164, 1800) print('resized!') img, img_copy = front_static(img_static, obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, member_name_dropdown, group_name_radio, group_name_txt, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial, preset) if horizontal: img = Image.open(img) img_rotate = img.rotate(90, expand=True) return img_rotate, img, None, img_copy return img, img, None, img_copy case "Gif": if img_dynamic_file == None: gr.Warning("Upload the gif first") return "resource/upload_first.png", "resource/upload_first.png", None _, overlay = front_static(img_blank, obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, member_name_dropdown, group_name_radio, group_name_txt, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial, preset) gif = Image.open(img_dynamic_file) edgif.attach_gif(gif, overlay) return "output.gif", "output.gif", "output.gif", None def front_static(img, obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, member_name_dropdown, group_name_radio, group_name_txt, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial, preset): meta = PngImagePlugin.PngInfo() meta.add_text('objektify', 'V1') meta.add_text("obj_color", obj_color if obj_color is not None else "") meta.add_text("obj_color_picker", obj_color_picker if obj_color_picker is not None else "") meta.add_text("ai_num", str(ai_num) if str(ai_num) is not None else "") meta.add_text("txt_color", txt_color if txt_color is not None else "") meta.add_text("txt_color_picker", txt_color_picker if txt_color_picker is not None else "") meta.add_text("member_name_dropdown", member_name_dropdown if member_name_dropdown is not None else "") meta.add_text("group_name_radio", group_name_radio if group_name_radio is not None else "") meta.add_text("group_name_txt", group_name_txt if group_name_txt is not None else "") meta.add_text("obj_num", obj_num if obj_num is not None else "") meta.add_text("obj_alphabet", obj_alphabet if obj_alphabet is not None else "") meta.add_text("obj_alphabet_txt", obj_alphabet_txt if obj_alphabet_txt is not None else "") meta.add_text("obj_serial", obj_serial if obj_serial is not None else "") match obj_color: case "Atom01": obj_color = "#ffdd00" front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "Binary01": obj_color = "#00ff01" front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "Cream01": obj_color = "#ff7477" front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "Divine01": obj_color = "#a226f5" front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "Ever01": obj_color = "#33ecfd" front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "Special": img.paste(side_special, (1039, 104), side_special) case "Premier": obj_color = "#2E3192" front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "OMA1": img = edimg.paste_img(img, side_oma1, (1016, 83)) case "OMA2": img = edimg.paste_img(img, side_oma2, (1016, 83)) case "Other color": obj_color = obj_color_picker front_side = edimg.color(side, obj_color) img.paste(front_side, (1039, 104), front_side) case "AI Color": front_side = edimg.color_ai(ai_num) img.paste(front_side, (1039, 0), front_side) match txt_color: case "White": txt_color_r = 255 txt_color_g = 255 txt_color_b = 255 txt_color_hex = "#FFFFFF" case "Black": txt_color_r = 0 txt_color_g = 0 txt_color_b = 0 txt_color_hex = "#000000" case _: txt_color_r, txt_color_g, txt_color_b = fn.hex_to_rgb(txt_color_picker) txt_color_hex = txt_color_picker if member_name_dropdown: img_name = edimg.txt(member_name_dropdown, txt_color_r, txt_color_g, txt_color_b, 61, 270, "l") img.paste(img_name, (1050, 161), img_name) match group_name_radio: case "tripleS": img_group_name = edimg.txt(group_name_radio, txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") img.paste(img_group_name, (1050, 1240), img_group_name) case "ARTMS": img_artms_side = edimg.color("resource/artms_side.png", txt_color_hex) img.paste(img_artms_side, (1081, 1371), img_artms_side) case "Other": img_group_name = edimg.txt(group_name_txt, txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") img.paste(img_group_name, (1050, 1240), img_group_name) group_name_radio = group_name_txt case None: pass case _: if os.path.isfile(f'/data/presets/{preset}/sideNameImage.png'): sideNameImage = edimg.color(f'/data/presets/{preset}/sideNameImage.png', txt_color_hex) #sideNameImage.show() img.paste(sideNameImage, (1058, 1640-sideNameImage.size[1]), sideNameImage) else: img_group_name = edimg.txt(group_name_radio, txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") img.paste(img_group_name, (1050, 1240), img_group_name) if obj_num != "" and obj_alphabet != "": if obj_alphabet == "Other": obj_alphabet = obj_alphabet_txt img_obj_num = edimg.txt(obj_num+obj_alphabet, txt_color_r, txt_color_g, txt_color_b, 75, 270, "r") img.paste(img_obj_num, (1050, 452), img_obj_num) if obj_serial: img_serial = edimg.serial(obj_serial, txt_color_r, txt_color_g, txt_color_b, 75) img.paste(img_serial, (1046, 871), img_serial) img_copy = deepcopy(img) img = edimg.rounded(img) krtime = fn.get_kr_time() img.save(f'/data/cache/objektify-{krtime}.png', pnginfo=meta) return f'/data/cache/objektify-{krtime}.png', img_copy sign_positions = { "SeoYeon": (31, 1068), "HyeRin": (0, 1103), "JiWoo": (34, 1101), "ChaeYeon": (1, 1053), "YooYeon": (79, 1100), "SooMin": (0, 1057), "NaKyoung": (0, 1116), "YuBin": (20, 1109), "Kaede": (26, 1109), "DaHyun": (42, 1104), "Kotone": (31, 1105), "YeonJi": (0, 1065), "Nien": (58, 1108), "SoHyun": (0, 1110), "Xinyu": (0, 1105), "Mayu": (49, 1056), "Lynn": (61, 1105), "JooBin": (84, 1118), "HaYeon": (39, 1167), "ShiOn": (170, 1105), "ChaeWon": (64, 1124), "Sullin": (41, 1146), "SeoAh": (52, 1113), "JiYeon": (65, 1129), "HeeJin": (38, 1148), "KimLip": (61, 1100), "JinSoul": (52, 1113), "Choerry": (46, 1022), "HaSeul": (62, 1131) } cream01 = Image.open("resource/cream01.png") atom01 = Image.open("resource/atom01.png") binary01 = Image.open("resource/binary01.png") divine01 = Image.open("resource/divine01.png") ever01 = Image.open("resource/ever01.png") back_ui = Image.open("resource/back_ui.png") outline_ui = Image.open("resources/outline.png") outline_ui_oma = Image.open("resources/outline_oma.png") back_ = Image.open("resources/back.png") right_modhaus = Image.open("resource/right_modhaus.png") img_qr_white_space = Image.new("RGB", (335, 335), 'white') def back(obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, group_name_radio, group_name_txt, member_name_dropdown, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial, obj_outline_color, obj_outline_color_picker, logo_radio, class_radio, class_txt, season_radio, season_txt, season_txt_outline, sign, qr_url_dropdown, qr_Logo_radio, right_btn, preset, qr_caption): match obj_color: case 'OMA1': frame = outline_ui_oma case 'OMA2': frame = outline_ui_oma case _: frame = outline_ui match obj_outline_color: case "White": outline = edimg.color(frame, "#FFFFFF") case "Black": outline = edimg.color(frame, "#000000") case _: outline = edimg.color(frame, obj_outline_color_picker) match obj_color: case "Atom01": obj_color = "#ffdd00" img = edimg.color(back_, obj_color) case "Binary01": obj_color = "#00ff01" img = edimg.color(back_, obj_color) case "Cream01": obj_color = "#ff7477" img = edimg.color(back_, obj_color) case "Divine01": obj_color = "#a226f5" img = edimg.color(back_, obj_color) case "Ever01": obj_color = "#33ecfd" img = edimg.color(back_, obj_color) case "Special": img = Image.open('resources/back_special.png') case "Premier": obj_color = "#2E3192" img = edimg.color(back_, obj_color) case "OMA1": img = Image.open('resources/back_oma1.png') case "OMA2": img = Image.open('resources/back_oma2.png') case "Other color": obj_color = obj_color_picker img = edimg.color(back_, obj_color) case "AI Color": img = edimg.color_ai_back(ai_num) case _: img = edimg.color(back_, '#ffffff') img.paste(outline, (0, 0), outline) match txt_color: case "White": txt_color_r = 255 txt_color_g = 255 txt_color_b = 255 txt_color_hex = "#FFFFFF" case "Black": txt_color_r = 0 txt_color_g = 0 txt_color_b = 0 txt_color_hex = "#000000" case _: txt_color_r, txt_color_g, txt_color_b = fn.hex_to_rgb(txt_color_picker) txt_color_hex = txt_color_picker back_ui_colored = edimg.color(back_ui, txt_color_hex) img.paste(back_ui_colored, (0, 0), back_ui_colored) match logo_radio: case "tripleS": logo = edimg.color('resource/triples_logo.png', txt_color_hex) img.paste(logo, (61, 162), logo) case "ARTMS": logo = edimg.color('resource/artms_logo.png', txt_color_hex) img.paste(logo, (61, 162), logo) case 'None': pass case _: logo = edimg.color(f'/data/presets/{preset}/top_logo.png', txt_color_hex) width, height = logo.size # Calculate the new width to maintain the aspect ratio new_height = 126 aspect_ratio = width / height new_width = int(new_height * aspect_ratio) # Resize the image logo = logo.resize((new_width, new_height), Image.Resampling.LANCZOS) img.paste(logo, (61, 162), logo) if member_name_dropdown: img_name = edimg.txt(member_name_dropdown, txt_color_r, txt_color_g, txt_color_b, 130, 0, "l", "font/Helvetica_Neue_LT_Std_65_Medium.otf", (800, 136)) img.paste(img_name, (54, 466), img_name) img_name_side = edimg.txt(member_name_dropdown, txt_color_r, txt_color_g, txt_color_b, 61, 270, "l") img.paste(img_name_side, (940, 159), img_name_side) if obj_num: if obj_alphabet == "Other": obj_alphabet = obj_alphabet_txt img_obj_num_side = edimg.txt(obj_num+obj_alphabet, txt_color_r,txt_color_g,txt_color_b, 75, 270, "r") img.paste(img_obj_num_side, (951, 448), img_obj_num_side) if obj_serial: img_obj_serial_side = edimg.serial(obj_serial, txt_color_r,txt_color_g,txt_color_b, 75) img.paste(img_obj_serial_side, (943, 867), img_obj_serial_side) match group_name_radio: case "tripleS": img_group_name_side = edimg.txt("tripleS", txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") img.paste(img_group_name_side, (943, 1238), img_group_name_side) case "ARTMS": img_group_name_side = edimg.color("resource/artms_side.png", txt_color_hex) img.paste(img_group_name_side, (974, 1371), img_group_name_side) case "Other": img_group_name_side = edimg.txt(group_name_txt, txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") img.paste(img_group_name_side, (943, 1238), img_group_name_side) group_name_radio = group_name_txt case None: pass case _: if os.path.isfile(f'/data/presets/{preset}/sideNameImage.png'): sideNameImage = edimg.color(f'/data/presets/{preset}/sideNameImage.png', txt_color_hex) sideNameImage.show() img.paste(sideNameImage, (951, 1638-sideNameImage.size[1]), sideNameImage) else: img_group_name = edimg.txt(group_name_radio, txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") img.paste(img_group_name, (943, 1238), img_group_name) #img_group_name_side = edimg.txt(group_name_radio, txt_color_r, txt_color_g, txt_color_b, 61, 270, "r") #img.paste(img_group_name_side, (943, 1238), img_group_name_side) group_name_radio = group_name_txt match class_radio: case "Zero" | "First" | "Double" | "Special" | "Premier": img_class = edimg.txt(class_radio, txt_color_r, txt_color_g, txt_color_b, 130, 0, "l", "font/Helvetica_Neue_LT_Std_65_Medium.otf", (800, 136)) img.paste(img_class, (54, 700), img_class) case "Other": img_class = edimg.txt(class_txt, txt_color_r, txt_color_g, txt_color_b, 130, 0, "l", "font/Helvetica_Neue_LT_Std_65_Medium.otf", (800, 136)) img.paste(img_class, (54, 700), img_class) match season_radio: case "Atom01": img_atom01 = edimg.color(atom01, txt_color_hex) img.paste(img_atom01, (63, 951), img_atom01) case "Binary01": img_binary01 = edimg.color(binary01, txt_color_hex) img.paste(img_binary01, (63, 951), img_binary01) case "Cream01": img_cream01 = edimg.color(cream01, txt_color_hex) img.paste(img_cream01, (62, 950), img_cream01) case "Divine01": img_divine01 = edimg.color(divine01, txt_color_hex) img.paste(img_divine01, (61, 951), img_divine01) case "Ever01": img_ever01 = edimg.color(ever01, txt_color_hex) img.paste(img_ever01, (61, 951), img_ever01) case "Other": img_season = edimg.txt(season_txt, txt_color_r, txt_color_g, txt_color_b, 130, 0, "l", "font/Helvetica_Neue_LT_Std_65_Medium.otf", (1100, 136)) x = edimg.calculate_size(season_txt, 130, "font/Helvetica_Neue_LT_Std_65_Medium.otf") img_season_outline = edimg.txt(season_txt_outline, txt_color_r, txt_color_g, txt_color_b, 135, 0, "l", "font/Helvetica_Neue_LT_Std_75_Bold_Outline.otf", (1100, 136)) img.paste(img_season,(53, 936), img_season) img.paste(img_season_outline, (53 + x, 936), img_season_outline) if sign in pre.names: colored_icon = edimg.color(f'sign/{sign}.png', txt_color_hex) position = sign_positions[sign] img.paste(colored_icon, position, colored_icon) elif sign: colored_icon = edimg.color(f'/data/presets/{preset}/{sign}.png', txt_color_hex) original_width, original_height = colored_icon.size ratio = min(480 / original_width, 330 / original_height) new_width = int(original_width * ratio) new_height = int(original_height * ratio) colored_icon = colored_icon.resize((new_width, new_height), Image.Resampling.LANCZOS) img = maker.space(img, (63, 1097), (488, 335), colored_icon) if qr_url_dropdown: url_txt = fn.url_mapping(qr_url_dropdown) img_qr = edimg.qr(url_txt) match qr_Logo_radio: case "tripleS": img_qr_icon = edimg.qr_icon("resource/qr_icon_triples.png") img_qr.paste(img_qr_icon, (126, 126), img_qr_icon) case 'Cosmo': img_qr_icon = edimg.qr_icon("resources/qr_icon_cosmo.png") img_qr.paste(img_qr_icon, (126, 126), img_qr_icon) case 'None': pass case _: img_qr_icon = Image.open(f'/data/presets/{preset}/qr_logo.png') colored_icon = img_qr_icon original_width, original_height = colored_icon.size ratio = min(84 / original_width, 84 / original_height) new_width = int(original_width * ratio) new_height = int(original_height * ratio) colored_icon = colored_icon.resize((new_width, new_height), Image.Resampling.LANCZOS) qr_logo_bg = Image.new(size=(84, 84), mode='RGB', color=(255,255,255)) qr_logo_bg = maker.space(qr_logo_bg, (0,0), (84,84), colored_icon) img_qr.paste(qr_logo_bg, (126, 126)) img_qr = edimg.resize_img(img_qr, 335,335) img.paste(img_qr, (555, 1098)) else: img.paste(img_qr_white_space,(555, 1098)) if qr_caption: qr_caption_img = maker.draw(36, qr_caption, color=txt_color_hex) img = maker.space(img, (553, 1453), (339,37), qr_caption_img) if right_btn: img_right_modhaus = edimg.color(right_modhaus, txt_color_hex) img.paste(img_right_modhaus, (61,1604), img_right_modhaus) krtime = fn.get_kr_time() img_jpeg = img.convert('RGB') img_jpeg.save(f'/data/cache/{member_name_dropdown}_{group_name_radio}_{class_radio}_{obj_num}{obj_alphabet}_{krtime}.jpg', quality=90) return img, img make_image_first = Image.open("resource/make_image_first.png") remove_black = Image.open("resource/remove_black.png") def ex_gen(img_front, file_front_path, img_back, count, blur, blur_strength, watermark): if img_back == None: gr.Warning("Make two images, front and back") return make_image_first, make_image_first if img_front == None and file_front_path == None: gr.Warning("Make two images, front and back") return make_image_first, make_image_first if file_front_path: file_front = Image.open(file_front_path) match count: case "One File": img_back = img_back.resize(file_front.size) img_remove_black = remove_black.resize(file_front.size) img_back.paste(img_remove_black,(0,0), img_remove_black) filename, _ = os.path.splitext(os.path.basename(file_front_path)) edimg.makegif(file_front, img_back) return f"exgif.gif", img_back case "Two Files": return file_front, img_back if blur: img_front = img_front.filter(ImageFilter.GaussianBlur(blur_strength)) img_back = img_back.filter(ImageFilter.GaussianBlur(blur_strength)) if watermark: img_txt = edimg.txt(watermark, 255, 255, 255, 210, 50, "l", "font/Helvetica_Neue_LT_Std_75_Bold.otf",(1338, 222), 120) img_front.paste(img_txt, (65, 333), img_txt) img_back.paste(img_txt, (65, 333), img_txt) match count: case "One File": img = Image.new("RGBA", (2328, 1800), (0, 0, 0, 0)) img_front = edimg.rounded(img_front) img_back = edimg.rounded(img_back) img.paste(img_front, (0, 0), img_front) img.paste(img_back, (1164, 0), img_back) return img, img_blank case "Two Files": img_front = edimg.rounded(img_front) img_back = edimg.rounded(img_back) return img_front, img_back def find_value(n, percent): return n * (percent / 100.0)