§
    Û×jÛ  ã                   óT   — d dl mZmZmZmZ d dlmZmZ dZdZ	dZ
 G d„ d¦  «        ZdS )	é   )ÚWORLDMIRROR_DEEPY_INFOSÚWORLDMIRROR_DEEPY_PROMPT_INFOSÚWORLDMIRROR_INFOSÚWORLDMIRROR_PROMPT_INFOS)ÚCUSTOM_SETTINGSÚMODEL_MODESÚworldmirror_2zDeepBeepMeep/3DÚworldmirrorc                   óî   — e Zd Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Z	edd„¦   «         Z
ed	„ ¦   «         Zedd„¦   «         Zed„ ¦   «         ZdS )Úfamily_handlerc                  ó   — t           gS ©N)ÚARCHITECTURE© ó    úNC:\Users\johnn\OneDrive\Documents\AI\models\worldmirror\worldmirror_handler.pyÚquery_supported_typesz$family_handler.query_supported_types   s
   € åˆ~Ðr   c                  ó
   — i i fS r   r   r   r   r   Úquery_family_mapsz family_handler.query_family_maps   s   € à�2ˆvˆr   c                  ó   — dS )Nr
   r   r   r   r   Úquery_model_familyz!family_handler.query_model_family   s   € àˆ}r   c                  ó
   — ddiS )Nr
   )ià  ÚWorldMirrorr   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € àÐ4Ð5Ð5r   c                 ó   — | S r   r   )Úbase_model_types    r   Úget_lora_dirzfamily_handler.get_lora_dir   s   € àÐr   c                 óF  — i dd“dd“dd“dddgddd	œ“d
d“dd“dddgddddœ“dd“dd“dd“dd“dd“dd“dd“dd“dd“d d“d!i t           ¥d"t          t           d"         ¦  «        i¥d#„ t          D ¦   «         g i dd$œ| gd%d&g d'¢d(d)œgt          t          t
          t          d*œ¥S )+NÚimage_outputsTÚmesh_outputsÚdevice_explicitÚimage_ref_choices)ÚPhotosÚI)z	No PhotosÚ r$   )ÚchoicesÚletters_filterÚdefaultÚno_background_removalÚfit_into_canvas_image_refsé    Úguide_custom_choices)zNo Videor%   )úReference VideoÚVr.   r%   ÚVideo)r&   r'   r(   ÚlabelÚreference_video_enabledÚskip_video_guide_preprocessÚvideo_guide_labelr-   Úprompt_descriptionz8Scene Name / Description (optional, used for file names)Úprompt_optionalÚno_negative_promptÚinference_stepsFÚlock_inference_stepsÚguidance_max_phasesÚembedded_guidancer   r&   c                 ó6   — g | ]}|                      ¦   «         ‘ŒS r   )Úcopy)Ú.0Úones     r   ú
<listcomp>z2family_handler.query_model_def.<locals>.<listcomp>5   s    € ÐFÐFÐF¨s §¢¡
¤
ÐFÐFÐFr   )Ú	selectionÚlabelsr(   Únativez3D reconstruction)zgaussian splatsÚ3DGSÚphotogrammetryzpoint cloudzcamera poseszvideo to 3DzrReconstructs a static scene as Gaussian splats or a point cloud, with the cameras, from several photos or a video.)ÚnameÚaliasesÚdescription)Úimage_batch_size_maxÚmodel_modesÚcustom_settingsÚprompt_enhancer_defÚprofiles_dirÚacceleratedÚspecialitiesÚinfosÚprompt_infosÚdeepy_infosÚdeepy_prompt_infos)r   Úlistr   r   r   r   r   )r   Ú	model_defs     r   Úquery_model_defzfamily_handler.query_model_def   s¢  € ð
Ø˜Tð
à˜Dð
ð ˜tð
ð  ¨oÐ?PÐ-QÐehÐuxÐ!yÐ!yð	
ð
 $ Tð
ð )¨!ð
ð #Ð1AÐC[Ð0\Ðpsð  ACð  NUð  %Vð  %Vð
ð & tð
ð *¨4ð
ð  Ð!2ð
ð !Ð"\ð
ð ˜tð
ð ! $ð
ð ˜uð
ð # Dð
ð  " 1ð!
ð"   ð#
ð$ %&ØS�kÐS¨9µd½;ÀyÔ;QÑ6RÔ6RÐSÐSØFÐFµoÐFÑFÔFØ13¸rÈbÐ#QÐ#QØ,Ð-Ø#Ø&9ð  Gbð  Gbð  Gbð .bðcð cð då&Ý4Ý2Ý"@ð;
ð 
ð 
ð 	
r   Nc                 ó,   — t           t          gg d¢gdœgS )N)zworldmirror_2_heads.safetensorszLicense.txtz
Notice.txt)ÚrepoIdÚsourceFolderListÚfileList)ÚREPOÚPROJECT)ÚcomputeListr   rT   s      r   Úquery_model_filesz family_handler.query_model_filesA   s=   € åµg°Yð  NNð  NNð  NNð  MOð  Pð  Pð  Qð  	Qr   c                 óþ   — |d         }d|vrd|vrdS d|v rt          |d         ¦  «        nd}t          |d         pi                      dt          d         d	         ¦  «        ¦  «        }d|v r||k    r	d
|› d|› d�S d S )NÚvideo_prompt_typer$   r.   z4WorldMirror needs Photos, a Reference Video or both.Ú
image_refsr+   rJ   Ú	max_viewsr(   zThe z! Photos already reach Max Views (zB): raise Max Views or remove photos to add Reference Video frames.)ÚlenÚintÚgetr   )r   rT   Úinputsr_   Úimagesra   s         r   Úvalidate_generative_settingsz+family_handler.validate_generative_settingsE   sÊ   € à"Ð#6Ô7ÐØÐ'Ð'Ð'¨CÐ7HÐ,HÐ,HØIÐIØ.1Ð5FÐ.FÐ.F•�V˜LÔ)Ñ*Ô*Ð*ÈAˆÝ˜Ð 1Ô2Ð8°b×=Ò=¸kÍ?Ð[\ÔK]Ð^gÔKhÑiÔiÑjÔjˆ	ØÐ#Ð#Ð#¨°)Ò(;Ð(;ð Q˜&ð  Qð  QÀ9ð  Qð  Qð  Qð  QØˆtr   Fc                 óX   — ddl m}  || |||¬¦  «        }||                     ¦   «         fS )Nr   )Úmodel_factory)Úsave_quantized)Úworldmirror_mainri   Úpipe)Úmodel_filenameÚ
model_typer   rT   rj   Úkwargsri   Úmodels           r   Ú
load_modelzfamily_handler.load_modelP   s@   € à3Ð3Ð3Ð3Ð3Ð3à�˜n¨j¸/ÐZhÐiÑiÔiˆØ�e—j’j‘l”lÐ"Ð"r   c           	      óT   — |                      ddt          d         ddddœ¦  «         d S )Nr   r$   r(   r+   r%   )Ú
image_moder_   Ú
model_modeÚ
batch_sizeÚremove_background_images_refÚprompt_enhancer)Úupdater   )r   rT   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsW   sE   € à×ÒØØ!$Ý% iÔ0ØØ,-Ø!ð
ð 
ñ 	ô 	ð 	ð 	ð 	r   r   )F)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r   r   r   r   rU   r]   rg   rq   rz   r   r   r   r   r   
   s/  € € € € € Øðð ñ „\ðð ðð ñ „\ðð ðð ñ „\ðð ð6ð 6ñ „\ð6ð ðð ñ „\ðð ð
ð 
ñ „\ð
ðB ðQð Qð Qñ „\ðQð ðð ñ „\ðð ð#ð #ð #ñ „\ð#ð ðð ñ „\ðð ð r   r   N)Úpromptsr   r   r   r   Úsettingsr   r   r   rZ   r[   r   r   r   r   ú<module>r�      sŒ   ðà yÐ yÐ yÐ yÐ yÐ yÐ yÐ yÐ yÐ yÐ yÐ yØ 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2à€Ø€Ø
€ðVð Vð Vð Vð Vñ Vô Vð Vð Vð Vr   