j< UddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl ZddlmZddlmZmZmZhdZhdZd Zd Zd Zd Zd ZdZddddddddddd ZejZda da!da"da#da$da%da&da'da(iZ)de*d<dZ+dZ,d:dZ-dZ.d:d Z/d!Z0d"Z1d#Z2d$Z3d%Z4d&Z5d'Z6d(Z7d)Z8d*Z9d:d+Z:d,Z;d:d-Zd0Z?d1Z@d2ZAd3ZBd4ZCd5ZDd6ZEd7ZFd8ZGeGgd9ZHdS);) annotationsN)Path)quote)FileData)_build_hdr_tonemap_filterprobe_video_stream_metadataresolve_media_binary>.mp4.ogg.ogv.webm> .mkvh264rh265rhevc.movrrrrrr rr rr rr theorar rr vp9i24veryfastTAV1DNxHDFFV1zH.264HEVCProResTheoraVP8VP9) av1dnxhdffv1rrrproresrvp8r Fzdict[str, str]_preview_to_sourceczt|to&|dS)N)zhttp://zhttps://) isinstancestrlower startswith)values CC:\Users\johnn\OneDrive\Documents\AI\shared\gradio\video_preview.py_is_http_url_liker:Cs. eS ! ! Wekkmm&>&>?V&W&WWctttj|SN)r5rosfspathresolvepaths r9 _abs_pathrCGs, tBIdOO$$,,.. / //r;c|p4ttjddptj}t |dz }|dd|S)N GRADIO_CACHEwangp_video_previewsT)parentsexist_ok)getattrgrVideotempfile gettempdirrmkdir) cache_dirbase directorys r9 _cache_dirrRKsY  Z NDAA ZXEXEZEZDT 33I OOD4O000 r;c|}td|d|jd|j}t j|dddS)N|utf-8) stat_PREVIEW_CACHE_VERSIONr@st_size st_mtime_nshashlibsha1encode hexdigest) source_pathrWdatas r9_source_signatureraRsx     D$ ` `{':':'<'< ` `t| ` `dN^ ` `D < G,, - - 7 7 9 9#2# >>r;crt|}|jddpd}t||d|dz S)NPvideo_r )rastemrR)r_rOsuffixrfs r9_preview_path_forrhXsI { + +F  CRC +GD i d#9#9V#9#9#9 99r;c<||jdzS)Nz .source.json) with_suffixrg) preview_paths r9 _sidecar_pathrl^s  # #L$7.$H I IIr;ct|}t|}|t|< tt|t jd|iddS#t$rYdSwxYw)NsourcerUencoding)rCr2rlr write_textjsondumpsOSError)r_rkrnpreviews r9_remember_previewrvbs { # #F %%G"(w d7mm$$// Hf;M0N0NY`/aaaaa      sAA11 A?>A?ct|ttjfs|Stj|rt |r|St |}|tvr t|Stt|}| s|S tj | dd}n#ttjf$r|cYSwxYw|r |t|<|S|S)NrUrorn)r4r5r>PathLiker?r:rCr2rlris_filerrloads read_textgetrtJSONDecodeError)rBrusidecarrns r9_restore_preview_pathrls dS"+. / / 9T??/55 ooG$$$!'**DMM**G ??   G--w-??@@DDXNN T) *  &,7# Ks!;CC87C8ct|dpddvo!t |dS)N codec_name)rr needs_tonemap)r5r|stripr6bool)metadatas r9 _is_hdr_hevcrs] x||L))/R 0 0 6 6 8 8 > > @ @DT T |Y]^f^j^jkz^{^{Y|Y||r;ctt|pi}t|dpd}t |dkr2|j|ftvot| S|jtvS)Nrrr) rr5r|rr6lenrg_BROWSER_PLAYABLE_VIDEO_CODECSr_BROWSER_VIDEO_EXTENSIONS)rBrrs r9_is_browser_playablers*3t9955;HX\\,//5266<<>>DDFFJ : !!##Z04RRq[ghp[q[qWqq ;    "; ;;r;ctjt|d}t|o|dS)Nrzvideo/) mimetypes guess_typer5rr7)rB mime_types r9_is_video_pathrs>$SYY//2I  r?ryrr) video_pathrBs r9needs_fast_video_previewrsq *:66u  *%% & &D <<>>u $  u#D)) ))r;c0t|pidpd}t|pidpd}|dkrP|dr)|dddS|tvr t|St|dkr|nd S) Nrr codec_profiler.DNXHRDNxHRrz Unknown Codec) r5r|rr6upperr7replace_CODEC_DISPLAY_NAMESr)rrrs r9_format_codec_labelrshn")),77=2>>DDFFLLNNJR,,_==CDDJJLLMW!4!4!6!6!A!A'!J!J""$$,,WgqAAA)))#J//!$Z1!4!4:     /Ir;ct|pidpdt |}t |jdpd}t|rdStdkr)tfdtDr d|d |d Sd |d S) Nrr. ContainerzHDR Video, SDR Previewrc3*K|] \}}|kVdSr=).0re known_codecrs r9 z(_format_preview_label..s,"r"rK;*#<"r"r"r"r"r"rr;z Container z Not Supported for z, Low Res PreviewzCodec z Not Supported, Low Res Preview) r5r|rr6rrrglstriprrranyr)r_r codec_labelcontainer_labelrs @r9_format_preview_labelrshn")),77=2>>DDFFLLNNJ%h//K;''.55c::@@BBQkOH('' :s"r"r"r"rSq"r"r"rrr^O^^ ^^^^ @K @ @ @@r;ctt|pi}t|dp|dpd}|tkr dtd}nd}t ||}|ddd d d d }d |d}|dr%dt|dznd}|d||S)N display_widthwidthrzscale=z:-2:flags=fast_bilinearz5scale=trunc(iw/2)*2:trunc(ih/2)*2:flags=fast_bilinear\z\\:z\:'z\'zdrawtext=text='zb':fontcolor=white@0.9:fontsize=max(11\,h/44):box=1:boxcolor=black@0.45:boxborderw=6:x=w-tw-12:y=12r,z,format=yuv420p,r) rr5intr|_PREVIEW_MAX_WIDTHrrjoinr)r_rr scale_filtertext label_filtertonemap_filters r9_preview_video_filterrs'*3{+;+;<<BH  _--Kg1F1FK! L LE !!!K 2KKK N h 7 7D <<f % % - -c5 9 9 A A#u M MDOTOOOL[c[g[ghw[x[xASXX7AABBEWWWAN ; ;^ ;\ ; ;;r;c t|sdStd}|dSttj|}t ||}|rt||t|S| |j dz|j z}g|dddddt|dd dd d ttd t|d ddtdt dddddddddddt|}t"5|r+t||t|cdddS t%j|ddddd }n#t($rd}YnwxYw||jd!ks|s6 |d"n#t($rYnwxYw ddddStj||dddn #1swxYwYt||t|S)#Nffmpegz.tmpz-yz-verrorz-nostdinz-iz-mapz0:v:0z0:a:0?z-tz-vfz-c:vlibx264z-presetz-crfz-pix_fmtyuv420pz-c:aaacz-b:a96kz-ac2z -movflagsz +faststartz -shortestTrUignoreF)capture_outputrrperrorscheckr) missing_ok)rr rr>r?r@rhryrvr5 with_namerfrg_PREVIEW_SECONDSr_PREVIEW_PRESET _PREVIEW_CRF_lock subprocessrunrt returncodeunlinkr)rrOrr_rktmp_pathcommandresults r9ensure_fast_video_previewrs #J / /t !( + +F ~try,,--5577K$[)<V.!338;K;K;M;M3 40000    ++++++++ 8\***+++++++++++++++k<000 |  s`3H4F*)H4* F96H48F99$H4G54H45 H?H4HH4H44H8;H8cDdtt|dzS)Nz/gradio_api/file=r)safe)rrCrAs r9_gradio_file_urlrs yR!@!@!@ @@r;c|t|dds|S|j}t|j|}||S||_t||_|jpt |j|_d|_|S)NrB video/mp4) rIrBrrurl orig_namernamer) file_datarOr_rks r9_attach_previewrs 64 @ @.K,Y^YGGL!IN$\22IM#-Gk1B1B1GI%I r;ct|ts|Sg}|D](}t|}||p|)|Sr=)r4listrappend)pathsreplaced_pathsrBrks r9#_replace_upload_paths_with_previewsrs\ eT " " N44066 l2d3333 r;c:|t|s|j|jrt||St t j|}t||j}|t||St|t||j dS)Nr)rBrrr) r:format watermark_original_video_format_videorr>r?rrErrr)selfrdrrks r9_patched_video_format_videor s })%00}DK4Kt~4K+D%888bi&&''J,Z9JKKL+D%888 +;L+I+IU_Udp{ | | ||r;ct||}|+t|ddt|j|j|S)Nrd)_original_video_postprocessrIrrdrE)rr8r`s r9_patched_video_postprocessr*sB &tU 3 3D GD'488D D$5666 Kr;c|Jt|dd9t|jddr#t|jj|j_t ||}t|S)NrdrB)rIrdrrB_original_video_preprocess)rpayloadrs r9_patched_video_preprocessr1sdww>>JwW^WdflnrOsOsJ27=3EFF  'g 6 6F  ( ((r;ct||}t|dgD]*}t|dd}|t||j+|S)Nrootrd)_original_gallery_postprocessrIrrE)rr8r`itemrds r9_patched_gallery_postprocessr8s\ (u 5 5Dfb))66gt,,   E4#4 5 5 5 Kr;c|St|dgpgD]?}t|dd}|*t|ddrt|j|_@t||}||Sd|DS)NrrdrBc6g|]\}}t||fSr)r)rmediacaptions r9 z/_patched_gallery_preprocess..Js* O O Ow "5 ) )7 3 O O Or;)rIrrB_original_gallery_preprocess)rrrrdr`s r9_patched_gallery_preprocessrAsGVR006B ? ?DD'400E WUFD%A%A 25:>> 'g 6 6D | O O$ O O OOr;cz|*t|ddrt|j|_t||SNrB)rIrrB"_original_file_process_single_filerrs r9!_patched_file_process_single_filerMs:FD!A!A.y~>>  -dI > >>r;cz|*t|ddrt|j|_t||Sr)rIrrB+_original_upload_button_process_single_filers r9*_patched_upload_button_process_single_filerSs:FD!A!A.y~>>  6tY G GGr;cvt|ddrdS|dd}d|_dS)N&_wangp_video_preview_upload_middlewareFhttpcK||d{V}|jdks*|jjdr |jdkr|Sd}|j23d{V}||z } 6 t j|}nL#t j$r:ddl m }|||jt|j |j |jcYSwxYwt|}||kr7ddl m }|||jt|j |j |jSddlm}d |j D} |||j| |j S) NPOSTz/uploadr;r)Response)content status_codeheaders media_type background) JSONResponsecFi|]\}}|dv||S)> content-typecontent-length)r6)rkeyr8s r9 zf_install_upload_preview_middleware.._wangp_video_preview_upload_middleware..rsRCCC*#uCIIKK`BMBMB3MBMBMBr;)r rr)methodrrBendswithr  body_iteratorrrrzr}fastapir dictrrrrfastapi.responsesritems) request call_nextresponsebodychunkrr rrrs r9rzR_install_upload_preview_middleware.._wangp_video_preview_upload_middleware]s"7++++++++ >V # #7;+;+D+DY+O+O #S[SgknSnSnO#1       % EMDD2 lJt$$EE# l l l ( ( ( ( ( (8Dh6JTXYaYiTjTjwxKX`Xklll l l l l=UCC U " " ( ( ( ( ( (8Dh6JTXYaYiTjTjwxKX`Xklll l222222CC0@0F0F0H0HCCC|N8LV]jrj}~~~~sA!#A88ACCT)rI middlewarer)apprs r9"_install_upload_preview_middlewarer%YsUs8BH5BH3BH#DBG +UBO(9BJ7BJ!"566CNJJJr;)rr;rr=)I __future__rr[rrrr>rrL threadingpathlibr urllib.parsergradiorJgradio.data_classesrshared.utils.video_decoderrr rrrrrrrXr.rLockrr/rrrrrr'rrr2__annotations__r:rCrRrarhrlrvrrrrrrrrrrrrrrrrrrrr%r*r;__all__rr;r9rFs"""""""  ((((((rrrrrrrrrr>== " " " !              #"!%)".2+ $#%'''''XXX000??? :::: JJJ   *}}}<<<>>> ***JJJAAA < < <????DAAA    }}}))) P P P??? HHH 666> :  O N Nr;