feat: update more size

app.py

@@ -6,14 +6,12 @@ from src.layoutCreate import generate_layout_photo, generate_layout_image
 import pathlib
 import numpy as np
 from image_utils import resize_image_to_kb
+from data_utils import csv_to_size_list
+
+# 获取尺寸列表
+size_list_dict = csv_to_size_list("size_list_CN.csv")
+print(size_list_dict)
 
-size_list_dict = {
-    "一寸": (413, 295),
-    "二寸": (626, 413),
-    "教师资格证": (413, 295),
-    "国家公务员考试": (413, 295),
-    "初级会计考试": (413, 295),
-}
 color_list_dict = {
     "蓝色": (86, 140, 212),
     "白色": (255, 255, 255),
@@ -233,7 +231,7 @@ if __name__ == "__main__":
     sess = onnxruntime.InferenceSession(HY_HUMAN_MATTING_WEIGHTS_PATH)
 
     size_mode = ["尺寸列表", "只换底", "自定义尺寸"]
-    size_list = ["一寸", "二寸", "教师资格证", "国家公务员考试", "初级会计考试"]
+    size_list = list(size_list_dict.keys())
     colors = ["蓝色", "白色", "红色", "自定义底色"]
     render = ["纯色", "上下渐变(白)", "中心渐变(白)"]
     image_kb = ["不设置", "自定义"]

data_utils.py

@@ -0,0 +1,18 @@
+import csv
+
+
+def csv_to_size_list(csv_file: str) -> dict:
+    # 初始化一个空字典
+    size_list_dict = {}
+
+    # 打开CSV文件并读取数据
+    with open(csv_file, mode="r") as file:
+        reader = csv.reader(file)
+        # 跳过表头
+        next(reader)
+        # 读取数据并填充字典
+        for row in reader:
+            size_name, h, w = row
+            size_list_dict[size_name] = (int(h), int(w))
+
+    return size_list_dict

size_list_CN.csv

@@ -0,0 +1,16 @@
+Name,Height,Width
+一寸,413,295
+二寸,626,413
+小一寸,378,260
+小二寸,531,413
+大一寸,567,390
+大二寸,626,413
+五寸,1499,1050
+教师资格证,413,295
+国家公务员考试,413,295
+初级会计考试,413,295
+英语四六级考试,192,144
+计算机等级考试,567,390
+研究生考试,709,531
+社保卡,441,358
+电子驾驶证,378,260