map colors

app.py

@@ -16,7 +16,7 @@ if not EMULATED:
 else:
     from models_stub import make_image_controlnet, make_inpainting, segment_image
 
-from config import HEIGHT, WIDTH, POS_PROMPT, NEG_PROMPT, COLOR_MAPPING, map_colors
+from config import HEIGHT, WIDTH, POS_PROMPT, NEG_PROMPT, COLOR_MAPPING, map_colors, map_colors_rgb
 from palette import COLOR_MAPPING_CATEGORY
 from preprocessing import preprocess_seg_mask, get_image, get_mask
 
@@ -166,7 +166,7 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
                 st.session_state['unique_colors'] = unique_colors
                 # multiselect to choose multiple colors
                 chosen_colors = st.multiselect(
-                    label="Choose colors", options=unique_colors, default=None, key='unique_colors'
+                    label="Choose colors", options=unique_colors, default=None, key='unique_colors', format_func=map_colors_rgb
                 )
                 print(chosen_colors)
 

config.py

@@ -27,6 +27,9 @@ def map_colors(color: str) -> str:
     """
     return COLOR_MAPPING[color]
 
+def map_colors_rgb(color: tuple) -> str:
+    return COLOR_MAPPING_RGB[color]
+
 
 POS_PROMPT = "tree, sky, cloud, scenery, outdoors, grass, flowers, sunlight, beautiful, ultra detailed beautiful landscape, architectural renderings vegetation, high res, best high quality landscape, outdoor lighting, sunshine, 4k, 8k, realistic"
 NEG_PROMPT= "lowres, deformed, blurry, bad anatomy, disfigured, poorly drawn face, mutation, mutated, extra limb, ugly, poorly drawn hands, missing limb, blurry, floating limbs, disconnected limbs, malformed hands, blur, out of focus, long neck, long body, mutated hands and fingers, out of frame"