feat: more color

Zeyi-Lin · Sep 8, 2024 · 7e066a6 · 7e066a6
1 parent 1d0a38b
commit 7e066a6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -22,12 +22,16 @@
     "蓝色": (86, 140, 212),
     "白色": (255, 255, 255),
     "红色": (233, 51, 35),
+    "黑色": (0, 0, 0),
+    "深蓝色": (69, 98, 148),
 }
 
 color_list_dict_EN = {
     "Blue": (86, 140, 212),
     "White": (255, 255, 255),
     "Red": (233, 51, 35),
+    "Black": (0, 0, 0),
+    "Dark blue": (69, 98, 148),
 }
 
 
@@ -321,8 +325,8 @@ def idphoto_inference(
     size_list_CN = list(size_list_dict_CN.keys())
     size_list_EN = list(size_list_dict_EN.keys())
 
-    colors_CN = ["蓝色", "白色", "红色", "自定义底色"]
-    colors_EN = ["Blue", "White", "Red", "Custom Color"]
+    colors_CN = ["蓝色", "白色", "红色", "黑色", "深蓝色", "自定义底色"]
+    colors_EN = ["Blue", "White", "Red", "Black", "Dark blue", "Custom Color"]
 
     render_CN = ["纯色", "上下渐变 (白)", "中心渐变 (白)"]
     render_EN = ["Solid Color", "Up-Down Gradient (White)", "Center Gradient (White)"]