jL d Z ddlZddlZddlmZ ddlmZ ddlmZ ed G d d Z d
Z
dS ) zFCPU encoding and browser rendering for bounded TinyVAE video previews. N) dataclass)Fraction)ImageT)frozenc 6 e Zd ZU ej ed< eed< ddZdS ) VideoPreviewimagevideoFc dt j | j d z }|rdnd}|rdnd}|rdnd}|s|dz
}d | d
| | d | d S )
Nzdata:video/mp4;base64,asciiz,display:block;max-width:100%;max-height:80vhz-display:block;max-width:100%;max-height:200pxz controls z title="Click to Pause or Resume" onclick="event.stopPropagation();window.__wangpGenerationPreviewPaused = !this.paused;this.paused ? this.play() : this.pause()" onplay="if(window.__wangpGenerationPreviewPaused) this.pause()"z;cursor:pointerz=)base64 b64encoder
decode)selfmodaluristylecontrolsplaybacks >M|"'/;;R H22 %H ' & &E is i iu} i @H i i RW i i i i N)F)__name__
__module__ __qualname__r __annotations__bytesr r r r r sG ; LLLi i i i i ir r c
dd l }t j }| |ddddi 5 }| dt | d }| d j \ |_ |_ d
|_
d |_ d d
dd|_ t | D ]k\ }} | r d d d d S |j | } || _ | | D ]}
| |
l| D ]}
| |
d d d n # 1 swxY w Y t' | d | S )Nr wmp4movflagsz*frag_keyframe+empty_moov+default_base_moof)modeformatoptionslibx264i' )rateyuv420p ultrafast25 zerolatency)presetcrftune)avioBytesIOopen
add_streamr limit_denominatorsizewidthheightpix_fmt thread_countr& enumerate
VideoFrame
from_imageptsencodemuxr getvalue) framesfps abort_checkr1 buffer containerstreamindexr framepackets r encode_videorL s
III
Z\\F c%*Fr9s t t " yB%%ihsmm6U6UV[6\6\%]]&,Qin# fm" $/mTT %f - - & &LE5{}}
" " " " " " " " M,,U33E EI --..
&
&
f%%%%
& mmoo " "F MM& ! ! ! ! "" " " " " " " " " " " " " " " q 6??#4#4 5 55s B
E
A;EEE) __doc__r r2 dataclassesr fractionsr PILr r rL r r r rQ s L L
! ! ! ! ! ! $ i i i i i i i i 6 6 6 6 6r