o @Tj@s2UdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZdd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZeeZededgZ de!d<hdZ"Gddde Z#Gddde Z$Gddde Z%d-ddZ&d.d d!Z' "  d/d0d+d,Z(dS)1aWardrobe seeding: populate the unknown user's wardrobe from preset images. Shared orchestration used by both the CLI script (scripts/seed_wardrobe.py) and the admin HTTP endpoint (POST /api/v1/admin/seed). Modes: - "reset": wipe DB + storage (preserves GLOBAL_USER_ID), then seed - "add": keep existing data, append new garments to the unknown user Ghost mannequin is always generated so VTON works immediately after seeding. ) annotationsN)Callable)Path)Literal) BaseModel)deleteselect) AsyncSession)GLOBAL_USER_IDsettings) FaceEmbedding)Garment)User)GarmentServicez,/home/ubuntu/ootd-poc/resources/men_garmentsz./home/ubuntu/ootd-poc/resources/women_garments list[Path]DEFAULT_RESOURCE_DIRS>.jpg.png.jpeg.webpc@s.eZdZUded<ded<ded<ded<dS) ResetSummaryintdeleted_garments deleted_faces deleted_users deleted_filesN__name__ __module__ __qualname____annotations__r!r!9/home/ubuntu/htdocs/ootd-api/app/services/seed_service.pyr*s  rc@s&eZdZUded<ded<ded<dS) SeedErrorstrfilez#Literal['read', 'analyze', 'ghost']stagemessageNrr!r!r!r"r#1s  r#c@sReZdZUded<dZded<dZded<dZded<gZded <d Zd ed <d S) SeedSummaryLiteral['reset', 'add']moderrimages_processedgarments_createdghosts_generatedzlist[SeedError]errorsNzResetSummary | None reset_summary) rrrr r+r,r-r.r/r!r!r!r"r(7s     r(dbr returncst|ttIdH}|ttIdH}t|}t|}|ttIdH|ttIdH|tt tj t kIdH}t |}t|}|D] }||IdHqW| IdHd} ttj} dD]6} | | } | s}qr| D]&} | r| tdd| dD7} t| q| r| | d7} qqrtd|||| t|||| d S) zDelete all garments/faces/non-global users and wipe storage dirs. Preserves the unknown user (GLOBAL_USER_ID). Shared by /admin/reset and the seed service's reset mode. Nr) originalsghost_mannequinfaces segmentationcss|] }|rdVqdS)N)is_file).0fr!r!r" ^sz!reset_all_data..*r6z8Reset: %d garments, %d faces, %d users, %d files deleted)rrrr)executerr scalarsallr lenrrwhereidr listcommitrr UPLOAD_DIRis_diriterdirsumrglobshutilrmtreer7unlinkloggerinfor)r0garmentsr4 n_garmentsn_facesresultnon_global_usersn_usersurbasesubsub_diritemr!r!r"reset_all_data@sL    rYdirscCsZg}|D]&}|std|qt|D]}|r)|jtvr)| |qq|S)Nz*Resource directory not found, skipping: %s) rErLwarningsortedrFr7suffixlowerIMAGE_EXTENSIONSappend)rZpathsdpr!r!r"_collect_imagesqs  rdaddgarment_servicerr*r) resource_dirslist[Path] | None progress_cbCallable[[str], None] | Nonec s<|dur|nt}t|d}|dkrt|IdH|_t|}t|}|dkr.|r,|d|St|dD]\} } d| d|d | j} |rL|| d z| } Wn!t ys} z|j t t | d t | d WYd} ~ q3d} ~ wwz |j|| dd IdH}Wn'ty} ztd| |j t t | dt | d WYd} ~ q3d} ~ ww|jd7_|s|j t t | ddd q3|jt|7_|D]O}|r|| d|jd|jdz|||jIdH|jd7_Wqty} ztd|j|j t t | dt | d WYd} ~ qd} ~ wwq3|S)u Seed preset garment images into the unknown user's wardrobe. Each image is analyzed (LLM), saved to DB, and has a ghost mannequin generated. Per-image failures are recorded and do not abort the batch. Note: `add` mode is not idempotent — rerunning creates duplicates. N)r*resetrz(No images found in resource directories.r6[/z] u — analyzingread)r%r&r')user_idzanalyze_and_save failed for %sanalyzezLLM returned 0 garmentsu — ghost ()z$generate_ghost failed for garment %sghost)rr(rYr/rdr? enumeratename read_bytesOSErrorr.r`r#r$analyze_and_save ExceptionrL exceptionr+r, category_main category_subgenerate_ghostrAr-)r0rfr*rgrirZsummaryratotalidxpathlabel image_byteserNgr!r!r" seed_wardrobe}sl      r)r0r r1r)rZrr1r)reNN) r0r rfrr*r)rgrhrirjr1r())__doc__ __future__rloggingrIcollections.abcrpathlibrtypingrpydanticr sqlalchemyrrsqlalchemy.ext.asyncior app.configr r app.models.facer app.models.garmentr app.models.userrapp.services.garment_servicer getLoggerrrLrr r_rr#r(rYrdrr!r!r!r"s:             1