a Ad @s<ddlZddlmZmZmZmZmZddlm Z ddZ dS)N)scriptsshared ui_commonpostprocessing call_queuec Cstjdd}tjdddtjdd`tjddtjd d d d $}tjd ddddd}Wdn1sx0Ytjdddd  }tjdddd}Wdn1s0Ytjdddd `}tjfddit j ddd}tjfdd it j d!d"d}tj d#dd$d%}Wdn1s:0YWdn1sZ0Ytj d&d'd(d)} t j} Wdn1s0Yt(td*t jj\} } } }Wdn1s0YWdn1s0Y|jd+d,g|gd-|jd.d,g|gd-|jd/d,g|gd-| jtjtjdd0gd1||||||g| | | | gd-td*|d|jt jjggd-dS)2Nr)valueFcompact) equal_heightvariant)r Z mode_extras)elem_idz Single ImageZ single_imageZextras_single_tab)idr SourceuploadTpil extras_image)labelsource interactivetyper z Batch ProcessZ batch_processZextras_batch_process_tabZextras_image_batch)rrr zBatch from DirectoryZbatch_from_directoryZextras_batch_directory_tabrzInput directoryzzcreate_ui..)fninputsoutputscSsdS)NrrrrrrrcSsdS)Nrrrrrr r) extra_outputs)grStateRowColumnTabsTabItemImageFilesTextboxr hide_dirsCheckboxButtonrscripts_postprocsetup_uircreate_output_paneloptsoutdir_extras_samplesselectclickrwrap_gradio_gpu_callrZrun_postprocessingparameters_copypasteadd_paste_fieldschange image_changed)Z tab_indexZ tab_singler tab_batchZ image_batchZ tab_batch_dirrrZshow_extras_resultssubmitZ script_inputsZ result_images html_info_x html_infohtml_logrrr create_uisP 2.  P* X rC) gradior&modulesrrrrr'modules.generation_parameters_copypastegeneration_parameters_copypaster:rCrrrrs