enesbol commited on
Commit
40e91a1
·
1 Parent(s): 3ff7c43

Update handler.py

Browse files
Files changed (1) hide show
  1. 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
- return names[index]
 
 
 
 
 
 
 
 
 
 
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