o e.K@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlZddlmZddlmZmZmZddlmZejsYddlmZmZmZe e d dd l m!Z!Wdn1sswYe"e#Z$d:ddZ%d;ddZ&dd$d%Z-d?d'd(Z.d@d*d+Z/d,d-Z0dAd/d0Z1dBd2d3Z2dCd5d6Z3dDd8d9Z4dS)E) annotationsN)BytesIO)Path)utils)ImageImageOpsPngImagePlugin) wasm_utils)FFmpegFFprobeFFRuntimeErrorignore) AudioSegmentx str | dictreturnbytescCs@t|tr|dr|d}n t|d}n|}tt|S)zUConverts a base64 string or dictionary to a binary string that can be sent in a POST.dataname) isinstancedictget client_utilsZencode_url_or_file_to_base64base64 b64decodeextract_base64_data)rZ base64strrGD:\Projects\ConvertPro\env\Lib\site-packages\gradio/processing_utils.py to_binary#s   rstrcCs|dddS)z;Just extracts the base64 data from a general base64 string.,)rsplit)rrrrr/srencoding Image.ImagecCsbt|}ttt|}zttdrt|}W|SW|St y0t j d|ddY|Sw)Nexif_transposez0Failed to transpose image %s based on EXIF data.T)exc_info) rropenrrrhasattrrr& Exceptionlogwarning)r$Z image_encodedimgrrrdecode_base64_to_image9s    r.cCsTt}|j|dd|}Wdn1swYtt|d}d|S)Npngformatutf-8data:image/png;base64,)rZsavefiggetvaluerr b64encode)Zplt output_bytes bytes_data base64_strrrrencode_plot_to_base64Hs  r9cCs@t}|jD]\}}t|trt|tr|||q |SN)rZPngInfoinfoitemsrrZadd_text) pil_imagemetadatakeyvaluerrrget_pil_metadataPs  rAr/cCsFt}|j||t|d|WdS1swYdS)N)Zpnginfo)rsaverAr4)r=r1r6rrrencode_pil_to_bytesZs$rCcCs t|}tt|d}d|S)Nr2r3)rCrrr5)r=r7r8rrrencode_pil_to_base64`srDcCsht}tt|tjdd}||d|}Wdn1s#wYtt |d}d|S)NF) force_copyZPNGr2r3) rrZ fromarray_convertnpuint8rBr4rrr5)Z image_arrayr6r=r7r8rrrencode_array_to_base64fs  rIcentercCsl|dkrd}n |dkrd}ntt|}|ddur!|jd|d<|ddur.|jd|d<tj|||dS) a Resize and crop an image to fit the specified size. args: size: `(width, height)` tuple. Pass `None` for either width or height to only crop and resize the other. crop_type: can be 'top', 'middle' or 'bottom', depending on this value, the image will cropped getting the 'top/left', 'middle' or 'bottom/right' of the image to fit the size. raises: ValueError: if an invalid `crop_type` is provided. top)rrrJ)?rLrNr!)Z centering) ValueErrorlistsizerfit)r-rOZ crop_typerJresizerrrresize_and_cropos   rRdc Cszt|}Wn'ty.}zt|}|r#d|rdn|ddnd}t||d}~ww|dks7|dkrMt||d}t||d}|||}t| } |j dkr`| d |j } |j | fS) NzCannot load audio from file: `Zffprobez ` not found.zr Please install `ffmpeg` in your system to use non-WAV audio file formats and make sure `ffprobe` is in your PATH.rrSr!r") r from_fileFileNotFoundErrorris_file RuntimeErrorlenrGarrayZget_array_of_sampleschannelsZreshape frame_rate) filenameZcrop_minZcrop_maxaudioeisfilemsgZ audio_startZ audio_endrrrraudio_from_files*      rbwavcCsZ|dkrt|}t|||jjt|jdkrdn|jdd}|j||d}|dS)Nrcr!)r\Z sample_widthr[r0) convert_to_16_bit_wavrtobytesdtypeitemsizerYshapeZexportclose)Z sample_raterr]r1r^filerrr audio_to_files rkcCs d}|jtjtjtjfvr+t||j|t| }|d}| tj }|S|jtj krFt||j|d}| tj }|S|jtj krO |S|jtj krjt||j|d}| tj }|S|jtjkrt||j|dd}| tj }|Std|jd)NzCTrying to convert audio automatically from {} to 16-bit int format.iiiiz2Audio data cannot be converted automatically from z to 16-bit int format.)rfrGfloat64float32float16warningswarnr1absmaxastypeZint16Zint32Zuint16rHrM)rr,rrrrds<         rdFcstdtjdtjdtdtjdtjdtjdtjdi}dd}d ddd!fd d }t |}|j }|tj ur:t d nt |}|j } |j } |j } |j } |j} |j}t| t|rb|r`|}|S| d vrrt| j}t| j}| d vrt| j}t| j}| d kr|| || ddkS| d kr|| }| dkr|| || d9}|S| dkrA| dkr|| St|dkst|dkrtd||| tjtj}|s| dkrtj|||d}ntj|||d|d}|d8}tj||dtj||||dn7| dkrtj||d|d}tj|d||dntj|||dd|d}tj||dtj||||d|| S| dkru|| | tjtj}| dkr_tj|d||d}ntj|d|d}|d||9}t || S| dkr| dkr||d| d|d}|| S||d| d|S| dkr||d| dd|}t|j| }tj|d||j dd|S| |kr||d| dd|dS|d|d}||8}||d| d|dd}||7}|| S)"a Adapted from: https://github.com/scikit-image/scikit-image/blob/main/skimage/util/dtype.py#L510-L531 Convert an image to the requested data-type. Warnings are issued in case of precision loss, or when negative values are clipped during conversion to unsigned integer types (sign loss). Floating point values are expected to be normalized and will be clipped to the range [0.0, 1.0] or [-1.0, 1.0] when converting to unsigned or signed integers respectively. Numbers are not shifted to the negative side when converting from unsigned to signed integer types. Negative values will be clipped when converting to unsigned integers. Parameters ---------- image : ndarray Input image. dtype : dtype Target data-type. force_copy : bool, optional Force a copy of the data, irrespective of its current dtype. uniform : bool, optional Uniformly quantize the floating point range to the integer range. By default (uniform=False) floating point values are scaled and rounded to the nearest integers, which minimizes back and forth conversion errors. .. versionchanged :: 0.15 ``_convert`` no longer warns about possible precision or sign information loss. See discussions on these warnings at: https://github.com/scikit-image/scikit-image/issues/2602 https://github.com/scikit-image/scikit-image/issues/543#issuecomment-208202228 https://github.com/scikit-image/scikit-image/pull/3575 References ---------- .. [1] DirectX data conversion rules. https://msdn.microsoft.com/en-us/library/windows/desktop/dd607323%28v=vs.85%29.aspx .. [2] Data Conversions. In "OpenGL ES 2.0 Specification v2.0.25", pp 7-8. Khronos Group, 2010. .. [3] Proper treatment of pixels as integers. A.W. Paeth. In "Graphics Gems I", pp 249-256. Morgan Kaufmann, 1990. .. [4] Dirty Pixels. J. Blinn. In "Jim Blinn's corner: Dirty Pixels", pp 47-57. Morgan Kaufmann, 1998. )FT)r"r!cstfdd|DS)aReturn first of `dtypes` with itemsize greater than `itemsize` Parameters ---------- itemsize: int The data type object element size. Other Parameters ---------------- *dtypes: Any Object accepted by `np.dtype` to be converted to a data type object Returns ------- dtype: data type object First of `dtypes` with itemsize greater than `itemsize`. c3s$|] }t|jkr|VqdSr:)rGrfrg).0dtrgrr s"z4_convert.._dtype_itemsize..)next)rgZdtypesrrvr_dtype_itemsize sz!_convert.._dtype_itemsizer!cs0tfdd|fdD}tt|S)aReturn dtype of `kind` that can store a `bits` wide unsigned int Parameters: kind: str Data type kind. bits: int Desired number of bits. itemsize: int The data type object element size. Returns ------- dtype: data type object Data type of `kind` that can store a `bits` wide unsigned int c3s4|]}|dks|dkrdkr|VqdS)uNr)rtibitskindrrrw-s z0_convert.._dtype_bits..)rz)rxrGrfr)rr~rgsrr}r _dtype_bitssz_convert.._dtype_bitsTcs|jj}||kr|d|kr|||S||kr$|r"|S|S||krO|rEt|j||}tj|d||||jdd|S|d||}|S||dkr|rwt|j||}tj |d|dd|d||jd|S|j|||jj dd}|d|dd|d9}|S||d|}|rt|j||}tj |d|dd|d||jd|d||}|S|j|||jj dd}|d|dd|d9}|d||}|S) aScale an array of unsigned/positive integers from `n` to `m` bits. Numbers can be represented exactly only if `m` is a multiple of `n`. Parameters ---------- a : ndarray Input image array. n : int Number of bits currently used to encode the values in `a`. m : int Desired number of bits to encode the values in `out`. copy : bool, optional If True, allocates and returns new array. Otherwise, modifies `a` in place. Returns ------- out : array Output image array. Has the same kind as `a`. runsafeoutrfZcastingrr!)rrfFcopy) rfrrrrsrrGemptyrhZ floor_dividemultiplyrg)anmrrborrr_scale5s< ((z_convert.._scalerlZuirrfgg?z.Images of type float must be between -1 and 1.r{)rfrL)rrg@r|rzrrFrN)r!)T) boolrGZbool_Zbool8floatZfloat_rnrmrlZasarrayrfZfloatingtyperrgZ issubdtypeZ obj2sctyperZiinfominrrrsrMrZrintZclipflooraddviewrrhmaximum)imagerfrEuniformZ dtype_rangeryrZ dtypeobj_inZ dtypeobj_outZdtype_inZ dtype_outZkind_inZkind_outZ itemsize_inZ itemsize_outZimin_inZimax_inZimin_outZimax_outresultZcomputation_typeZ image_outrrrrFs,  :                     rFrcCstjrdStdduS)NFZffmpeg)r IS_WASMshutilwhichrrrrffmpeg_installedsrvideo_filepathc Csxz.t|j}td|did}|jtjtjd}t|d}|ddd}||fdvWSt t t fy;Yd Sw) zDetermines if a video is playable in the browser. A video is playable if it has a playable container and codec. .mp4 -> h264 .webm -> vp9 .ogg -> theora z?-show_format -show_streams -select_streams v -print_format jsonN)global_optionsinputs)stderrstdoutrstreamsZ codec_name)).mp4Zh264)z.oggZtheora)z.webmZvp9T) rsuffixlowerr run subprocessPIPEjsonloadsr IndexErrorKeyError)r containerZprobeoutputZ video_codecrrrvideo_is_playablesr video_pathc Csznz:tjdd*}t|d}t||jtt|jdit|didd}| Wdn1s5wYWnt yX}zt dt||}WYd}~nd}~wwWt |jt|SWt |jt|St |jw)zLConvert the video to mp4. If something goes wrong return the original video.F)deleterNz-y -loglevel quiet)rZoutputsrz2Error converting video to browser-playable format )tempfileNamedTemporaryFiler with_suffixrcopy2rr rrr printosremove)rZtmp_fileZ output_pathffr_rrrconvert_video_to_playable_mp4s0      r)rrrr)rrrr)r$rrr%)r/)rJ)rrS)rc)FF)rr)rrrr)rrrr)5 __future__rrrloggingrrrrroiorpathlibrnumpyrGZ gradio_clientrrZPILrrrZgradior rZffmpyr r r catch_warnings simplefilterZpydubr getLogger__name__r+rrr.r9rArCrDrIrRrbrkrdrFrrrrrrrsP             #