Kkit.color

  1import requests
  2from haishoku.haishoku import Haishoku
  3import numpy as np
  4import colorsys
  5
  6def hex_to_rgb(value):
  7    """
  8    Convert a hex color string to an RGB tuple.
  9
 10    Parameters
 11    ----------
 12    value : str
 13        hex color string
 14
 15    Returns
 16    -------
 17    tuple
 18        RGB color tuple
 19    """
 20    value = value.lstrip('#')
 21    lv = len(value)
 22    return tuple(int(value[i:i + lv // 3], 16) for i in range(0, lv, lv // 3))
 23
 24def rgb_to_hex(rgb):
 25    """
 26    Convert an RGB tuple to a hex color string.
 27
 28    Parameters
 29    ----------
 30    rgb : tuple
 31        RGB color tuple
 32
 33    Returns
 34    -------
 35    str
 36        hex color string
 37    """
 38    return '#%02x%02x%02x' % rgb
 39
 40def __normalize_rgb_color(color):
 41    return tuple([i/255 for i in color])
 42
 43def __denormalize_rgb_color(color):
 44    return tuple([i*255 for i in color])
 45
 46def rgb2hsv(color):
 47    """
 48    Convert an RGB color to an HSV color.
 49
 50    Parameters
 51    ----------
 52    color : tuple
 53        RGB color tuple
 54
 55    Returns
 56    -------
 57    tuple
 58        HSV color tuple
 59    """
 60    color = __normalize_rgb_color(color)
 61    color = colorsys.rgb_to_hsv(*color)
 62    return color
 63
 64def __int_rgb(color):
 65    return tuple([int(i) for i in color])
 66
 67def hsv2rgb(color):
 68    """
 69    Convert an HSV color to an RGB color.
 70
 71    Parameters
 72    ----------
 73    color : tuple
 74        HSV color tuple
 75
 76    Returns
 77    -------
 78    tuple
 79        RGB color tuple
 80    """
 81    color = colorsys.hsv_to_rgb(*color)
 82    color = __denormalize_rgb_color(color)
 83    color = __int_rgb(color)
 84    return color
 85
 86def str2color(string, Azure_key, num=5, verbose=False, vi=True):
 87    """
 88    Convert a string to a color.
 89
 90    This function uses the Bing Image Search API to search for images related to the input string, and then extracts the dominant color from the images to generate a color.
 91
 92    Parameters
 93    ----------
 94    string : str
 95        input string
 96
 97    Azure_key : str
 98        Microsoft Azure API key
 99
100    num : int
101        number of images to search
102        
103    verbose : bool
104        print the dominant color of each image
105
106    Returns
107    -------
108    tuple
109        RGB color tuple
110    list
111        list of RGB color tuples for each image
112    """
113    if num>150:
114        raise Exception("num must small than 150")
115    search_url = "https://api.bing.microsoft.com/v7.0/images/search"
116    headers = {"Ocp-Apim-Subscription-Key" : Azure_key}
117    # headers2 = {"user-agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.0.0 Safari/537.36"}
118    params  = {"q": string}
119    response = requests.get(search_url, headers=headers, params=params)
120    response.raise_for_status()
121    search_results = response.json()
122    thumbnail_urls = [img["thumbnailUrl"] for img in search_results["value"]]
123    color_list = []
124    for i in range(num):
125        # time.sleep(0.5)
126        try:
127            # img_data = requests.get(thumbnail_urls[i], headers=headers2)
128            # do_color = np.array(Haishoku.getDominant(io.BytesIO(img_data.content)))
129            do_color = np.array(Haishoku.getDominant(thumbnail_urls[i]))
130            color_list.append(do_color)
131            if verbose:
132                print(i, thumbnail_urls[i], do_color)
133        except:
134            print("error: %s"%thumbnail_urls[i])
135    color_list = np.vstack(color_list)
136    rgb_color = np.mean(color_list, axis=0)
137    rgb_color = rgb_color.astype("int")
138    rgb_color = tuple(rgb_color)
139    hsl_color = rgb2hsv(rgb_color)
140    hsl_color = (hsl_color[0], 1, 0.7)
141    modified_rgb = hsv2rgb(hsl_color)
142    
143    if vi:
144         pass       
145
146    return modified_rgb, color_list