o پi*@sddlZddlZddlZddlZddlmZmZmZmZddl Z ddl m Z ddl m Z mZddlmZmZmZmZmZmZmZddlmZddlmZddlmZdd lmZmZm Z gd Z!ee"Z#d d d ddZ$dZ%dZ&de'de(e)e)fe(dBfddZ*dee'dee'de'fddZ+de'defddZ,dee e'fde'de'fd d!Z-d"e de'de'fd#d$Z.d%e'de'de'dBfd&d'Z/d(e'de'de'fd)d*Z0d+e'de'de'fd,d-Z1d.ede(e2e'effd/d0Z3d1eeedfdefd2d3Z4d4e5de'd5ede5fd6d7Z6djpgpngjpegwebpr9r7auto transparentr8)r()r5r6fmtr2r2r3choose_output_image_ext=s r> request_idc Kst}|dd}|ddu}|dd}|r.t|\}}|dur.|d||d|dd|D}|d d tjd |j||d |}|s]|dur]t ||j } | dur]| |_ |S) aEBuild SamplingParams from request parameters. Handles size parsing, output_quality resolution, and None filtering before delegating to SamplingParams.from_user_sampling_params_args. Callers pass only the parameters they have; None values are stripped automatically so that SamplingParams defaults apply. output_qualityNoutput_compressionr widthheightcSsi|] \}}|dur||qSNr2).0kvr2r2r3 `sz)build_sampling_params.. save_outputT) model_path server_argsr?r2) rpopgetr4 setdefaultitemsrfrom_user_sampling_params_argsrJadjust_output_quality data_typerA) r?kwargsrKr@has_explicit_compressionr r0r1sampling_paramsresolvedr2r2r3build_sampling_paramsHs.       rWimage target_pathcs.t||IdH}|durt||IdH}|SrD)_maybe_url_image_save_upload_to_path)rXrY input_pathr2r2r3save_image_to_pathus r]uploadcs`tjtj|dd|IdH}t|d}||Wd|S1s)wY|S)NTexist_okwb)osmakedirspathdirnamereadopenwrite)r^rYcontentfr2r2r3r[}s   r[img_urlcsXt|tsdS|drt||IdH}|S|dr(t||IdH}|Std)N)zhttp://zhttps://z data:imagezUnsupported image url format) isinstancestrr( startswith_save_url_image_to_path_save_base64_image_to_pathr,)rkrYr\r2r2r3rZs  rZ image_urlc stjtj|ddztjdd4IdH}|j|ddIdH}|tj|ds|j dd  }| d d }tj|\}}| }|d vrW|d krTdn|}n/| drxd|vsdd|vrgd}nd|vrnd}nd|vrud}nd}n|dkrd}nt d|||}t|d} | |jWdn1swY|WdIdHWS1IdHswYWdSty} z tdt| d} ~ ww)z0Download image from URL and save to target path.Tr_)follow_redirectsNg$@)timeoutr'z content-typer$?r>.bmp.gif.jpg.png.jpeg.webpryrwimage/r9r7r8rxr:rzzapplication/octet-streamz.URL does not point to an image. Content-Type: raz#Failed to download image from URL: )rbrcrdrehttpx AsyncClientrMraise_for_statussplitextheadersr(r*rnr,rgrhrir.rm) rqrYclientresponse content_typeurl_path_url_extextrjer2r2r3rosH   4$ro base64_datac s*d}d}t||}|st||d}|d}|s$t|d|d}|s2t|d|drG|d d }|d krFd }nd }|d |}tjtj |ddz$t |} t |d} | | Wd|WS1sxwY|WSty} z tdt| d} ~ ww)z1Decode base64 image data and save to target path.zSFailed to decode base64 image. Expected format: `data:[];base64,`zdata:(.*?)(;base64)?,(.*)r'r&z (missing ;base64 marker)z (empty data payload)r{/r9r7.Tr_raNzFailed to decode base64 image: )rematchr,grouprnr*r(rbrcrdrebase64 b64decodergrhr.rm) rrY _B64_FMT_HINTpatternr media_type is_base64datar image_datarjrr2r2r3rpsD         rpscheduler_clientc st}ttjK|gIdH}|jdur+|jdur+|jp#d}t d||jr2|j}nt |jt |jj j jfdd|j|jjd}Wdn1sYwYt|}ttd||jr}|jdkr}td|jd d ||fS) Nz Unknown errorz;Model generation returned no output. Error from scheduler: cst|SrD)rmoutput_file_path)idxbatch num_outputsr2r3sz*process_generation_batch..)audioaudio_sample_raterAr'rzPeak memory usage: z.2fz MB)time perf_counterrloggerpromptforwardoutputoutput_file_pathserror RuntimeErrorr+rrRfpsrIrrrArpeak_memory_mbinfo)rrtotal_start_timeresult error_msgsave_file_path_list total_timer2rr3process_generation_batchs:     rinputscGs:g}|D]}|durt|tr||q||q|S)a4 Merge multiple image input sources into a single list. This function handles both single items and lists of items, merging them into a single flattened list. Useful for processing images, URLs, or other multimedia inputs that can come as either single items or lists. Args: *inputs: Variable number of inputs, each can be None, single item, or list Returns: List: Flattened list of all non-None inputs Example: >>> merge_image_input_list(["img1", "img2"], "img3", None) ["img1", "img2", "img3"] N)rllistextendappend)rr input_itemr2r2r3merge_image_input_lists   rrrcCsD|jr |jdkr |j|d<|jr|jjdkr|jj|d<||d<|S)Nrrinference_time_sid)rmetricstotal_duration_s)rr?rr2r2r3add_common_data_to_response-s   rr@rRcCs&|dkr |tjkr dSdSt|dS)Ndefault2K)rVIDEOOUTPUT_QUALITY_MAPPERrM)r@rRr2r2r3rQ;s rQrD)8rrbrrtypingrrrrr|fastapir4sglang.multimodal_gen.configs.sample.sampling_paramsrr/sglang.multimodal_gen.runtime.entrypoints.utilsr r r r r rr;sglang.multimodal_gen.runtime.pipelines_core.schedule_batchr.sglang.multimodal_gen.runtime.scheduler_clientr)sglang.multimodal_gen.runtime.server_argsr1sglang.multimodal_gen.runtime.utils.logging_utilsrrr__all____name__rr DEFAULT_FPSDEFAULT_VIDEO_SECONDSrmtupler-r4r>rWr]r[rZrorprrrdictrrQr2r2r2r3s` $   "   -.( &