
    +j;                        S SK Jr  S SKJrJr   " S S\5      r " S S\5      r " S S\5      r " S	 S
\5      r " S S\5      r	 " S S\5      r
 " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S \5      r " S! S"\5      r " S# S$\5      r " S% S&\5      r " S' S(\5      rg))*    )Optional)	BaseModelFieldc                      \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   \" S5      r	\\S'   \" S5      r
\\S	'   \" S
SS9r\\S'   \" SSS9r\\S'   \" S5      r\\S'   \" S5      r\\   \S'   \" S5      r\\   \S'   \" S5      r\\S'   \" SSS9r\\   \S'   \" SSS9r\\S'   \" SSS9r\\S'   \" SSS9r\\S '   S!rg)"ImageEnhanceRequest   	Reimaginemodeljpegoutput_formatAllsubject_detectionTface_enhancementr   z'Is ignored if face_enhancement is falsedescriptionface_enhancement_creativityg?face_enhancement_strength.
source_urlNoutput_widthoutput_heightFcrop_to_fillz+Text prompt for creative upscaling guidanceprompt   z%Creativity settings range from 1 to 9
creativitytruez&To preserve the identity of charactersface_preservationzTo preserve the original colorcolor_preservation )__name__
__module____qualname____firstlineno__r   r
   str__annotations__r   r   r   boolr   floatr   r   r   r   intr   r   r   r   r   r   __static_attributes__r       :/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/topaz.pyr   r      s    {#E3#vM3&"5\s)"4[d().q>g)hh',S>g'huhCjJ "'+L(3--#(;M8C=.uL$%!$4abFHSMbA+RSJS"67_`s`#F8XYYr)   r   c                   0    \ rS rSr% \" S5      r\\S'   Srg)ImageAsyncTaskResponse   .
process_idr   N)	r   r    r!   r"   r   r.   r#   r$   r(   r   r)   r*   r,   r,      s    CjJ r)   r,   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   \" S5      r	\
\   \S'   \" S5      r\\S'   Srg)	ImageStatusResponse   .r.   statusNprogresscreditsr   )r   r    r!   r"   r   r.   r#   r$   r2   r3   r   r'   r4   r(   r   r)   r*   r0   r0      s>    CjJ *FC#DkHhsm):GSr)   r0   c                   J    \ rS rSr% \" S5      r\\S'   \" S5      r\	\S'   Sr
g)ImageDownloadResponse"   .download_urlexpiryr   N)r   r    r!   r"   r   r8   r#   r$   r9   r'   r(   r   r)   r*   r6   r6   "   s    c
L#"*FCr)   r6   c                   J    \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   Sr	g)
Resolution'   .widthheightr   N)
r   r    r!   r"   r   r=   r'   r$   r>   r(   r   r)   r*   r;   r;   '   s    sE3*FCr)   r;   c                       \ rS rSr% \" S5      r\\S'   \" SSS9r\	\S'   \" SSS9r
\	\S'   \" SS	S9r\	\S
'   \" S5      r\	\S'   \" S5      r\\S'   Srg)CreateVideoRequestSource,   .	containerzSize of the video file in bytesr   sizez%Duration of the video file in secondsdurationz#Total number of frames in the video
frameCount	frameRate
resolutionr   N)r   r    r!   r"   r   rB   r#   r$   rC   r'   rD   rE   rF   rG   r;   r(   r   r)   r*   r@   r@   ,   sb    3ZIsc'HID#I#+RSHcSC-RSJS3ZIs"3ZJ
'r)   r@   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\	\
   \S'   \" S5      r\
\S'   \" S5      r\\S'   \" S5      r\\S'   S	rg)
VideoFrameInterpolationFilter5   .r
   Nslowmofps	duplicateduplicate_thresholdr   )r   r    r!   r"   r   r
   r#   r$   rK   r   r'   rL   rM   r%   rN   r&   r(   r   r)   r*   rI   rI   5   sM    sE3!$KFHSM'SzCCjIt !&s+r)   rI   c                   4   \ rS rSr% \" S5      r\\S'   \" SSS9r\	\   \S'   \" SSS9r
\	\   \S	'   \" SS
S9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\	\   \S'   \" SSS9r\\-  S-  \S'   \" SS S9r\	\   \S!'   \" SS"S9r\S-  \S#'   \" SS$S9r\S-  \S%'   \" SS&S9r\S-  \S''   S(rg))VideoEnhancementFilter=   .r
   NzAuto, Manual, Relativer   autoz9Downscales video input for correction of blurred subjectsfocusFixLevelz Strength of compression recoverycompressionzAmount of detail reconstructiondetailsz8Amount of noise to add to input to reduce over-smoothingprenoisezAmount of noise reductionnoisezAmount of halo reductionhaloz%Anti-aliasing and deblurring strengthpreblurzAmount of sharpness appliedblurzGrain after AI model processinggrainzSize of generated grain	grainSizez$Source details into the output videorecoverOriginalDetailValuez>slc-1/slp-2.5: enum (low/middle/high). ast-2: decimal 0.0-1.0.r   z/Set to true for Starlight Creative (slc-1) onlyisOptimizedModez%Descriptive scene prompt (ast-2 only)r   zast-2 pre-enhance sharpnesssharpzast-2 realism controlrealismr   )r   r    r!   r"   r   r
   r#   r$   rR   r   rS   rT   r&   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r   r^   r%   r   r_   r`   r(   r   r)   r*   rP   rP   =   s   sE32JKD(3-K#(;v#wM8C=w#(;]#^K%^$T7XYGXe_Y %d8r sHhuos"45PQE8E?Q!$4NOD(5/O$T7^_GXe__!$4QRD(5/R"45VWE8E?W!&t9R!SIxS27Jp2qq%*4=}%~Jd"~&+D>o&pOXd^pt1XYFC$JY2OPE54<P!$4KLGUT\Lr)   rP   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\	\S'   \" SSS9r
\\   \S'   \" SSS9r\\S	'   \" SS
S9r\\S'   Srg)OutputInformationVideoR   .rG   rF   z,Required if audioTransfer is Copy or Convertr   
audioCodeczCopy, Convert, NoneaudioTransferzLow, Mid, HighdynamicCompressionLevelr   N)r   r    r!   r"   r   rG   r;   r$   rF   r'   rd   r   r#   re   rf   r(   r   r)   r*   rb   rb   R   sY    "3ZJ
'3ZIs %c7e fJfs0EFM3F#(:J#KSKr)   rb   c                   0    \ rS rSr% \" S5      r\\S'   Srg)	OverridesZ   TisPaidDiffusionr   N)	r   r    r!   r"   r   rj   r%   r$   r(   r   r)   r*   rh   rh   Z   s    !$KOT'r)   rh   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\	\
\-     \S'   \" S5      r\\S'   \" \" SS95      r\\S'   S	rg
)CreateVideoRequest^   .sourcefiltersoutputT)rj   	overridesr   N)r   r    r!   r"   r   rn   r@   r$   ro   listrI   rP   rp   rb   rh   rq   r(   r   r)   r*   rl   rl   ^   sO    ',SzF$1LQRUJGT/2HHIV%*3ZF"/ 4!@AIyAr)   rl   c                   0    \ rS rSr% \" S5      r\\S'   Srg)CreateVideoResponsee   .	requestIdr   N)	r   r    r!   r"   r   rv   r#   r$   r(   r   r)   r*   rt   rt   e   s    3ZIsr)   rt   c                   P    \ rS rSr% \" S5      r\\S'   \" S5      r\	\   \S'   Sr
g)VideoAcceptResponsei   .uploadIdurlsr   N)r   r    r!   r"   r   rz   r#   r$   r{   rr   r(   r   r)   r*   rx   rx   i   s"    #JHcCjD$s) r)   rx   c                   J    \ rS rSr% \" S5      r\\S'   \" S5      r\	\S'   Sr
g)VideoCompleteUploadRequestPartn   .partNumeTagr   N)r   r    r!   r"   r   r   r'   r$   r   r#   r(   r   r)   r*   r}   r}   n   s    :GSc
D#r)   r}   c                   6    \ rS rSr% \" S5      r\\   \S'   Sr	g)VideoCompleteUploadRequests   .uploadResultsr   N)
r   r    r!   r"   r   r   rr   r}   r$   r(   r   r)   r*   r   r   s   s    :?*M467Dr)   r   c                   .    \ rS rSr% \" SSS9r\\S'   Srg)VideoCompleteUploadResponsew   .zConfirmation messager   messager   N)	r   r    r!   r"   r   r   r#   r$   r(   r   r)   r*   r   r   w   s    *@AGSAr)   r   c                   6    \ rS rSr% \" S5      r\\   \S'   Sr	g)VideoStatusResponseEstimates{   .costr   N)
r   r    r!   r"   r   r   rr   r'   r$   r(   r   r)   r*   r   r   {   s    CjD$s) r)   r   c                   0    \ rS rSr% \" S5      r\\S'   Srg)VideoStatusResponseDownloadUrl   .urlr   N)	r   r    r!   r"   r   r   r#   r$   r(   r   r)   r*   r   r      s    SzCr)   r   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\	\
   \S'   \" S5      r\	\   \S'   \" S5      r\	\   \S'   \" S5      r\	\   \S	'   S
rg)VideoStatusResponse   .r2   N	estimatesr3    r   downloadr   )r   r    r!   r"   r   r2   r#   r$   r   r   r   r3   r&   r   r   r   r(   r   r)   r*   r   r      s\    *FC8=dIx45C %dHhuo+"2YGXc]&9>tHh56Dr)   r   N)typingr   pydanticr   r   r   r,   r0   r6   r;   r@   rI   rP   rb   rh   rl   rt   rx   r}   r   r   r   r   r   r   r)   r*   <module>r      s     %Z) Z"!Y !) I 
 
(y (,I ,MY M*LY L(	 (B B )  !) !
Y 
E EB) B!9 !Y E) Er)   