^jz'dZddlmZddlZddlmZmZddlZddlZ ddl m Z m Z m Z mZmZddlmZerddlmZddZdd Z dd Zdd Zdd Zdd ZeeddZej6f ddZy)aFunctional implementations for text manipulation and rendering. This module provides utility functions for manipulating text in strings and rendering text onto images. Includes functions for word manipulation, text drawing, and handling text regions in images. ) annotationsN) TYPE_CHECKINGAny)MONO_CHANNEL_DIMENSIONSNUM_MULTI_CHANNEL_DIMENSIONSNUM_RGB_CHANNELSpreserve_channel_dimuint8_io)PAIRImagec|t|k\ry|jtt||}t|Dcgc] \}}||vs |}}}dj |Scc}}w)a&Delete a specified number of random words from a list. This function randomly removes words from the input list and joins the remaining words with spaces to form a new string. Args: words (list[str]): List of words to process. num_words (int): Number of words to delete. py_random (random.Random): Random number generator for reproducibility. Returns: str: New string with specified words removed. Returns empty string if num_words is greater than or equal to the length of words.  )lensamplerange enumeratejoin)words num_words py_randomindices_to_deleteidxword new_wordss w/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/albumentations/augmentations/text/functional.pydelete_random_wordsrsh CJ!((s5z):IF'0'7X)#t3FW;WXIX 88I Ys  A'A'c$|dk(st|tkrdj|S|j}t |D]:}|j t t|d\}}||||c||<||<<dj|S)aSwap random pairs of words in a list of words. This function randomly selects pairs of words and swaps their positions a specified number of times. Args: words (list[str]): List of words to process. num_words (int): Number of swaps to perform. py_random (random.Random): Random number generator for reproducibility. Returns: str: New string with words swapped. If num_words is 0 or the list has fewer than 2 words, returns the original string. rr)rr rcopyrr)rrr_idx1idx2s rswap_random_wordsr%6s A~Ud*xx JJLE 9 <%%eCJ&7; d#(;d  d U4[< 88E?c|d}t|D]>}|jdt|}|j||j |@dj |S)a;Insert random stopwords into a list of words. This function randomly inserts stopwords at random positions in the list of words a specified number of times. Args: words (list[str]): List of words to process. num_insertions (int): Number of stopwords to insert. stopwords (tuple[str, ...] | None): Tuple of stopwords to choose from. If None, default stopwords will be used. py_random (random.Random): Random number generator for reproducibility. Returns: str: New string with stopwords inserted. )andtheisinatofrr)rrandintrinsertchoicer)rnum_insertions stopwordsrr"rs rinsert_random_stopwordsr3Qsb,: > "73u:. S)**9567 88E?r&c ddlm}|jtk(r|j |S|jt k(r0|jddk(r|j |dddddfdS|jt k(r(|jdtk(r|j |Std |j#t$rtdtwxYw) z+Convert a NumPy array image to a PIL image.rr Pillow is not installedr NL)modezUnsupported image shape: ) PILr ImportErrorndimr fromarrayrshaper TypeError)imager s rconvert_image_to_pilr@psF zz,,uu%% zz11ekk!n6IuuQ1W~C88 zz11ekk!nHX6Xuu%% / }= >> F34+EFs CCc ddlm}|j|}|D]}|d}|d}|d}|d}|jdk(rVt |t rht|d k\rtt|d d d z }n?t|d k(r1t|d}n"t |t rt d |D}|d d } |j| ||||S#t$rtdtwxYw)zDraw text on a PIL image.r) ImageDrawr5 bbox_coordstextfont font_colorr7Nr6c32K|]}t|yw)N)int).0cs r z)draw_text_on_pil_image..s:!s1v:sr rEfill) r9rBr:Drawr8 isinstancetuplerrIsumrD) pil_image metadata_listrBdrawmetadatarCrDrErFpositions rdraw_text_on_pil_imagerXs F! >>) $D!>}- l+  >>S *e,z?a'!$SBQ%81%* 3 F34+EFs CC1c ` ddlm}m}t |j dDcgc]}|j |dddd|f }}|Dcgc]}|j|}}|D]}|d} |d} |d} |d } t| |j dkr5tt| dg|j dt| z zz} n-t| |j dkDr| d|j d} | D cgc] } t| } } | dd}t|D]\}}|j|| | | | tj|Dcgc]}tj |c}d S#t$rtdtwxYwcc}wcc}wcc} wcc}w) zADraw text on a multi-channel image with more than three channels.r)r rBr5r NrCrDrErFrM)axis)r9r rBr:rr=r<rOrrQlistrIrrDnpstackarray)r?rTr rBichannelschannel pil_imagesrVrCrDrErFrKrW channel_idrSs r draw_text_on_multi_channel_imagerdsF(:?u{{1~9NOAaAg/OHO9ABg)..)BJB!S}- l+  z?U[[^ +tJ/1#Q#j/9Y2ZZ[J _u{{1~ -#$4ekk!n5J'11c!f1 1r?&/z%: S !J NN8T:j;QN R S+S2 88h?7RXXg&?a HH? F34+EFPB$2@s"E?#F F!F&F+?Fc,|r t||}t|jtk(s3t|jtk(rC|jddt hvr,t |}t||}tj|St||S)a5Render text onto an image based on provided metadata. This function draws text on an image using metadata that specifies text content, position, font, and color. It can optionally clear the background before rendering. The function handles different image types (grayscale, RGB, multi-channel). Args: image (np.ndarray): Image to draw text on. metadata_list (list[dict[str, Any]]): List of metadata dictionaries containing: - bbox_coords: Bounding box coordinates (x_min, y_min, x_max, y_max) - text: Text string to render - font: PIL ImageFont object - font_color: Color for the text clear_bg (bool): Whether to clear (inpaint) the background under the text. Returns: np.ndarray: Image with text rendered on it. r r6) inpaint_text_backgroundrr=rrrr@rXr\r^rd)r?rTclear_bgrSs r render_textrhs.'}= 5;;22 EKK88U[[^PQScOd=d(/ *9mD xx "" +E= AAr&c$|j}tj|jd|jdftj}|D]"}|d\}}}} d||| ||f<d||| ||f<$t j ||d|S)a+Inpaint (clear) regions in an image where text will be rendered. This function creates a clean background for text by inpainting rectangular regions specified in the metadata. It removes any existing content in those regions to provide a clean slate for rendering text. Args: image (np.ndarray): Image to inpaint. metadata_list (list[dict[str, Any]]): List of metadata dictionaries containing: - bbox_coords: Bounding box coordinates (x_min, y_min, x_max, y_max) method (int, optional): Inpainting method to use. Defaults to cv2.INPAINT_TELEA. Options include: - cv2.INPAINT_TELEA: Fast Marching Method - cv2.INPAINT_NS: Navier-Stokes method Returns: np.ndarray: Image with specified regions inpainted. rr6)dtyperCrG) inpaintRadiusflags)r!r\zerosr=uint8cv2inpaint) r?rTmethod result_imagemaskrVx_miny_minx_maxy_maxs rrfrfs0::rs# % 606 &   >?"@#IL B BJ##%J %J'%J %J %Jr&