j jdZddlmZddlZddlmZddlmZddlZddl Z ddl m Z edGd d Z dd Zdd Zdd ZddZy)uImage preprocessing for EKG strips. For PTB-XL rendered images this is essentially a no-op — the renderer already produces clean grids on a parchment background. For user-uploaded phone photos, the pipeline below trims margins, corrects light skew, and normalizes contrast so the multimodal model sees the cleanest possible representation. Each step is conservative: skip if the input looks already-good. The cost of doing nothing on a clean image is small; the cost of mangling a clean image with aggressive deskew is large. ) annotationsN) dataclass)Path)ImageT)frozenc>eZdZUded<ded<ded<ded<d dZy ) PreparedImagez Image.Imagepilintwidthheightz list[str]notesctj}|jj|dd|j S)NPNGT)formatoptimize)ioBytesIOr savegetvalue)selfbufs L/var/www/adampowell.pro/apps/ekg-tutor/interpret/src/pipelines/image_prep.py to_png_byteszPreparedImage.to_png_bytes s/jjl  c%$ 7||~N)returnbytes)__name__ __module__ __qualname____annotations__rrrr r s  J K rr c^t|}|js t|tj|j d}g}t j|}tj|tj}t|\}}|r|jdt|\}}|r|jdt|}tj|tj}tj |}t#||j$|j&|S)zLoad an image from disk and return a preprocessed copy. The original is never modified. Notes accumulate for any operation that was applied so the upstream service can attach them to image_quality_notes on the canonical interpretation. RGBzauto-cropped uniform borderszrotated to correct skew)r r r r)rexistsFileNotFoundErrorropenconvertnparraycv2cvtColor COLOR_RGB2BGR_autocrop_bordersappend _maybe_deskew_normalize_contrast COLOR_BGR2RGB fromarrayr r r ) pathpr rnp_imgbgrappliedrgbouts rpreparer;&s T A 88:"" **Q-   &CE XXc]F ,,vs00 1C$S)LC 34 %LC ./ c "C ,,sC-- .C //# C S #**E RRrctj|tj}|dkD|dkzjtj dz}tj |}tj|}||dfStj|\}}}}|jdd\} } |dkr|dkr||z| dz kDr||z| dz kDr|dfS|d| zks|d| zkr|dfS||||z|||zfd fS) zfTrim solid-color borders. Conservative: bails out if it would crop more than 12% in any direction.NFg)\(?T) r+r,COLOR_BGR2GRAYastyper)uint8 bitwise_not findNonZero boundingRectshape) r7graymaskinvcoordsxywhHWs rr.r.Fs  <<S// 0D CZD2I & . .rxx 83 >D //$ C __S !F ~Ez!!&)JAq!Q 99Ra=DAq1uQAEQU?AQEz4!8|q4!8|Ez q1q5y!a!e)# $d **rctj|tj}tj|ddd}tj|dt j dz ddd }||d fSg}|d d D]e}|d \}}}} ||z } | |z } | d k(rt jt j| | } d| cxkrdksRnU|j| gt|dkr|d fStt j|} t| dkr|d fS|jd d\}}tj|dz |dz f| d}tj ||||ftj"tj$}|dfS)zrDetect dominant near-horizontal lines (EKG grid) and rotate to align. Falls back to identity on any ambiguity.2) apertureSizex ) threshold minLineLength maxLineGapNFrig?r@g?)flags borderModeT)r+r,rBCanny HoughLinesPr)pidegreesarctan2r/lenfloatmedianabsrHgetRotationMatrix2D warpAffine INTER_CUBICBORDER_REPLICATE)r7rIedgeslinesangleslinex1y1x2y2dxdyangrjrQrRmatrixrotateds rr0r0Zs <<S// 0D IIdB! 4E OO q"%%#+CB E }EzFds aBB "W "W 7 jjB+, >r> MM#  6{REz 299V$ %F 6{SEz 99Ra=DAq  $ $a!eQU^VS AFnn  Aoo'' G D=rc@tj|tj}tj|\}}}tjdd}|j |}tjtj |||gtjS)zqMild CLAHE on the L channel of LAB. Conservative clip limit so the parchment-on-grid look stays recognizable.g?)r~) clipLimit tileGridSize)r+r, COLOR_BGR2LABsplit createCLAHEapplymerge COLOR_LAB2BGR)r7lablabclahel2s rr1r1sp ,,sC-- .CiinGAq! OOc ?E QB << 2q!*-s/@/@ AAr)r4z str | Pathrr )r7 np.ndarrayrztuple[np.ndarray, bool])r7rrr)__doc__ __future__rr dataclassesrpathlibrr+numpyr)PILrr r;r.r0r1r"rrrsW # !  $   S@+("JBr