o @Tj@s2dZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z e eZdZd$d ed ed efd d Zdedejd efddZdedejd efddZdedejd efddZded efddZdedBd dfddZdejd efddZd%deded efdd Zd&deded efd"d#ZdS)'z0Image save / resize / path management utilities.N)Path)urlparse)Image)settingsi@.@urltimeoutreturnc sZt|}|jdvrtd|jpd|jstdz+tj|dd4IdH}||IdH}WdIdHn 1IdHs@wYWntjy[}ztd||d}~ww|jd kritd |j|j }t |d krvtd t |t krtd t |dt d|j dd dd }|r|ds|dkstd||S)aDownload an image from an HTTP(S) URL and return raw bytes. Performs server-side (no browser CORS), with scheme/size/content-type guards suitable for untrusted user-supplied URLs in a POC context. Raises ``ValueError`` on any failure so the caller can surface a 400. )httphttpsu)지원하지 않는 URL 스킴입니다: z(empty)u,URL 호스트를 해석할 수 없습니다.T)rfollow_redirectsNu+이미지를 가져오지 못했습니다: iu0이미지를 가져오지 못했습니다: HTTP ru빈 이미지 응답입니다.u2이미지 크기가 제한을 초과했습니다: z > z bytesz content-type;zimage/zapplication/octet-streamu5이미지가 아닌 콘텐츠입니다: content-type=)rscheme ValueErrornetlochttpx AsyncClientget RequestError status_codecontentlenMAX_IMAGE_FETCH_BYTESheaderssplitstriplower startswith)rrparsedclientresperctyper$//home/ubuntu/htdocs/ootd-api/app/utils/image.pyfetch_image_bytess< (     r& image_bytesuser_idcCVtdt|}ttj|}|jdddtjd}|||t||S)zDSave original uploaded image and return path relative to UPLOAD_DIR. originalsTparentsexist_okz.jpg rstrr UPLOAD_DIRmkdiruuiduuid4hex write_bytesr'r(rel_pathdest_dirfilenamer$r$r%save_original_image:  r:cCr))zBSave ghost mannequin image and return path relative to UPLOAD_DIR.ghost_mannequinTr+.pngr.r6r$r$r%save_ghost_mannequin_imageDr;r>cCr))zHSave VTON composite result image and return path relative to UPLOAD_DIR.vtonTr+r=r.r6r$r$r%save_vton_imageNr;r@cCsNtd}ttj|}|jdddtjd}|||t||S)zFSave segmentation result image and return path relative to UPLOAD_DIR. segmentationTr+r=) rrr0r1r2r3r4r5r/)r'r7r8r9r$r$r%save_segmentation_imageXs  rB relative_pathcCsZ|sdSttj|}z|jddtd|WdSty,tjd|ddYdSw)zDelete an image file by its path relative to UPLOAD_DIR. Logs a warning and continues if the file doesn't exist or deletion fails. NT) missing_okzDeleted image: %szFailed to delete image: %sexc_info)rrr0unlinkloggerinfoOSErrorwarning)rCfilepathr$r$r% delete_imagebs  rMc Csd}ttj}dD]E}||t|}|rN|D]"}|r=z ||d7}Wqty<t j d|ddYqwqz| Wq tyMYq wq |S)zmDelete all storage files for a user across all subdirectories. Returns the number of files deleted. r)r*r<facesr?zFailed to delete: %sTrE) rrr0r/is_diriterdiris_filerGrJrHrKrmdir)r(countbasesubuser_dirfr$r$r%delete_user_storageqs*       rYmax_sizec Cstt|}|jdvs|jdkr2d|jvr2|d}td|jd}|j || dd|}n |jdkr<|d}|j\}}t |||kr`|t ||}| t ||t ||ftj}t}|j|d d d |S) uNResize image if larger than max_size on any side. Returns JPEG bytes. RGBA inputs (e.g., pre-segmented person PNGs) are composited onto white before the JPEG encode — a plain ``convert("RGB")`` would drop alpha onto a black canvas, which downstream image models then preserve as a black backdrop in the result. )RGBALAP transparencyr\RGB)rara)maskJPEGZ)formatquality)ropenioBytesIOmoderIconvertnewsizepastermaxresizeintLANCZOSsavegetvalue) r'r[imgrgbabgwhratiobufr$r$r%resize_image_if_neededs    "r}cCstt|}|j\}}t|||kr|S|t||}t||t||f}||tj}t}|j |ddd| S)a+Resize PNG if larger than max_size on any side, preserving transparency. Returns the original bytes if no resize needed; otherwise returns re-encoded PNG bytes with the alpha channel intact (required for ghost mannequin images so the model can distinguish garment from background). PNGT)rfoptimize) rrhrirjrnrprrrqrsrtru)r'r[rvryrzr{new_sizer|r$r$r%resize_png_if_neededs r)r)rZ)r~)__doc__riloggingr2pathlibr urllib.parserrPILr app.configr getLogger__name__rHrr/floatbytesr&UUIDr:r>r@rBrMrrrYr}rr$r$r$r%s(     '