Update src/pyscripts/img_process.py

src/pyscripts/img_process.py

@@ -1,975 +1,179 @@
-from PIL import Image, ImageDraw, ImageFont
-import re
+from fastapi import FastAPI, Path, Query, Response, BackgroundTasks
+from fastapi.responses import FileResponse, HTMLResponse
+import gradio as gr
+import sys
 import os
-import io
-from dataclasses import dataclass
+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 requests
-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"):
-        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 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"):
-        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",
-    )
-
-
-    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)))
-
-        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
-        text_ = text_info(
-            text,
-            font_size,
-            [0, 0],
-            max_height=font_size,
-            max_width=bg_width * (1 - 2 * H_PADDING_RATION) - info_name.text_width - bar.width,
-            font_path=font_path, alignment="left")
-        info_content, background, _ = 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 - bar.width,
-            start_y=next_height - info_name.font_size + (info_name.font_size - text_.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)
+# 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")
     
-    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",
-    )
+    start_period = "上午" if start_datetime.hour < 12 else "下午"
+    end_period = "上午" if end_datetime.hour < 12 else "下午"
     
-    people = [p for p in people if p["photo_link"] != ""]
-    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
+    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")
     
-    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)
+    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"
+    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())
+    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"
+    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}
 
-    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 = "\n".join(person["title"].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
+interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
+app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)
 
 
 if __name__ == "__main__":
-    # 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)
+    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)