Update handler.py
Browse files- handler.py +13 -3
handler.py
CHANGED
|
@@ -59,14 +59,24 @@ class EndpointHandler():
|
|
| 59 |
rgb_tuple = tuple(int(hex_color[i:i+2], 16) for i in (1, 3, 5))
|
| 60 |
names = []
|
| 61 |
rgb_values = []
|
| 62 |
-
|
| 63 |
for color_hex, color_name in CSS3_HEX_TO_NAMES.items():
|
| 64 |
names.append(color_name)
|
| 65 |
rgb_values.append(hex_to_rgb(color_hex))
|
| 66 |
-
|
| 67 |
kdt_db = KDTree(rgb_values)
|
| 68 |
distance, index = kdt_db.query(rgb_tuple)
|
| 69 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
|
|
|
|
| 59 |
rgb_tuple = tuple(int(hex_color[i:i+2], 16) for i in (1, 3, 5))
|
| 60 |
names = []
|
| 61 |
rgb_values = []
|
| 62 |
+
|
| 63 |
for color_hex, color_name in CSS3_HEX_TO_NAMES.items():
|
| 64 |
names.append(color_name)
|
| 65 |
rgb_values.append(hex_to_rgb(color_hex))
|
| 66 |
+
|
| 67 |
kdt_db = KDTree(rgb_values)
|
| 68 |
distance, index = kdt_db.query(rgb_tuple)
|
| 69 |
+
|
| 70 |
+
color_mapping = {
|
| 71 |
+
'tomato': 'red',
|
| 72 |
+
'chocolate': 'brown-black',
|
| 73 |
+
'darkgoldenrod': 'yellow',
|
| 74 |
+
}
|
| 75 |
+
|
| 76 |
+
color_name = names[index]
|
| 77 |
+
mapped_color = color_mapping.get(color_name, color_name)
|
| 78 |
+
|
| 79 |
+
return mapped_color
|
| 80 |
|
| 81 |
|
| 82 |
|