/fTddlZddlZddlZddlZddlZddlZddlZddlZddlZ ddl Z ejdZ ejdZejdZejdZejdZej$j'ej$j)eZej$j/edZdZej4ed d Zd Zd ZdZdZdZ dZ!ejDddZ#dZ$y)NApiUrlOpenIdApiKeyOssUrlRegionsDatastmpTexist_okctjjtd}g}t tj |}|D]7}|}tjj||}|j ||g9|ddd}|S)N ClothImgs)ospathjoindata_dirsortedlistdirappend) cloth_direxamplesfilesfcloth_id cloth_paths k/Users/zhengchengyao/Documents/MeshFitSolver/samples/projs/mm/ClothCodes/huggingface/IDM-VTON-PLUS/utils.pyget_cloth_examplesrs{ X{3IH 2::i( )E 0WW\\)Q/ :./0"~H Octjjtd}g}tj|D]7}|}tjj||}|j ||g9|S)NPoseImgs)rrrrrr)pose_dirrrpose_id pose_paths rget_pose_examplesr$(sdww||Hj1HH ZZ !.GGLL1- ),-. Orctjjtd}tjj||d|}|S)N ResultImgs_)rrrr)rr" result_dirres_paths rget_result_exampler*1s:h 5Jww||J8*AgY(?@H Orc |||dd}tj}tj|dtj|}g}|j dk(rwd|jvre|jdd}|D]J} |j t| dzd zt| d zd zt| d t| d d L|S)Nr)openIdapiKeyipIdpagez/api/inf/get_batch_fast_resultsdatar1recordsbodyUrlz?thumbnail/768x768>showUrlstateposition)poseresr6r7) requestssessionpostjsondumps status_coderrint) apiUrlr,r-clientIpparamsr;retr9r3records rgetAllFastInfsrF6sx JF G --6("ABTZI[ \C C  SXXZ hhj(3G! 9 6&*;#;r?openputexistsremove)rAr,r-rB cloth_image pose_imagefolderr#rrCr;rDr9r1fileresponses r upload_imgsrcEs WW\\)X%5%5c2%> ?FKK& VZ0Ifk2JKK :a$B$h/0KK K!DbD12xww'' 2((4F G --6("9:FAS TC C  SXXZ 88:f%DE{f$T%[&8j$' E4#<<Z(8tD Ei& D$#<<YdC Dww~~i("))I*>ww~~j)299Z+@= J E E D Ds$I%I1%I.1I:c t|dk(rd}n|dk(rd}n|dk(rd}n|dk(rd}tt||dd||d }tj}tjt d t j| } | jd k(r2d | j vr td| j d Syy)N upper_cloth lower_clothrNdressesrO full_body)r,r-rS denoise_steps auto_mask auto_cropcategorycaptionz/api/inf/public_fastinfr0r2r1zpublic task successfully!) rrr:r;r<rr=r>r?print) rAr,r-rSrnrorkrCr;rDs rpublicFastSwaprqas =  9  ; %1!w0F G --6("9:FAS TC  SXXZ   - .88:f% % rc|||d}tj}tj|d|}|jdk(r&d|j vry|j dSy)N)r,r-rSz/api/inf/get_fast_result)rCr2r1r)r:r;getr?r=)rAr,r-rSrCr;rDs r getFastInfResrtxse >F G ,,&!9:6 JC   #xxz&!!r ctj}tjd|}|jdD]=}|jd|d}|tvrt |dyt |d?y)Nz2https://webapi-pc.meitu.com/common/ip_location?ip=r1nationinvalidFvalidT)r:r;rsr=rrp)ipr;rDknats r check_funcr}s} G ,,KB4P QC XXZ  hhj #H- '> #y ! #w    rc\ t|S#t$r}t|Yd}~yd}~wwxYw)NT)r} Exceptionrp)rzes r check_warprs+"~  as + &+)%rsysrWr=randomtimer: func_timeoutnumpynpgradiogrenvironrrrrrrdirnameabspath__file__proj_dirrrrTrVrr$r*rFrcrqrtfunc_set_timeoutr}rrrrs  H  H  H  H  **Y  77??277??84 5 77<<' *   I%  8&. r" # r