Update src/pyscripts/img_process.py

src/pyscripts/img_process.py

@@ -531,13 +531,13 @@ def write_rundown(BACKGROUND_COLOR, rundown, point_bar, H_PADDING_RATION, bg_wid
     return background
 
 
-def write_person_line(background, person, H_PADDING_RATION, start_y, photo_size_ratio):
+def write_person_line(background, person, H_PADDING_RATION, start_y, photo_size_ratio, scale=1, text_start_padding=120, name_font_size=120, title_font_size=80):
     bg_width = background.size[0]
     text = person["name"]
-    font_size = 120
+    font_size = name_font_size * scale
     font_path = font_path_zihun59
     alignment = "left"
-    pioneer_part_width = H_PADDING_RATION * bg_width + font_size
+    pioneer_part_width = H_PADDING_RATION * bg_width + text_start_padding * scale
 
     radius = int(photo_size_ratio * bg_width)
     if person["photo_link"] != "":
@@ -546,10 +546,10 @@ def write_person_line(background, person, H_PADDING_RATION, start_y, photo_size_
 
     person["title"] = person["title"].split("\n")
 
-    name_padding = 50
-    title_padding = 20
+    name_padding = 50 * scale
+    title_padding = 20 * scale
 
-    people_text_y = start_y + (2 * radius - font_size - name_padding - 80 * len(person["title"])) // 2
+    people_text_y = start_y + (2 * radius - font_size - name_padding - title_font_size * scale * len(person["title"])) // 2
     conference_rundown_people_name, background, next_height = write_text(
         text,
         font_path,
@@ -561,7 +561,7 @@ def write_person_line(background, person, H_PADDING_RATION, start_y, photo_size_
         alignment=alignment
     )
     next_height += name_padding
-    font_size = 80
+    font_size = title_font_size * scale
     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"]):
@@ -585,8 +585,9 @@ def write_round_photo(background, person, start_x, start_y, radius):
     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
+    circle_center = (width // 2, height // 3)
+    circle_radius = min(width//2, height // 3)
+    print(width, height, circle_center, circle_radius)
     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)
@@ -713,7 +714,7 @@ def getGuestComposePoster(conference_name, conference_eng_name, people):
         max_height=700,
         alignment="medium",
     )
-    
+
     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))
@@ -777,6 +778,37 @@ def write_person_card(background, card, person, H_PADDING, start_y):
     return background
 
 
+def getGuestEScreen(root_type, person):
+    template_info = {
+        "full": {
+            "width": 3328,
+            "height": 1152
+        },
+        "half": {
+            "width": 3072,
+            "height": 1280
+        }
+    }
+    background = Image.open(f"src/assets/EScreen/{root_type}.jpg")
+    bg_width, bg_height = background.size
+    scale = bg_width / template_info[root_type]["width"]
+
+    radius = 600 // 2 * scale
+    radius_ratio = radius / bg_width
+    H_PADDING_RATION = 0.22
+    V_START = bg_height // 2 - radius
+    print(background.size)
+
+    write_person_line(background, person, H_PADDING_RATION, V_START, radius_ratio,
+                      text_start_padding=200, name_font_size=100, title_font_size=50, scale=scale)
+
+    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/hall_screen.jpg")
+    return img_stream
+
 if __name__ == "__main__":
     # image_path = 'src/assets/邀请函封面.jpg'  # 替换为你的图片文件路径
     # text_to_write = "你好，世界！"
@@ -827,7 +859,7 @@ if __name__ == "__main__":
 
     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/tkqv47pvn5dfHUX1w8qMwap_100.jpeg?Expires=1722618000&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=jV7%2BtqMNMaLmuwbFIcNih8x5atU%3D&response-cache-control=public%2Cmax-age%3D86400"
+    ORIGINAL_PHOTO_LINK = "https://weboffice-temporary.ks3-cn-beijing.wpscdn.cn/thumbnail/tn6C9byoghgm8cnQYtShMdb_100.jpeg?Expires=1723395600&KSSAccessKeyId=AKLTmoJhggaFT1CHuozGZqbC&Signature=%2BJX%2BaPumafx8mXpV7wfN1Z3FVpA%3D&response-cache-control=public%2Cmax-age%3D86400"
 
     conference_name = "教育行业规范管理发展论坛"
     conference_name_eng = "2024 World Education Conference"
@@ -960,7 +992,7 @@ if __name__ == "__main__":
     # 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_guest_poster(name, "", "", ORIGINAL_PHOTO_LINK, titles)
 
     # make_conference_poster(conference_name, conference_name_eng, conference_name_full, conference_info, conference_rundown)
     # 定义圆心和半径
@@ -973,3 +1005,6 @@ if __name__ == "__main__":
 
         
     # getGuestComposePoster("全球教育者公论教育", "EDUCATORS AROUND THE GLOBE TALKING ABOUT EDUCATION", people)
+    
+    getGuestEScreen("full", conference_rundown[0]["people"][0])
+