g8ddlZddlmZddlZddlZddZdZedkrejdZ e d e d e d d e dde ddde ZeejejZeraedeedejdejrdeDZeD] Zeen eeejddSejddSdS)N) ColorThiefctj|s!td|dtjdStj|s!td|dtjdS t|}||}|S#t$r,}td|d |tjYd}~dSd}~wwxYw) al Extracts a specified number of colors from an image using ColorThief. Args: image_path (str): The path to the image file. num_colors (int): The number of colors to extract in the palette. Returns: list: A list of RGB tuples representing the color palette, or None if the file is invalid or an error occurs. z Error: Image file not found at '')fileNz Error: Path 'z' is not a file.) color_countzError processing image 'z': ) ospathexistsprintsysstderrisfiler get_palette Exception) image_path num_colors color_thiefpalettees 3/Users/srajan/code/music-website/extract_palette.pyextract_colorsrs 7>>* % % >>>>SZPPPPt 7>>* % % ;z;;;#* M M M M ,, ))j)AA  ;;;;;#*MMMMttttts&B)) C3!CCcTd|d|d|dS)z,Converts an RGB tuple to a hex color string.z#{:02x}{:02x}{:02x}r)format)rgbs r rgb_to_hexr"s& ' 'AAA ? ??__main__z0Extract a color palette from an album art image.) description image_filez!Path to the album art image file.)typehelpz-nz --num_colorsz(Number of colors to extract (default: 6))r#defaultr$z--hex store_truez2Output colors in hex format instead of RGB tuples.)actionr$z Extracted z colors from 'z':c,g|]}t|S)r).0colors r r-5s BBB:e,,BBBrr)r)argparse colorthiefrr r rr__name__ArgumentParserparser add_argumentstrint parse_argsargsr"rrr lenhex hex_paletter,exitr*rrr<s!!!!!! 8@@@ z $X $1c d d dF  35XYYY n3GIII  QSSS     DnT_do>>G  J33w<<JJtJJJKKK 8 BB'BBBK$  e   E'NNN  -r