Update src/pyscripts/img_process.py

src/pyscripts/img_process.py

@@ -1,179 +1,966 @@
-from fastapi import FastAPI, Path, Query, Response, BackgroundTasks
-from fastapi.responses import FileResponse, HTMLResponse
-import gradio as gr
-import sys
+from PIL import Image, ImageDraw, ImageFont
+import re
 import os
-project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
-sys.path.append(project_root)
-from src.pyscripts import file_process, oss, img_process
-from datetime import datetime
-from collections import namedtuple
-from pydantic import BaseModel
+import io
+from dataclasses import dataclass
 import requests
-# from easydict import EasyDict
-
-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}
-
-class ConferenceInfo(BaseModel):
-    name: str
-    eng_name: str
-    full_name: str
-    start_time: str
-    end_time: str
-    member: list
-    guest: list = None
-
-class ConferencePosterInfo(BaseModel):
-    conference_info: ConferenceInfo
-    conference_rundown:list
-
-
-def format_time(start_time, end_time):
-    start_datetime = datetime.strptime(start_time, "%Y/%m/%d %H:%M:%S")
-    end_datetime = datetime.strptime(end_time, "%Y/%m/%d %H:%M:%S")
-    
-    start_period = "上午" if start_datetime.hour < 12 else "下午"
-    end_period = "上午" if end_datetime.hour < 12 else "下午"
-    
-    formatted_start_time = start_datetime.strftime(f"%Y年%m月%d日{start_period}%H:%M")
-    formatted_end_time = end_datetime.strftime(f"{end_period}%H:%M")
-    
-    return f"{formatted_start_time}-{formatted_end_time}"
-
-
-@app.post("/getConferencePoster")
-def getConferencePoster(item: ConferencePosterInfo):
-    # item.conference_rundown = [list(i) for i in item.conference_rundown]
-    sorted_conference_rundown = sorted(item.conference_rundown, key=lambda x: x['end_time'])
-    img_stream = img_process.make_conference_poster(
-        item.conference_info["name"],
-        item.conference_info["eng_name"],
-        item.conference_info["full_name"],
-        [
-            *item.conference_info["member"], 
-            ["论坛时间", format_time(item.conference_info["start_time"], 
-                              item.conference_info["end_time"])],
-            ["论坛地点", "国家会展中心4.2号馆 " + item.conference_info["address"]]
-        ],
-        sorted_conference_rundown)
-    file_dir = "ConferencePoster"
-    object_name = file_dir + "/" + item.conference_info["name"] + ".jpg"
+from pypinyin import pinyin, Style
+import rembg
+import math
+
+
+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"
+font_path_notosansM = "src/assets/NotoSansHans-Medium.otf"
+font_path_notosansB = "src/assets/NotoSansHans-Bold.otf"
+font_path_notosansL = "src/assets/NotoSansHans-Light.otf"
+font_path_pingfang = "src/assets/PingFang.ttc"
+font_path_pingfangM = "src/assets/pingfang-M.ttf"
+
+
+@dataclass
+class text_info:
+    text: str
+    font_size: int
+    position: list
+    color: tuple = (0, 0, 0)
+    max_height: int = 1000
+    max_width: int = 0
+    text_width: int = 0
+    text_height: int = 0
+    line_space: int = 10
+    font_path: str = None
+    em_font_path: str = None
+
+    def __post_init__(self):
+        self.wrap_text()
+        self.get_text_width()
+        self.get_text_height()
+
+    def wrap_text(self):
+        """
+        resize to fix max_height and max_width
+        """
+        words = re.findall(r"[a-zA-Z\d]+|[\u4e00-\u9fa5]|\n|.", self.text)
+        self.font = ImageFont.truetype(self.font_path, self.font_size)
+
+        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 = self.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
+        test_bbox = self.font.getbbox("\n".join(lines), anchor="la")
+        self.text_height = test_bbox[3] - test_bbox[1]
+        self.text_width = test_bbox[2] - test_bbox[0]  # 计算宽度
+        # print(lines, self.text_height, self.text_width)
+        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.text_height > self.max_height or self.text_width > self.max_width or (key_line and not re.search("^" + re.escape(key_line) + "$", self.text, re.M)):
+            self.font_size -= 2
+            self.wrap_text()
+
+    def get_text_width(self):
+        text_bbox = self.font.getbbox(self.text, anchor="la")
+        self.text_width = text_bbox[2] - text_bbox[0]
+
+    def get_text_height(self):
+        text_bbox = self.font.getbbox(self.text, anchor="la")
+        self.text_height = text_bbox[3] - text_bbox[1]
+
+
+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_space
+    return img, y_text
+
+
+def embed_image(background, foreground, offset_y=0, offset_x=None, standard_fg_size=None):
+    """
+    粘贴图片
+
+    Args:
+        background (_type_): 背景图片
+        foreground (_type_): 前景图片
+        offset_y (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 = foreground.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
+    if offset_x is None:
+        offset_x = (bg_width - fg_width) // 2  # 横轴偏移使小图居中
+
+    background.paste(foreground, (int(offset_x), int(offset_y)), foreground)
+    return background
+
+
+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
+
+
+def write_text(
+    text, font_path, image, H_PADDING_RATION, font_size, max_height=None, color=(255, 255, 255), em_font_path=None, alignment="left",
+    start_y=0, start_x=0, max_width=None, part_width=10000
+):
+    bg_width = image.size[0]
+    if max_width is None:
+        max_width = bg_width * (1 - 2 * H_PADDING_RATION)
+    if max_height is None:
+        max_height = font_size * len(text.split("\n"))
+    text = text_info(
+        text=text,
+        font_size=font_size,
+        position=[start_x, start_y],
+        color=color,
+        max_width=max_width,
+        max_height=max_height,
+        font_path=font_path,
+        em_font_path=em_font_path,
+    )
+    if alignment == "right":
+        text.position[0] = bg_width * (1 - H_PADDING_RATION) - text.text_width
+    elif alignment == "medium":
+        text.position[0] = (min(part_width, bg_width) - text.text_width) // 2 + start_x
+    elif alignment == "left":
+        text.position[0] = bg_width * H_PADDING_RATION
+    print(text)
+    image, next_height = write_text_on_image(image, text)
+    return text, image, next_height
+
+
+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, offset_y=750, standard_fg_size=(1600 // 2, 450 // 2))
+
+    text = slogan
+    font_size = 110 // 2
+    font_path = font_path_fangzhenghei
+    slogan, background, next_height = write_text(
+        text,
+        font_path,
+        background,
+        H_PADDING_RATION,
+        font_size,
+        color=(0, 0, 0),
+        alignment="medium",
+        start_y=V_START,
+    )
+
+    content, background, next_height = write_text(
+        content,
+        font_path_SourceHanSans,
+        background,
+        H_PADDING_RATION,
+        100,
+        max_height=MAX_HEIGHT,
+        color=(0, 0, 0),
+        em_font_path=font_path_fangzhenghei,
+        alignment="left",
+        start_y=next_height + V_PADDING,
+    )
+
+    img_stream = io.BytesIO()
+    background.save(img_stream, format="JPEG")
+    img_stream.seek(0)
+    background.save("output/展商海报.jpg")
+    return img_stream
+
+
+def cut_image_blank(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
+    if right < left or bottom < top:
+        return image
+    cropped_image = image.crop((left, top, right + 1, bottom + 1))
+    return cropped_image
+
+
+def rm_img_bg(image):
+    image = rembg.remove(image)
+    cropped_image = cut_image_blank(image)
+    return cropped_image
+
+
+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 = 3020
+    bg_width, bg_height = background.size
+    background = embed_image(background, guest_photo, offset_y=1300, standard_fg_size=(2160, 2600))
+    lower_mask = Image.open("src/assets/Guest03.png").convert("RGBA")
+    background = embed_image(background, lower_mask, standard_fg_size=background.size)
+    higher_mask = Image.open("src/assets/Guest02.png").convert("RGBA")
+    background = embed_image(background, higher_mask, standard_fg_size=background.size)
+    seal = Image.open("src/assets/Guest01.png").convert("RGBA")
+    background = embed_image(background, seal, standard_fg_size=background.size)
+    # background = background.convert("RGB")
+
+    font_size = 210
+    font_path = font_path_zihun59
+    text = name
+    name, background, next_height = write_text(
+        text,
+        font_path,
+        background,
+        H_PADDING_RATION,
+        font_size,
+        alignment="right",
+        start_y=V_START,
+    )
+
+    next_height += 50
+    font_size = 100
+    font_path = font_path_notosans
+    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()
+    text = name_pinyin
+    name_pinyin, background, next_height = write_text(
+        text,
+        font_path,
+        background,
+        H_PADDING_RATION,
+        font_size,
+        alignment="right",
+        start_y=next_height + V_PADDING,
+    )
+
+    next_height += 100
+    font_size = 70
+    font_path = font_path_notosans
+    for title in title_list:
+        text = title
+        title, background, next_height = write_text(
+            text,
+            font_path,
+            background,
+            H_PADDING_RATION,
+            font_size,
+            alignment="right",
+            start_y=next_height + V_PADDING,
+        )
+    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"):
-        with open(f"src/assets/output/{item.conference_info['name'] + '.jpg'}", 'wb') as f:
-            f.write(img_stream.getvalue())
+        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:
-        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}
-
-@app.post("/getGuestComposePoster")
-def getGuestComposePoster(item:ConferenceInfo):
-    # item.conference_rundown = [list(i) for i in item.conference_rundown]
-    sorted_conference_guest = sorted(item.guest, key=lambda x: x['rank'])
-    img_stream = img_process.getGuestComposePoster(
-        item.name,
-        item.eng_name,
-        sorted_conference_guest)
-    file_dir = "ConferenceGuestPoster"
-    object_name = file_dir + "/" + item.name + ".jpg"
+        return None
+    img_stream = compose_guest_poster(name, name_pinyin, guest_photo, titles)
+    return img_stream
+
+
+def crop_round_img(image, circle_center, circle_radius):
+    mask = Image.new("L", image.size, 0)
+    draw = ImageDraw.Draw(mask)
+    draw.ellipse((circle_center[0] - circle_radius, circle_center[1] - circle_radius, circle_center[0] + circle_radius, circle_center[1] + circle_radius), fill=255)
+    output_image = Image.new("RGBA", image.size)
+    output_image.paste(image, (0, 0), mask)
+    output_image = output_image.crop(
+        (circle_center[0] - circle_radius, circle_center[1] - circle_radius, circle_center[0] + circle_radius, circle_center[1] + circle_radius)
+    )
+    # output_image = cut_image_blank(output_image)
+
     if os.path.exists("src/assets/output"):
-        with open(f"src/assets/output/{item.name + '.jpg'}", 'wb') as f:
-            f.write(img_stream.getvalue())
-    else:
-        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}
+        output_image.save("src/assets/output/cropped_circle_image.png")
+    return output_image
+
+
+def make_conference_poster(conference_name, conference_name_eng, conference_name_full, conference_info, conference_rundown):
+
+    BACKGROUND_COLOR = (1, 6, 143)
+
+    background = Image.open("src/assets/schedule/top_background.jpg").convert("RGBA")
+    bar = Image.open("src/assets/schedule/bar.png").convert("RGBA")
+    point_bar = Image.open("src/assets/schedule/point_bar.png").convert("RGBA")
+
+    H_PADDING_RATION = 0.06
+    V_PADDING = 80
+    V_START = 1100
+    bg_width, bg_height = background.size
+    print(background.size)
+    print(point_bar.size)
+    schedule_top, top_next_height = make_conference_schedule_top(
+        background, conference_name, conference_name_eng, conference_name_full, conference_info, bar, H_PADDING_RATION, V_START
+    )
+    schedule_components = [schedule_top]
+
+    ##########################################################
+    for rundown in conference_rundown:
+        rundown_pic = write_rundown(BACKGROUND_COLOR, rundown, point_bar, H_PADDING_RATION, bg_width)
+        # if rundown["type"] == "主持人":
+        #     schedule_components = schedule_components[:1] + [rundown_pic] + schedule_components[1:]
+        # else:
+        schedule_components.append(rundown_pic)
+    widths, heights = zip(*(i.size for i in schedule_components))
+    schedule_poster = Image.new("RGB", (bg_width, sum(heights)))
+    x_offset = 0
+    for img in schedule_components:
+        schedule_poster.paste(img, (0, x_offset))
+        x_offset += img.size[1]
+
+    img_stream = io.BytesIO()
+    schedule_poster.save(img_stream, format="JPEG")
+    img_stream.seek(0)
+    if os.path.exists("src/assets/output"):
+        schedule_poster.save("src/assets/output/schedule.jpg")
+    return img_stream
+
+
+def write_rundown(BACKGROUND_COLOR, rundown, point_bar, H_PADDING_RATION, bg_width):
+    background = Image.new("RGB", (bg_width, 6000), BACKGROUND_COLOR)
+    next_height = 0
+    photo_size_ratio = 0.12
+    V_PADDING = 80
+    background = embed_image(
+        background,
+        point_bar,
+        offset_y = V_PADDING,
+    )
+    time_width = 0
+    font_size = 130
+    font_path = font_path_notosansM
+    # print(point_bar.size)
+    if rundown["type"] != "主持人":
+        text = rundown["start_time"][:-3] + "-" + rundown["end_time"][:-3]
+        text_ = text_info(text, font_size, [0, 0],
+                          max_height=font_size,
+                          max_width=bg_width * (1 - H_PADDING_RATION * 2.5),
+                          font_path=font_path)
+        start_y = V_PADDING + (point_bar.height - text_.font_size) // 2
+        conference_rundown_time, background, next_height = write_text(
+            text, font_path, background, H_PADDING_RATION * 1.5, font_size,
+            start_y=start_y,
+            alignment="left"
+        )
+        time_width = conference_rundown_time.text_width
+    rundown["title"] = rundown["title"] if "title" in rundown else ""
 
+    text = rundown["type"]  # rundown["title"] if rundown["type"] != "主持人" else "主持人"
+    pioneer_part_width = H_PADDING_RATION * 1.5 * bg_width + time_width + font_size // 2
+    text_ = text_info(
+        text,
+        font_size,
+        [0, 0],
+        max_height=font_size,
+        max_width=bg_width - pioneer_part_width - H_PADDING_RATION * bg_width, font_path=font_path)
+    start_y = V_PADDING + (point_bar.height - text_.font_size) // 2
+    conference_rundown_type, background, next_height = write_text(
+        text,
+        font_path,
+        background,
+        pioneer_part_width / bg_width,
+        font_size,
+        start_y=start_y,
+        max_width=bg_width - pioneer_part_width - H_PADDING_RATION * bg_width,
+        alignment="left",
+    )
 
 
-interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
-app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)
+    next_height = V_PADDING + point_bar.size[1] + V_PADDING
+
+    if rundown["type"] != "主持人":
+        text = rundown["title"]  # rundown["title"] if rundown["type"] != "主持人" else "主持人"
+        pioneer_part_width = H_PADDING_RATION * 1.5 * bg_width + time_width + font_size // 2
+        text_ = text_info(
+            text,
+            font_size,
+            [0, 0],
+            max_height=font_size,
+            max_width=bg_width - pioneer_part_width - H_PADDING_RATION * bg_width, font_path=font_path)
+        # start_y = V_PADDING + (point_bar.height - text_.font_size) // 2
+        conference_rundown_type, background, next_height = write_text(
+            text,
+            font_path,
+            background,
+            pioneer_part_width / bg_width,
+            font_size,
+            start_y=next_height,
+            max_width=bg_width - 2 * H_PADDING_RATION * bg_width,
+            alignment="medium",
+        )
+
+
+    for person in rundown["people"]:
+        # if person["photo_link"] != "":
+        next_height += V_PADDING
+        background, next_height = write_person_line(background, person, H_PADDING_RATION, next_height, photo_size_ratio)
+
+    background = background.convert("RGB")
+    background = background.crop((0, 0, background.size[0], next_height + V_PADDING))
+    if os.path.exists("src/assets/output"):
+        background.save("src/assets/output/schedule_rundown.jpg")
+    return background
+
+
+def write_person_line(background, person, H_PADDING_RATION, start_y, photo_size_ratio):
+    bg_width = background.size[0]
+    text = person["name"]
+    font_size = 120
+    font_path = font_path_zihun59
+    alignment = "left"
+    pioneer_part_width = H_PADDING_RATION * bg_width + font_size
+
+    radius = int(photo_size_ratio * bg_width)
+    if person["photo_link"] != "":
+        round_photo = write_round_photo(background, person, H_PADDING_RATION * bg_width + radius, start_y + radius, radius)
+        pioneer_part_width += round_photo.size[1]
+
+    person["title"] = person["title"].split("\n")
+
+    name_padding = 50
+    title_padding = 20
+
+    people_text_y = start_y + (2 * radius - font_size - name_padding - 80 * len(person["title"])) // 2
+    conference_rundown_people_name, background, next_height = write_text(
+        text,
+        font_path,
+        background,
+        pioneer_part_width / bg_width,
+        font_size,
+        start_y=people_text_y,
+        max_width=bg_width - pioneer_part_width - H_PADDING_RATION * bg_width,
+        alignment=alignment
+    )
+    next_height += name_padding
+    font_size = 80
+    font_path = font_path_notosansM
+    # pioneer_part_width = H_PADDING_RATION * bg_width + round_photo.size[1] + font_size
+    for i, text in enumerate(person["title"]):
+        conference_rundown_people_title, background, next_height = write_text(
+            text,
+            font_path,
+            background,
+            pioneer_part_width / bg_width,
+            font_size,
+            start_y=next_height + title_padding,
+            max_width=bg_width - pioneer_part_width - H_PADDING_RATION * bg_width,
+            max_height=radius * 1.2,
+            alignment=alignment,
+        )
+
+    return background, start_y + radius * 2
+
+
+def write_round_photo(background, person, start_x, start_y, radius):
+    photo = Image.new("RGBA", (100, 100), (125, 125, 125, 0))
+    res = requests.get(person["photo_link"])
+    photo = Image.open(io.BytesIO(res.content)).convert("RGBA")
+    width, height = photo.size
+    circle_center = (width // 2, height // 4)
+    circle_radius = min(width, height // 2) // 2
+    round_photo = crop_round_img(photo, circle_center, circle_radius)
+    round_photo = round_photo.resize((radius*2, radius*2))
+    background = embed_image(background, round_photo, offset_y=start_y-radius, offset_x=start_x-radius)
+
+    return round_photo
+
+
+def make_conference_schedule_top(background, conference_name, conference_name_eng, conference_name_full, conference_info, bar, H_PADDING_RATION, V_START):
+    font_size = 250
+    font_path = font_path_pingfangM
+    text = conference_name
+    conference_name, background, next_height = write_text(
+        text,
+        font_path,
+        background,
+        H_PADDING_RATION,
+        font_size,
+        start_y=V_START,
+        alignment="left",
+    )
+
+    font_size = 110
+    font_path = font_path_notosans
+    text = conference_name_eng
+    V_PADDING = 120
+    conference_name_eng, background, next_height = write_text(
+        text, font_path, background, H_PADDING_RATION, font_size, start_y=next_height + V_PADDING, alignment="left"
+    )
+
+    font_size = 160
+    font_path = font_path_notosansM
+    text = "——" + conference_name_full + "——"
+    V_PADDING = 180
+    conference_name_full, background, next_height = write_text(
+        text, font_path, background, H_PADDING_RATION, font_size, start_y=next_height + V_PADDING, alignment="medium"
+    )
+    # write_text(" ", font_path, background, H_PADDING_RATION, 200, start_height=next_height + 400, alignment="medium")
+    INNER_PADDING = 60
+    next_height = next_height + V_PADDING - INNER_PADDING
+    background, next_height = write_conference_info(conference_info, background, bar, H_PADDING_RATION, INNER_PADDING, next_height)
+    background = background.convert("RGB")
+    background = background.crop((0, 0, background.size[0], next_height + V_PADDING))
+    if os.path.exists("src/assets/output"):
+        background.save("src/assets/output/schedule_top.jpg")
+
+    return background, next_height
+
+
+def write_conference_info(conference_info, background, bar, H_PADDING_RATION, V_PADDING, next_height):
+    bg_width, vg_height = background.size
+    for i, info in enumerate(conference_info):
+        font_size = 110
+        font_path = font_path_notosansB
+        text = info[0]
+        bar = bar.resize((int(bg_width * 0.06), int(font_size * 0.7)))
+        # V_PADDING = 180
+        info_name, background, next_height = write_text(
+            text, font_path, background, H_PADDING_RATION, font_size, start_y=next_height + V_PADDING, alignment="left"
+        )
+        background = embed_image(
+            background, bar, offset_y=next_height - (bar.size[1] + info_name.font_size) // 2, offset_x=bg_width * H_PADDING_RATION + info_name.text_width
+        )
+
+        font_size = 80
+        text = info[1]
+        pioneer_part_width = bar.size[0] + info_name.text_width + font_size
+        info_content, background, next_height = write_text(
+            text,
+            font_path,
+            background,
+            H_PADDING_RATION + pioneer_part_width / bg_width,
+            font_size,
+            max_width=bg_width * (1 - 2 * H_PADDING_RATION) - info_name.text_width,
+            start_y=next_height - (font_size + info_name.font_size) // 2,
+            alignment="left",
+        )
+
+    return background, next_height
+
+
+def getGuestComposePoster(conference_name, conference_eng_name, people):
+
+    top = Image.open("src/assets/GuestComposePoster/top.png").convert("RGBA")
+    card = Image.open("src/assets/GuestComposePoster/card.png").convert("RGBA")
+    V_PADDING = 80
+    background = Image.new("RGB", (top.size[0], card.size[1] * math.ceil(len(people) / 3) + V_PADDING), (255, 255, 255))
+    bg_width, bg_height = background.size
+    H_PADDING = (bg_width - 3*card.size[0]) / 2
+    V_START = 1200
+    print(top.size)
+    print(background.size)
+    
+    font_size = 220
+    font_path = font_path_notosansL
+    text = conference_name
+    conference_name, top, next_height = write_text(
+        text,
+        font_path,
+        top,
+        H_PADDING / bg_width,
+        font_size,
+        start_y=V_START,
+        alignment="medium",
+    )
+    font_size = 250
+    font_path = font_path_notosansL
+    text = conference_eng_name
+    conference_eng_name, top, next_height = write_text(
+        text,
+        font_path,
+        top,
+        H_PADDING / bg_width,
+        font_size,
+        start_y=next_height + V_PADDING,
+        max_height=700,
+        alignment="medium",
+    )
+    
+    for (i, person) in enumerate(people):
+        background = write_person_card(background, card, person, H_PADDING + card.size[0] * (i % 3), card.height * (i // 3))
+
+    guest_compose_poster = Image.new("RGB", (bg_width, background.size[1] + top.size[1]))
+    guest_compose_poster.paste(top, (0, 0))
+    guest_compose_poster.paste(background, (0, top.size[1]))
+    img_stream = io.BytesIO()
+    guest_compose_poster.save(img_stream, format="JPEG")
+    img_stream.seek(0)
+    if os.path.exists("src/assets/output"):
+        guest_compose_poster.save("src/assets/output/guest_compose_poster.jpg")
+    return img_stream
+
+
+def write_person_card(background, card, person, H_PADDING, start_y):
+    bg_width = background.size[0]
+    # print(bg_width * photo_size_ratio)
+    white_bar = 70
+    radius = int(0.67 * card.width) // 2
+    
+    round_photo = write_round_photo(background, person, H_PADDING + card.width // 2, start_y+white_bar+radius, radius)
+    background = embed_image(background, card, offset_y=start_y, offset_x=H_PADDING)
+
+    text = person["name"]
+    font_size = 120
+    font_path = font_path_zihun59
+    pioneer_part_width = H_PADDING
+    people_text_y = start_y + 2 * radius + white_bar + 20
+    guest_compose_poster_people_name, background, next_height = write_text(
+        text, font_path, background, 
+        pioneer_part_width / bg_width, 
+        font_size, 
+        start_y=people_text_y,
+        start_x=H_PADDING,
+        part_width=card.width,
+        max_width=card.width,
+        alignment="medium",
+        # color=(0,0,0)
+    )
+
+    text = person["title"]
+    text = "\n".join(text.split("\n")[:5])
+    font_size = 40
+    font_path = font_path_notosansM
+    next_height += 30
+    for i, text in enumerate(text.split("\n")):
+        pioneer_part_width = H_PADDING
+        people_text_y = next_height + 10
+        guest_compose_poster_people_title, background, next_height = write_text(
+            text, font_path, background, 
+            pioneer_part_width / bg_width, 
+            font_size, 
+            start_y=people_text_y,
+            start_x=H_PADDING,
+            part_width=card.width,
+            max_width=card.width-white_bar,
+            alignment="medium",
+            # color=(0,0,0)
+        )
+
+    return background
 
 
 if __name__ == "__main__":
-    MOCK_URL = "https://mock.jsont.run/iKAdks2DIVBtx-R7Fbw3y"
-    res = requests.get(MOCK_URL)
-    mock_data = res.json()
-    # mock_data = ConferencePosterInfo(**mock_data)
-    # getConferencePoster(mock_data)
-    print(mock_data)
-    mock_data = ConferenceInfo(**mock_data["conference_info"])
-    getGuestComposePoster(mock_data)
+    # image_path = 'src/assets/邀请函封面.jpg'  # 替换为你的图片文件路径
+    # text_to_write = "你好，世界！"
+    # font_path = "src/assets/FangZhengHeiTiJianTi-1.ttf"
+    # position = (120, 140)
+    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=1721322000&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=qZCdG1n42uwCFGJ8%2BAlF6VK8KEM%3D&response-cache-control=public%2Cmax-age%3D86400"
+
+    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=1721743200&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=CvxnLBCIf%2B0aXEfaJ9XK6Vq%2FXU8%3D&response-cache-control=public%2Cmax-age%3D86400"
+
+    conference_name = "教育行业规范管理发展论坛"
+    conference_name_eng = "2024 World Education Conference"
+    conference_name_full = "2024世界教育者大会上海主会"
+    conference_info = [
+        ["指导单位", "上海市民办教育协会、长三角教育发展研究院"],
+        ["主办单位", "上海市民办教育协会、长三角教育发展研究院"],
+        ["论坛时间", "2024年8月21日上午9:00-12:30"],
+        ["论坛地址", "国家会展中心4.2号馆A"],
+    ]
+
+    conference_rundown = [
+        {
+            "type": "主持人",
+            "title": "上海市教科规划办副主任",
+            "people": [
+                {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                },
+                {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": "",
+                },
+            ],
+            "start_time": "9:00:00",
+            "end_time": "9:20:00",
+        },
+        {
+            "type": "致辞",
+            "title": "上海市教科规划办副主",
+            "people": [
+                {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                }
+            ],
+            "start_time": "9:00:00",
+            "end_time": "9:20:00",
+        },        {
+            "type": "致辞",
+            "title": "上海市教科规划办副主任上海市教科规划办副主任上海市教科规",
+            "people": [
+                {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                }
+            ],
+            "start_time": "9:00:00",
+            "end_time": "9:20:00",
+        },
+    ]
+    people = [
+                        {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                },
+                                        {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                },
+                                                        {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                },
+                                                                        {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                },
+                                                                                        {
+                    "name": "方法名",
+                    "title": "上海市教科规划办副主任\n上海教科院改革发展研究部研究员\n华东师范大学国家宏观教育研究院博士生导师",
+                    "photo_link": ORIGINAL_PHOTO_LINK,
+                },
+        {
+            "name": "方法名",
+            "title": "上海市教科规划办副主任",
+            "photo_link": ORIGINAL_PHOTO_LINK,
+        },
+        {
+            "name": "方法名",
+            "title": "上海市教科规划办副主任",
+            "photo_link": ORIGINAL_PHOTO_LINK,
+        },
+        {
+            "name": "方法名",
+            "title": "上海市教科规划办副主任",
+            "photo_link": ORIGINAL_PHOTO_LINK,
+        },
+        {
+            "name": "方法名",
+            "title": "上海市教科规划办副主任",
+            "photo_link": ORIGINAL_PHOTO_LINK
+        },
+        {
+            "name": "方法名",
+            "title": "上海市教科规划办副主任",
+            "photo_link": ORIGINAL_PHOTO_LINK,
+        },
+    ]
+
+    # 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)
+
+    # 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")))
+
+    # image = Image.open('src/assets/苏建诚.png')
+    # width, height = image.size
+    # circle_center = (width // 2, height // 4)
+    # circle_radius = min(width, height//2) // 2
+    # round_img = crop_round_img(image, circle_center, circle_radius)
+    # print(round_img.size)
+
+    # res = requests.get(ORIGINAL_PHOTO_LINK)
+    # photo = Image.open(io.BytesIO(res.content)).convert("RGBA")
+
+    # make_exhibitor_poster(LOGO_LINK, slogan_list[SLOGAN_ID], text_list[TEXT_ID])
+    # make_guest_poster(name, "", "", ORIGINAL_PHOTO_LINK, titles)
+
+    # make_conference_poster(conference_name, conference_name_eng, conference_name_full, conference_info, conference_rundown)
+    # 定义圆心和半径
+
+    # res = requests.get(ORIGINAL_PHOTO_LINK)
+    # print(res)
+    # with open("src/assets/output/guest_compose_poster.png", "wb") as f:
+    #     f.write(res.content)
+    # photo = Image.open(io.BytesIO(res.content)).convert("RGBA")
+
+        
+    getGuestComposePoster("全球教育者公论教育", "EDUCATORS AROUND THE GLOBE TALKING ABOUT EDUCATION", people)