o g@sddlZddlmZddlZddlZdddZddZedkrejd d Z e j d e d d e j dde ddde j dddde ZeejejZeryedeedejdejrnddeDZeD]ZeeqfneeeddSeddSdS)N) ColorThiefc Cstj|std|dtjddStj|s&td|dtjddSz t|}|j|d}|WSt yS}ztd|d |tjdWYd}~dSd}~ww) 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.)Z color_countzError processing image 'z': ) ospathexistsprintsysstderrisfiler get_palette Exception) image_path num_colorsZ color_thiefpaletteer3/Users/srajan/code/music-website/extract_palette.pyextract_colorss   rcCsd|d|d|dS)z,Converts an RGB tuple to a hex color string.z#{:02x}{:02x}{:02x}r)format)Zrgbrrr rgb_to_hex"sr__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))rdefaultrz--hex store_truez2Output colors in hex format instead of RGB tuples.)actionrz Extracted z colors from 'z':cCsg|]}t|qSr)r).0colorrrr 5sr%r)r)argparseZ colorthiefrr rrr__name__ArgumentParserparser add_argumentstrint parse_argsargsrrrr lenhex hex_paletter$exitrrrrs8