Update src/pyscripts/img_process.py

src/pyscripts/img_process.py

@@ -1,92 +1,377 @@
-from fastapi import FastAPI, Path, Query, Response, BackgroundTasks
-from fastapi.responses import FileResponse, HTMLResponse
-import gradio as gr
-from src.pyscripts import file_process, oss, img_process
-from datetime import datetime
-from collections import namedtuple
-from pydantic import BaseModel
-
-Person = namedtuple('Person', ['name', 'title', 'category', 'conference', 'respond_person'])
-
-CUSTOM_PATH = "/gradio"
-
-app = FastAPI()
-
-pdf_path = "src/assets/InvitationTemplate.pdf"
-output_path = 'output/output.pdf'
-person = Person(name="雷军", title="先生", category="观礼", conference="世界教育者大会", respond_person="Eazy")
-
-# data = {
-#     'name': '雷军',
-#     'title': '先生',
-#     "conference": "世界教育者大会"
-# }
-
-
-@app.get("/")
-def read_main():
-    return {"message": "This is your main app"}
-
-# @app.post("/getInvitationPDF/{name}/{title}/{conference}")
-@app.get("/getInvitationPDF")
-def getInvitationPDF(name, title, conference, category, respond_person):
-    person = Person(name=name, title=title, category=category, conference=conference, respond_person=respond_person)
-    formatted_date = datetime.now().strftime('%Y-%m-%d')
-    file_name = f"致{name+title}的WWEC2024邀请函.pdf"
-    file_dir = f"invitation/{category}"
-    object_name = file_dir + "/" + file_name
-    pdfStream = file_process.writeInvitationPDF(person)
-    # with open(f"output/{file_name}", 'wb') as f:
-    #     f.write(pdfStream.getvalue())
-    oss.upload_file_stream_to_s3(pdfStream, "host", object_name)
-    file_url = f"https://isidoresong.us.kg/{object_name}"
-    return {"url": file_url}
-
-class GuestPosterInfo(BaseModel):
-    exhibitor: str
-    logo_link: str
-    slogan: str
-    content: str
-
-@app.post("/getExhibitorPoster")
-def getExhibitorPoster(item: GuestPosterInfo):
-    logo_link = item.logo_link
-    exhibitor = item.exhibitor
-    slogan = item.slogan
-    content = item.content
-    img_stream = img_process.make_exhibitor_poster(logo_link, slogan, content)
-    file_dir = "ExhibitorPoster"
-    object_name = file_dir + "/" + exhibitor + ".jpg"
-    # with open(f"output/{exhibitor + '.jpg'}", 'wb') as f:
-    #     f.write(img_stream.getvalue())
-    oss.upload_file_stream_to_s3(img_stream, "host", object_name)
-    file_url = f"https://isidoresong.us.kg/{object_name}"
-    return {"url": file_url}
-
-
-class GuestPosterInfo(BaseModel):
-    name: str
-    name_pinyin: str
-    photo_link: str
-    original_photo_link: str
-    titles: str
-
-@app.post("/getGuestPoster")
-def getGuestPoster(item: GuestPosterInfo):
-    photo_link = item.photo_link
-    original_photo_link = item.original_photo_link
-    name = item.name
-    name_pinyin = item.name_pinyin
-    titles = item.titles
-    img_stream = img_process.make_guest_poster(name, name_pinyin, photo_link, original_photo_link, titles)
-    file_dir = "GuestPoster"
-    object_name = file_dir + "/" + name + ".jpg"
-    #     f.write(img_stream.getvalue())
-    oss.upload_file_stream_to_s3(img_stream, "host", object_name)
-    file_url = f"https://pub-429c75a96a8f4597984dd7ebc525d652.r2.dev/{object_name}"
-    return {"url": file_url}
-
-interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
-app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)
-
-# if __name__ == "__main__":
+from PIL import Image, ImageDraw, ImageFont
+import re
+import os
+import io
+from dataclasses import dataclass
+import requests
+from pypinyin import pinyin, Style
+import rembg
+
+
+font_path_fangzhenghei = "src/assets/FangZhengHeiTiJianTi-1.ttf"
+font_path_zihun59 = "src/assets/字魂59号-创粗黑.ttf"
+font_path_notosans = "src/assets/NotoSansHans-Regular.otf"
+font_path_feihuasong = "src/assets/FeiHuaSongTi-2.ttf"
+font_path_wendingPLjianbaosong = "src/assets/gbsn00lp-2.ttf"
+font_path_SourceHanSans = "src/assets/SourceHanSansCN-VF-2.otf"
+
+
+
+
+@dataclass
+class text_info:
+    text: str
+    font_size: int
+    position: list
+    color: tuple= (0,0,0)
+    max_height: int=1000
+    max_width: int=0
+    line_spacing: int = 10
+    font_path: str = None
+    em_font_path: str = None
+
+
+    def __post_init__(self):
+        self.wrap_text()
+
+
+    def wrap_text(self):
+        """
+        resize to fix max_height and max_width
+        """
+        font = ImageFont.truetype(self.font_path, self.font_size)
+        words = re.findall(r'[a-zA-Z\d]+|[\u4e00-\u9fa5]|\n|.', self.text)
+        lines = []
+        current_line = ''
+        for word in words:
+            if word == '\n':
+                lines.append(current_line.strip())
+                current_line = ''
+                continue
+            test_line = current_line + word
+            test_bbox = font.getbbox(test_line, anchor="la")
+            test_width = test_bbox[2] - test_bbox[0]  # 计算宽度
+            if test_width <= self.max_width and '\n' not in test_line:
+                current_line = test_line
+            else:
+                lines.append(current_line.strip())
+                current_line = word
+        if current_line:
+            lines.append(current_line.strip())
+        self.lines = lines
+        self.height = len(lines) * self.font_size + (len(lines) - 1) * self.line_spacing
+        key_lines = [line for line in lines if line.startswith("【】")]
+        if len(key_lines) > 0:
+            key_line = max(key_lines, key=len)
+        else:
+            key_line = False
+        if self.height > self.max_height or (key_line and not re.search("^"+re.escape(key_line)+"$", self.text, re.M)):
+            self.font_size -= 2
+            self.wrap_text()
+
+
+def write_text_on_image(img, text_info):
+    # 创建一个可以在给定图像上绘图的对象
+    draw = ImageDraw.Draw(img)
+    # 加载字体，如果没有提供字体路径，则使用系统默认字体
+    font = ImageFont.load_default().font_variant(size=text_info.font_size)
+    em_font = ImageFont.load_default().font_variant(size=text_info.font_size)
+    if text_info.font_path:
+        font = ImageFont.truetype(text_info.font_path, text_info.font_size)
+    if text_info.em_font_path:
+        em_font = ImageFont.truetype(text_info.em_font_path, text_info.font_size)
+    # 在图片上写入文字
+    # draw.text(text_info.position, text_info.text, fill=text_info.color, font=font)
+    y_text = text_info.position[1]
+    wrapped_lines = text_info.lines
+    text_bbox_chinese = font.getbbox("中文行高")
+    for line in wrapped_lines:
+        text_bbox = font.getbbox(line)
+        line_width = text_bbox[2] - text_bbox[0]  # 获取宽度
+        line_height = max(text_bbox[3] - text_bbox[1], text_bbox_chinese[3] - text_bbox_chinese[1])
+        # print(text_bbox[3] - text_bbox[1], text_bbox_chinese[3] - text_bbox_chinese[1], text_info.font_size)
+        if re.search(r"^【.*】.*?"+re.escape(line), text_info.text, re.M) or re.match(r"^【.*】.*?$", line):
+            draw.text((text_info.position[0], y_text), line.replace("【】", ""), fill=text_info.color, font=em_font)
+        else:
+            draw.text((text_info.position[0], y_text), line, fill=text_info.color, font=font)
+        # 更新y坐标以添加行间距并为下一行做准备
+        y_text += line_height + text_info.line_spacing
+    return img, y_text
+
+
+def embed_image(background, foreground, vertical_offset=0, standard_fg_size=None):
+    """
+    粘贴图片
+
+    Args:
+        background (_type_): 背景图片
+        foreground (_type_): 前景图片
+        vertical_offset (int, optional): 纵向偏移. Defaults to 1500.
+        standard_fg_size (tuple, optional): 前景缩放大小. Defaults to (1600, 450).
+
+    Returns:
+        _type_: _description_
+    """
+    # 获取背景图片的尺寸
+    if standard_fg_size is None:
+        standard_fg_size = background.size
+    standard_fg_width, standard_fg_height = standard_fg_size
+    bg_width, bg_height = background.size
+    fg_width, fg_height = foreground.size
+    fg_resize_percent = max(fg_width / standard_fg_width, fg_height / standard_fg_height)
+    # print((fg_width / standard_fg_width, fg_height / standard_fg_height))
+    foreground = foreground.resize((int(fg_width // fg_resize_percent), int(fg_height // fg_resize_percent)))
+    fg_width, fg_height = foreground.size
+    offset_x = (bg_width - fg_width) // 2  # 横轴偏移使小图居中
+    offset_y = vertical_offset  # 纵轴固定偏移300px
+
+    background.paste(foreground, (offset_x, offset_y), foreground)
+    return background
+
+def make_exhibitor_poster(logo_link, slogan, content):
+    background = Image.open("src/assets/展商海报白板.jpg")
+    H_PADDING_RATION = 0.08
+    V_PADDING = 25
+    V_START = 950
+    MAX_HEIGHT = 520
+    res = requests.get(logo_link)
+    foreground = Image.open(io.BytesIO(res.content))
+    # foreground = Image.open(logo_path_list[FOREGROUND_ID])
+    background = embed_image(background, foreground, vertical_offset=750, standard_fg_size=(1600//2, 450//2))
+    slogan = text_info(text=slogan, font_size=110//2,
+        position=[background.size[0] * H_PADDING_RATION, V_START],
+        max_width=background.size[0]*(1-2*H_PADDING_RATION),
+        max_height=110//2,
+        font_path=font_path_fangzhenghei)
+    slogan.position[0] = (background.size[0] - len(slogan.text) * slogan.font_size) // 2
+    background, next_height = write_text_on_image(background, slogan)
+
+    content = text_info(text=content, font_size=100,
+        position=[background.size[0] * H_PADDING_RATION, next_height+V_PADDING],
+        max_width=background.size[0]*(1-2*H_PADDING_RATION),
+        max_height=MAX_HEIGHT,
+        font_path=font_path_SourceHanSans, em_font_path=font_path_fangzhenghei)
+    print(content.height)
+    background, next_height = write_text_on_image(background, content)
+    img_stream = io.BytesIO()
+    background.save(img_stream, format="JPEG")
+    img_stream.seek(0)
+    background.save("output/展商海报.jpg")
+    return img_stream
+
+
+def crop_image(image):
+    # 获取图片的宽度和高度
+    width, height = image.size
+    # 初始化边界框
+    left = width
+    top = height
+    right = 0
+    bottom = 0
+    # 遍历图片的每个像素，找到非透明的区域
+    for x in range(width):
+        for y in range(height):
+            pixel = image.getpixel((x, y))
+            if pixel[3] != 0:  # 检查透明度通道
+                if x < left:
+                    left = x
+                if x > right:
+                    right = x
+                if y < top:
+                    top = y
+                if y > bottom:
+                    bottom = y
+    cropped_image = image.crop((left, top, right + 1, bottom + 1))
+    return cropped_image
+
+
+def rm_img_bg(image):
+    image = rembg.remove(image)
+    cropped_image = crop_image(image)
+    return cropped_image
+
+def get_text_width(font_path, font_size, text):
+    font = ImageFont.truetype(font_path, font_size)
+    test_bbox = font.getbbox(text, anchor="la")
+    test_width = test_bbox[2] - test_bbox[0]
+    return test_width
+def compose_guest_poster(name, name_pinyin, guest_photo, titles):
+
+    background = Image.open("src/assets/Guest04.jpg").convert("RGBA")
+    background = background.resize((1080*2, 2400*2))
+
+    title_list = titles.split("\n")
+    # guest = Image.open("src/assets/Guest.png").convert("RGBA")
+    # higher_mask = Image.open("src/assets/Guest01.png").convert("RGBA")
+    H_PADDING_RATION = 0.08
+    V_PADDING = 5
+    V_START = 950
+    MAX_HEIGHT = 520
+    # foreground = Image.open(logo_path_list[FOREGROUND_ID])
+    bg_width, bg_height = background.size
+    background = embed_image(background, guest_photo, vertical_offset=1300, standard_fg_size=(2160, 2600))
+    lower_mask = Image.open("src/assets/Guest03.png").convert("RGBA")
+    background = embed_image(background, lower_mask)
+    higher_mask = Image.open("src/assets/Guest02.png").convert("RGBA")
+    background = embed_image(background, higher_mask)
+    seal = Image.open("src/assets/Guest01.png").convert("RGBA")
+    background = embed_image(background, seal)
+    # background = background.convert("RGB")
+    
+    font_size=210
+    name, background, next_height = write_text(name, font_path_zihun59, background, H_PADDING_RATION, 3020, font_size)
+
+    next_height += 50
+    font_size = 100
+    if name_pinyin == "":
+        pinyin_list = pinyin(name.text, style=Style.NORMAL)
+        name_pinyin = (pinyin_list[0][0]+' ' + "".join([item[0] for item in pinyin_list[1:]])).title()
+    name_pinyin, background, next_height = write_text(name_pinyin, font_path_notosans, background, H_PADDING_RATION,next_height+V_PADDING, font_size)
+    
+
+    next_height += 100
+    font_size = 70
+    for title in title_list:
+        title, background, next_height = write_text(title, font_path_notosans, background, H_PADDING_RATION,next_height+V_PADDING, font_size)
+
+    background = background.convert("RGB")
+    img_stream = io.BytesIO()
+    background.save(img_stream, format="JPEG")
+    img_stream.seek(0)
+    if os.path.exists("src/assets/output"):
+        background.save("src/assets/output/嘉宾海报.jpg")
+    return img_stream
+
+def make_guest_poster(name, name_pinyin, photo_link, original_photo_link, titles):
+    if photo_link != '':
+        res = requests.get(photo_link)
+        guest_photo = Image.open(io.BytesIO(res.content)).convert("RGBA")
+    elif original_photo_link != '':
+        res = requests.get(original_photo_link)
+        original_guest_photo = Image.open(io.BytesIO(res.content))
+        guest_photo = rm_img_bg(original_guest_photo)
+        if os.path.exists("src/assets/output"):
+            guest_photo.save("src/assets/output/rmbg.png")
+    else:
+        return None
+    img_stream = compose_guest_poster(name, name_pinyin, guest_photo, titles)
+    return img_stream
+
+def write_text(text, font_path, image, H_PADDING_RATION, start_height, font_size):
+    bg_width = image.size[0]
+    text_width = get_text_width(font_path, font_size, text)
+    text = text_info(text=text, font_size=font_size,
+        position=[bg_width*(1-H_PADDING_RATION) - text_width, start_height],
+        color=(255,255,255),
+        max_width=bg_width*(1-2*H_PADDING_RATION),
+        font_path=font_path)
+    image, next_height = write_text_on_image(image, text)
+    return text,image,next_height
+    # return img_stream
+
+
+
+img_path_dict = {
+    "old": "src/assets/InvitationTemplateCompressed.pdf",
+    "common": "src/assets/旧空白邀请函图片.jpg",
+    "观礼": "src/assets/带名字.jpg",
+    "演讲": "src/assets/带名字和会.jpg",
+    "致辞": "src/assets/带名字和会.jpg",
+}
+font_path = "src/assets/ZhouZiSongTi7000Zi-2.otf"
+font_path = "src/assets/FangZhengHeiTiJianTi-1.ttf"
+
+def writeInvitationPDF(person, debug=False):
+    "旧空白邀请函图片.jpg"
+    font_size=40
+    resized_name_font_size = font_size*5/len(person.name+person.title)
+    resized_conference_font_size = font_size*9/len(person.conference)
+    img_path = img_path_dict[person.category]
+    image = Image.open(img_path)
+    draw = ImageDraw.Draw(image)
+    text_color = (255, 255, 255)  # 白色 RGB
+    
+    draw.text((182, 150+max(0, font_size-resized_name_font_size)), person.name+person.title, fill=text_color, font=ImageFont.truetype(font_path, min(font_size, resized_name_font_size)))
+    draw.text((102, 532+max(0, font_size-resized_conference_font_size)), person.conference, fill=text_color, font=ImageFont.truetype(font_path, min(font_size, resized_conference_font_size)))
+    imgStream = io.BytesIO()
+    # image.save(imgStream)
+    # doc.save(pdfStream)
+    # doc.close()
+    if debug:
+        image.save("output/image_with_text.jpg")
+
+    return imgStream
+
+
+
+
+if __name__ == "__main__":
+    # image_path = 'src/assets/邀请函封面.jpg'  # 替换为你的图片文件路径
+    # text_to_write = "你好，世界！"
+    # font_path = "src/assets/FangZhengHeiTiJianTi-1.ttf"
+    # position = (120, 140)
+    # write_text_on_image(image_path, text_to_write, position=position, font_path=font_path)
+    from collections import namedtuple
+    Person = namedtuple('Person', ['name', 'title', 'category', 'conference', 'respond_person'])
+    person = Person(name="一二三四五六", title="先生", category="演讲", conference="8月21日中欧智慧论坛，做主题演讲", respond_person="Eazy")
+    output_path = 'output/output.pdf'
+    # writeInvitationPDF(person, debug=True)
+
+
+    text_list = ["""Shanghai 协砺科技有限公司专注于青少年科技教育12年，通过STEAM25教育模式培养青少年的科学思维和创新能力。在全国设有数十家学习中心，服务上万名学员，提供科创、机器人和编程课程。公司还设有赛事研究中心，由专业工程师团队提供赛事研究和策略支持。学员在各类科技赛事中屡获佳绩。协砺科技致力于推动中国科技教育的创新与发展，为国家繁荣贡献力量。
+
+【主要产品】科创、编程、赛事
+【核心特点】为青少年提供优质的科创学习规划。
+【核心亮点】高效提供更优质的赛事解决方案——快速赋能当地传统教育企业转型智能化教育赛道。""",
+"""天天乐学创办于2015年6月，总部位于北京，创始团队均来自清北等知名高校，旗下产品天天乐学定制app，主打立体化教学和趣味性学习，可深度匹配教学需求，实现个性化教学。目前合作校近三万家，遍布全国31个省，300座城市，是国内知名的教育科技公司。
+
+
+【主要产品】独立教师·机构·绘本馆英语APP定制
+【核心特点】内容深度定制,帮助机构实现线上线下联动教学
+""",
+"""【】星瀚之光是明德传承集团旗下文化传媒品牌
+成立于2023年,立足深圳,布局全国九大中心城市,是国内首家定位于价值观驱动的P产业链开发的MCN机构,拥有知识P孵化事业群和商业P孵化事业群两大主营板块,旗下涵盖了创始人P孵化直播电商、私域消费电商、企业整合营销、艺人经纪等业务,致力于成为中国规模最大的P生态搭建领军企业。
+
+【】坚守“自利利他、奋斗为本、价值为纲、成就客户”的企业价值观
+公司成立以来发展迅猛，迅速孵化出各大赛道的多个头部P，霸榜各大短视频平台榜单，包括：有家庭教育赛道:白瑞(家庭教育赛道头部主播)；身心灵赛道：尚致胜(新中式心理学首创者)；女性成长赛道：黄欢；商业赛道：苏建诚(亚细亚集团联合创始人)、王怀南(宝宝树创始人)、杨晓燕(长江商学院助理院长)、何慕(联众智达董事长)等；国学赛道：沈德斌(了凡四训学者)、钱锦国老师等。""",
+"""学趋坊是一家为学校、机构、智能自习室等提供在线化和智能化解决方案的数字化教育内容供应商。已经为多个知名的在线教育品牌、智能自习室品牌提供0DM和OEM服务,产品用户累计超过1000万。
+
+【主要产品】智能校本
+【核心特点】为本地化内容插上科技的翅膀。
+【核心亮点】高质量高效本地化教研—快速赋能当地传统教育企业转型智能化教育赛道。"""
+    ]
+    logo_path_list = [
+        "src/assets/logo/学区房logo.png",
+        "src/assets/logo/星瀚之光logo.jpg",
+        "src/assets/logo/软科起点logo.png",
+        "src/assets/logo/天天乐学logo.png",
+    ]
+    slogan_list = [
+        "探索科技，启迪未来。",
+        "价值观驱动的知识P全域孵化领先平台",
+        "定制英语APP学习系统"
+    ]
+    FOREGROUND_ID = 3 # 0, 1, 2, 3
+    SLOGAN_ID = 1 # 0, 1, 2
+    TEXT_ID = 2 # 0, 1, 2, 3
+
+    LOGO_LINK = "https://weboffice-temporary.ks3-cn-beijing.wpscdn.cn/thumbnail/tJUjbgWdD6R5k5rECKka1yK_100.png?Expires=1720890000&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=IxmcUo8eX3VzsRltml9F8AQ5%2FDY%3D&response-cache-control=public%2Cmax-age%3D86400"
+    # make_exhibitor_poster(LOGO_LINK, slogan_list[SLOGAN_ID], text_list[TEXT_ID])
+    name, titles = "王千里", "全国政协文员\n好好先生"
+    PHOTO_LINK = "https://weboffice-temporary.ks3-cn-beijing.wpscdn.cn/thumbnail/tpcdnbPEKEJ3ge831dSHEd3_100.png?Expires=1720890000&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=jeKkEy%2BTx3BAMNzqrmjUbg6uBK8%3D&response-cache-control=public%2Cmax-age%3D86400"
+    ORIGINAL_PHOTO_LINK = "https://weboffice-temporary.ks3-cn-beijing.wpscdn.cn/thumbnail/tZCeTRiNFQbksEEF4WMR8ZX_100.jpeg?Expires=1721070000&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=Hc2iv3U3EBQ86xqQLztLqdU3zuk%3D&response-cache-control=public%2Cmax-age%3D86400"
+    
+    
+    # res = requests.get(ORIGINAL_PHOTO_LINK)
+    # # guest_photo = Image.open(io.BytesIO(res.content)).convert("RGBA")
+    # original_guest_photo = Image.open(io.BytesIO(res.content)).convert("RGBA")
+    # guest_photo = rm_img_bg(original_guest_photo)
+    
+    # compose_guest_poster(name, guest_photo, titles)
+
+    make_guest_poster(name, "", "", ORIGINAL_PHOTO_LINK, titles)
+    # for file in os.listdir("src/assets/GuestPoster/original_guest"):
+    #     image = Image.open(os.path.join("src/assets/GuestPoster/original_guest", file))
+    #     image = rm_img_bg(image)
+    #     image.save(os.path.join("src/assets/output", file.replace("jpg", "png")))
+