o Ej@sTddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZe ejd Zed dddZed dddZede fZddddDddZdEd&d'Z ed(dFd,d-Z!dGd0d1Z"dHd4d5Z#dId7d8Z$dJd9d:Z%dKdd?Z'dMdBdCZ(dS)N) annotationsN)Callable)Path)Any)ALLOWED_ACTIONS)$provider_call_metrics_from_build_dir) LocalPetApi)LocalJsonStore)build_pet_package_from_photos)LocalPrivateStoragetestsfixturespublicu周六-合成样例u一能静-合成样例.) build_modepackage_builder state_dirr storage_dir output_dirr str | NonerPackageBuilder | Nonereturnrcs|jddd|jddd|jdddt||p!tjdd}|dvr-t}n|dvr>|p4tdfdd }nt dt t |d||dS)NTparentsexist_okAI_PET_BUILD_RUNNERdemo>rfixturer>realmodelexternalbuilddict[str, Any]petmedialist[dict[str, Any]] output_rootrrcst||||dS)N)r!r#r$r& media_storager) production_external_build_runner)r!r#r$r&r'selected_builder;/opt/sixxie/releases/current/services/api/production_app.py build_runner-sz+create_production_api..build_runnerz,AI_PET_BUILD_RUNNER must be demo or externalzlocal-api-store.json)storer-r&r' r!r"r#r"r$r%r&rrr") mkdirr osenvirongetstriplowerproduction_demo_build_runnerr ValueErrorrr )rrrrr runner_moder-r+r)r,create_production_apis"  r9r!r"r#r$r%r&c Cst||d}t|t|d}|d}|d}|r!t||jdd|jdddD]}||} | rAt| ||q/|djt t| d pNd d d d d|dr^dnd} |d| t t ddddS)N)r#r$build_idpackageqaT)r)pet.jsonzspritesheet.webpspritesheet.pngzcontact-sheet.htmlpet_namer#z../package/spritesheet.webp) display_name spritesheetutf-8encodingr>zpackage/spritesheet.pngzpackage/spritesheet.webppackage/pet.json) build_dir package_ref preview_refactionschecksum_validruntime_compatible validation_ok) _select_public_fixturerstrexistsshutilrmtreer0copy2 write_text_demo_contact_sheetr3listr) r!r#r$r&rrF package_dirqa_dirfilenamesourcerHr+r+r,r6As4    r6)rr'r PackageBuilderc Cs|stdt|d}|jdddtjtt|dpdd|d$}t||t|d }|t|d p8d t ||t|d } Wdn1sMwYt| } | d d sbtd| dd rldnd} | |  sxtdt | } | d| t t dd| t| dS)Nz+at least one uploaded pet photo is required_private_upload_inputsTrr:r!-)prefixdir)r$r'tmp_dirr?r#)r?notes photo_pathsr&r;r=z-generated package is missing package/pet.jsonr<zcontact-sheet.pngzqa/contact-sheet.pngzqa/contact-sheet.htmlz2generated package is missing preview contact sheetrE)rFrGrHrIrJrKrLprovider_call_metrics)r7rr0tempfileTemporaryDirectory_safe_temp_prefixrNr3_materialize_private_media_pet_generation_notesrO_generated_validation_okrUrr) r!r#r$r&r'rtmp_roottmprarFrHrLr+r+r,r(_s:  (  r(r_ list[Path]c Csg}t|D]8\}}|ddvrtdt|dpd}|s$tdt|}|d||}|||||q|S)Ndeletion_status>Nactivez!active uploaded media is required storage_refz%uploaded media storage ref is missingzinput-) enumerater3r7rN _media_suffix write_bytesreadappend) r$r'r_raindexitemrnsuffix photo_pathr+r+r,rfs rfrFboolcCsP|d}|s dSz t|jdd}Wn tjy YdSw|dduS)Nzvalidation.jsonFrBrCokT)rOjsonloads read_textJSONDecodeErrorr3)rFvalidation_path validationr+r+r,rhsrhrNcCsPtt|dpdj}|dvr|Sdddddt|d p"dd S) Noriginal_filenamero>.jpeg.gif.jpg.png.webprrrr)z image/pngz image/jpegz image/webpz image/gif mime_typez.img)rrNr3rwr5)r$rwr+r+r,rqsrqcCs@t|dpdt|dpdg}ddd|DS)Npersonality_notesrotypical_gestures css|]}|r|VqdS)Nr+).0partr+r+r, sz(_pet_generation_notes..)rNr3r4join)r#partsr+r+r,rgsrgvaluecCs"ddd|Dd}|pdS)Nrocss$|] }|s |dvr|VqdS)>_r\N)isalnum)rchr+r+r,rs"z$_safe_temp_prefix..z._-r!)rr4)rcleanedr+r+r,resrecCsZdt|dp dgdd|Ddd|D}d|vs)d|vs)d |vr+tStS) Nrr?rocSg|] }t|dp dqS)rnrorNr3rrvr+r+r, z*_select_public_fixture..cSr)rrorrr+r+r,rr yinengjingu 一能静mimi)rrNr3r5YI_NENG_JING_FIXTUREZHOU_LIU_FIXTURE)r#r$haystackr+r+r,rMs  rMr@rAc Cs8dddtD}d|d|d|d|d|d S) Nrocss|] }d|dVqdS)z
  • z
    • Nr+)ractionr+r+r,rsz&_demo_contact_sheet..zD z) production preview

    us 本地生产模式预览

    公开演示素材，仅用于 production-mode 链路验证。

    z spritesheet preview
      z
    )rr)r@rArowsr+r+r,rTsrT) rrrrrrrrrrrrr/)r!r"r#r"r$r%r&rr'r rrZrr")r$r%r'r r_rrrk)rFrrry)r$r"rrN)r#r"rrN)rrNrrN)r#r"r$r%rr)r@rNrArNrrN)) __future__rr{r1rPrccollections.abcrpathlibrtypingrpackages.pet_package_schemarservices.ai.provider_metricsrservices.api.local_apirservices.api.local_storer ,services.pet_builder.photo_generation_workerr &services.storage.local_private_storager __file__resolverROOTrrrZr9r6r(rfrhrqrgrerMrTr+r+r+r,s>            &% *