From 7e066a6d7205476a3b8a53f4db856d6d16f39b97 Mon Sep 17 00:00:00 2001
From: ZeYi Lin <944270057@qq.com>
Date: Sun, 8 Sep 2024 15:52:16 +0800
Subject: [PATCH] feat: more color

---
 app.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 4be42580..da470bd0 100644
--- 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)"]