+j^ 6SSKrSSKrSSKrSSKrSSKrSSKJr SSKrSSKr SSK r SSK J r SSK Jr SSKJrJrJr SSKJr SBS\S \S \ R,4S jjrS \ R,S \ R,S \ R,4SjrSSS.S\ R,S\S-S\S-S \4SjjrSCS\ R,S\S-S \ R4SjjrSDS\ R,S\S-S\S \4SjjrSES\ RS\S \4SjjrS\S\S\S \\\4S-4SjrSFS\ R,S\S \ R,4Sjjr S\ R,S\S \ R,4S jr!SDS\ R,S\S-S\S \4S!jjr"SGS"\RFS#\S$\S \4S%jjr$SHS&\RJS#\RLS-S'\RNS-S \4S(jjr(SGS)\ RRS*\S#\S$\S \4 S+jjr*S,\ R,S \ RR4S-jr+S"\RFS \4S.jr,S&\RJS/\-S \RJ4S0jr.S&\RJS1\S \RJ4S2jr/S\S\S\S \\\4S-4S3jr0S&\RJS4\S \RJ4S5jr1S&\RJS6\\\4S \RJ4S7jr2S8\ R,S \ R,4S9jr3S:\4S \54S;jr6SIS<\ R,S\ R,4S=jjr7S<\RS \ R,4S>jr8S?\S \4S@jr9S?\S \4SAjr:g)JN)BytesIO)Image)common_upscale)Input InputImplTypes)mimetype_to_extension image_bytesiomodereturnc[R"U5nURU5n[R"U5R [R 5S- n[R"U5RS5$)aConverts image data from BytesIO to a torch.Tensor. Args: image_bytesio: BytesIO object containing the image data. mode: The PIL mode to convert the image to (e.g., "RGB", "RGBA"). Returns: A torch.Tensor representing the image (1, H, W, C). Raises: PIL.UnidentifiedImageError: If the image data cannot be identified. ValueError: If the specified mode is invalid. go@r) ropenconvertnparrayastypefloat32torch from_numpy unsqueeze)r r image image_arrays @/home/wildlama/comfy/ComfyUI/comfy_api_nodes/util/conversions.pybytesio_to_image_tensorrs] JJ} %E MM$ E((5/((4u>wG  rrFcU(dSn[5nURS5SR5nUS:XaSnURX#S9 UR S5 U$)z)Converts a PIL Image to a BytesIO object.r+/r JPGJPEGformatr)rsplituppersaveseek)rFr-rN pil_formats rr3r3us\  9L%b)//1JU HH\H-a rsrc_wsrc_hcX-nX2::ag[R"X#- 5n[S[X-55n[S[X-55nXUS--nXfS--nXV4$)a Return downscaled (w, h) with even dims fitting ``total_pixels``, or None if already fits. Source aspect ratio is preserved; output may drift by a fraction of a percent because both dimensions are rounded down to even values (many codecs require divisible-by-2). Nr!)mathsqrtmaxintr]r^r,pixelsscalenew_wnew_hs r_compute_downscale_dimsrisk ]F  IIl+ ,E 3u}% &E 3u}% &E QYE QYE <rcURSS5n[URSURS[U55nUcU$UupE[ X$USS5RSS5$)aDownscale input image tensor to roughly the specified total pixels. Output dimensions are rounded down to even values so that the result is guaranteed to fit within ``total_pixels`` and is compatible with codecs that require even dimensions (e.g. yuv420p). r r r;r!lanczosdisabled)r'rir&rcr)rr,samplesdimsrgrhs rr?r?sj mmB"G "7==#3W]]15Es contiguousrAdtyperrr)rrs rrrs}}X^^A.!3PQQ~~a1A;$$Q'++-88:@@BMbjj(%,,RZZ8 rcdUSR5n[5n[R"USSS9nUR SUSS9nSUl[R RURS S 5RS S 5R5R5S URS S :XaS OSS9nUSUl S UlURUR!U55 URUR!S55 UR#5 UR%S 5 U$)Nrrmp3r libmp3lamerrirr r fltrrr)r>rrrrbit_raterrr'reshapefloatrAr&rrrrcloser[)rr output_bufferr out_streamrs raudio_input_to_mp3rsZ $$&HIMww}3uE!,,\m@T,UJ J MM & &A&&q"-335;;=!*a/vX ' E m,EEI**512**401q r duration_secc[5nSnSnUR5n[R"USS9n[R"USSS9nSnSnURGHSn[ R "SUR[U55 [U[R5(a|URSURS 9nURUl URUl S Ul[ R "S URURUR5 M[U[R5(dMURS UR S 9nUR UlUR"Ul[ R "S UR UR$5 GMV URR&SRn [)X-5n U S-S-n U S:Xa [+S5eSn Sn U(aUR-SS9H8nX:a O2UR/U5HnUR1U5 M U S- n M: UR/5HnUR1U5 M [ R "SX5 U(aUR3S5 UR-SS9HCnUR4U:a O2UR/U5HnUR1U5 M U S- n ME UR/5HnUR1U5 M [ R "SU 5 UR75 UR75 UR3S5 [8R:"U5$![<aCnUbUR75 UbUR75 [?S[AU535UeSnAff=f)a1 Returns a new VideoInput object trimmed from the beginning to the specified duration, using av to avoid loading entire video into memory. Args: video: Input video to trim duration_sec: Duration in seconds to keep from the beginning Returns: VideoFromFile object that owns the output buffer Nrr rmp4rzFound stream: type=%s, class=%sh264ryuv420pz!Added video stream: %sx%s @ %sfpsaacz%Added audio stream: %sHz, %s channelsrz7Video too short: need at least 16 frames for Moonvalleyrr z$Encoded %s video frames (target: %s)rzEncoded %s audio frameszFailed to trim video: )!rget_stream_sourcerrstreamslogginginfotype isinstance VideoStreamr average_ratersrrpix_fmt AudioStreamrrchannelsrrcrrMrrr[timerr VideoFromFile Exception RuntimeErrorstr)rrrinput_containerr input_source video_streamrstreamfpsestimated_frames target_frames frame_countaudio_frame_countrres r trim_videor6shIMOYE..0 '',S977=s5I  %--F LL:FKKf V&"..11/::6H[H[:\ %+\\ "&,mm #'0 $ @&,,PVP]P]_e_r_rsFBNN33/::5vGYGY:Z +1+=+= (&,mm # DfFXFXZ`ZiZij."%%++A.;;|12)R/25 A VW W  (//a/8/+11%8F$((09q 9'--/ $$V,0 LL? \    #(//a/8::-+11%8F$((09!Q&!9'--/ $$V,0 LL24E F   1&&}55 E  &  ! ! #  '  " " $3CF8<=1D EsD2NH>N O>O  O max_pixelsc^UR5up#[X#U5nUcU$[X5$)a Downscale a video to fit within ``max_pixels`` (w * h), preserving aspect ratio. Returns the original video object untouched when it already fits. Preserves frame rate, duration, and audio. Aspect ratio is preserved up to a fraction of a percent (even-dim rounding). )get_dimensionsri_apply_video_scale)rrr]r^ scale_dimss rdownscale_video_to_max_pixelsrs6 '')LE(zBJ e 00rcX-nX2:ag[R"X#- 5n[R"X-5n[R"X-5nUS-(aUS- nUS-(aUS- nXV4$)aOReturn upscaled (w, h) with even dims meeting at least ``total_pixels``, or None if already large enough. Source aspect ratio is preserved; output may drift by a fraction of a percent because both dimensions are rounded up to even values (many codecs require divisible-by-2). The result is guaranteed to be at least ``total_pixels``. Nr!r )r`raceilrds r_compute_upscale_dimsrsm]F  IIl+ ,E IIem $E IIem $E qy   qy   <r min_pixelsc^UR5up#[X#U5nUcU$[X5$)aUpscale a video to meet at least ``min_pixels`` (w * h), preserving aspect ratio. Returns the original video object untouched when it already meets the minimum. Preserves frame rate, duration, and audio. Aspect ratio is preserved up to a fraction of a percent (even-dim rounding). Note: upscaling a low-resolution source does not add real detail; downstream model quality may suffer. )rrr)rrr]r^rs rupscale_video_to_min_pixelsrs6'')LE&uZ@J e 00rrcUup#[5nSnSnUR5upx[U=(d U5n UR5n [R "U SS9n[R "USSS9nUR SUR5S9n X+lX;l S U l Sn URH`n [U [R5(dM$UR S U RS9n U RU lU RU l O URR S nU (a[#X~R$- 5OS nU(a[#Xx-UR$- 5OSnU(aUR'XS 9 S nUR)S S 9HnU (a4UR*bUR*U:aM)UbUR*U:a OmUR-X#S S9n[R.R1UR3S S9S S9nU R5U5HnUR7U5 M US- nM U R55HnUR7U5 M US :Xa[9SUSUS35eU bUR'S 5 UR)S S9HtnU (a:UR:bUR:U:aM)U(aUR:Xx-:a O4SUlU R5U5HnUR7U5 M Mv U R55HnUR7U5 M UR=5 UR=5 UR'S 5 [>R@"U5$![BaCnUbUR=5 UbUR=5 [ES[GU535UeSnAff=f)zNRe-encode ``video`` scaled to ``scale_dims`` with a single decode/encode pass.Nrrrrrrrrrr)rr)rsrrrWrVr z&resize produced no frames (start_time=z , duration=z" selected nothing from the source)rzFailed to resize video: )$rget_active_trim_windowboolrrrrget_frame_ratersrrrrrrrrrrc time_baser[rMrreformat VideoFramer to_ndarrayrrrrrrrrrr)rrout_wout_hrrr start_timedurationtrimmingrrrrin_video start_ptsend_ptsencodedrr audio_framers rrrsLEIMO!779JJ*(+HHG..0 '',S977=s5I'226@T@T@V2W "#(  %--F&"..11/::5vGYGY:Z +1+=+= (&,mm # .#**003 "''/;3C3Cj3P K$4$4z7L$L"& *11+>F$((0? ?'--/ $$V,0  1&&}55 G  &  ! ! #  '  " " $5c!fX>?QF GsBNJ7N O>OOwavc0URR(aU$UR[R:XaUR 5S- $UR[R :XaUR 5S- $[ SUR35e)zOConvert audio to float 32 bits PCM format. Copy-paste from nodes_audio.py file.ilzUnsupported wav dtype: )ris_floating_pointrint16rint32r)rs r_f32_pcmr(sm yy"" ekk !yy{e$$ ekk !yy{e$$ .syyk: ;;rrc [R"[U55nURR(d [ S5eURRSn[ URR5nUn/nUR=(d SnURURS9HnUR5n[R"U5n U RS:XaU R!S5n OU R"SU:wa4U R"SU:Xa!U R%SS5R'5n OAU R"SU:wa.U R)SU5R+5R'5n UR-U 5 M SSS5 W(d [ S5e[R."USS9n [1U 5n U R!S5R'5WS .$!,(df  Nb=f) z Decode any common audio container from bytes using PyAV and return a Comfy AUDIO dict: {"waveform": [1, C, T] float32, "sample_rate": int}. z"No audio stream found in response.rr )rr NzDecoded zero audio frames.r$)rr)rrrrrrrc codec_contextrrrMindexrrrrrr& transposerrtappendr(r) rafrin_srout_srframes n_channelsrarrbufrs raudio_bytes_to_audio_inputr3s % &"zzAB B!!!$F((445%'__) YYv||Y4E""$C""3'Cxx1}mmA&1+ " 0KmmAq)4461+kk"j1335@@B MM# 5 ', 566 ))F "C 3-C a(335f MM7 ' &s E9G55 HmaskcURupgpURS5nURSS5n[XXrUS9nURSS5nU(dUR S5nU(dUS:R 5nU$)zaResize mask to be the same dimensions as an image, while maintaining proper format for API calls.r r )rsrrupscale_methodcropg?)r&rr'rr@r) rrrrallow_gradientadd_channel_dim_rrrss rresize_mask_to_imagerVsz ++Au >>" D <<A D $F`d eD <<2 D ||B s !!# KrcVURS5n[R"U/S-SS9$)z]Make mask have the expected amount of dims (4) and channels (3) to be recognized as an image.r r;r$)rrr()rs rconvert_mask_to_imagerks' >>" D 99dVaZR ((rfilepathc[US5nUR5nSSS5 [R"W5R S5$!,(df  N3=f)z(Converts a text file to a base64 string.rbNrI)rreadrKrLrM)rf file_contents rtext_filepath_to_base64_stringr$qsC h vvx     L ) 0 0 99  s A  Acb[U5n[R"U5up#UcSnSUSU3$)z#Converts a text file to a data URI.zapplication/octet-streamr{r|)r$ mimetypes guess_type)rr}r-rs rtext_filepath_to_data_urir(xs<28r7sC     &44+676#65<<6(.u||.U\\.ell.( +'  <<*Tz   2  S4Z RWR]R]  + !,,!*!! !.   g 3s#%PSUXPX/\`J`" W%,, Wc WTYT`T` W  ell    QVQ]Q]  " + 6,,6*66 6&9%++99Z]9jm959%)G ;;G**T1G   d "G G4" ::   >"**4ekkg.iEekkiEiE5;;iEX 1 1# 1%++ 1ScSVhZ^H^( 1u{{ 1 1 1TGekkTGuS#XTG5;;TGn<%,,<5<<< NE Nd NL#   ,, <<*) ) ) :S:S:666r