z
z
z)joinsplit)text classnamecontentr r r r s r c ! C s dd l }g }g }G dd d}t| }||} tjj}
tjj} tjj} d}
d}|dkr~tjjr~||d kr~d}|| g}|}
t j
tjjdd t t j
tjjd d
d d d
4}| dk}||}|r|g d t||
D ]\}}t|}|| jk }|rdn|| j }|d | _tjj||
d | j| | j| | | j| || | d
\}}t j ||
}| | | | |r| t j | | | q||d |d |d |d |d |d |d |d |d g W d n1 s0 Y |r|r| j|d n| jd }tj | || jd |d}| tjj pDd} t j
|
| d} ddl!m"} | | d j} t#t$|D ]L}t || d d $} | %|| | & W d n1 s0 Y qW d n1 s0 Y |'d| t(j)j*|dd t+d |d fS )!Nr c @ s e Zd ZdddZdS )z save_files..MyObjectNc S s* |d ur&| D ]\}}t| || qd S r )itemssetattr)selfdkeyvaluer r r __init__- s z%save_files..MyObject.__init__)N)__name__
__module__ __qualname__r& r r r r MyObject, s r* Findex_of_first_imageT)exist_okzlog.csvautf8 )encodingnewline) promptseedwidthheightsamplercfgsstepsfilenamenegative_prompt )r4 r3 extensioninfogridp save_to_dirsr3 r4 r5 r6 sampler_name cfg_scaler9 r; z,[datetime]_[[model_name]]_[seed]-[seed_last]z.zip)ZipFilewrb)mode)r% visiblezSaved: ),csvr r r opts outdir_saveuse_save_to_dirs_for_uisamples_formatsave_selected_onlyosmakedirsopenpathr tellwriterwriterow enumerater r, batch_indexmodulesimages
save_image all_seeds all_promptsr relpathappendbasenameFilenameGeneratorapplygrid_zip_filename_patternzipfilerD ranger writestrreadinsertr
Filer r )!Zjs_datarY Z do_make_zipindexrI filenamesZfullfnsr* datar@ rR rA r= start_indexZonly_onefileat_startrT Z image_indexfiledataimageZis_gridifullfn
txt_fullfnr: Z zip_fileseednamegen zip_filenameZ zip_filepathrD zip_filefr r r
save_files& s\
"
6
` X rx c s dd t jd| ddp t j| dd4 t jdd | d
d d tjjpRd d
}W d n1 sl0 Y d }t t jd| dd tt | dtj
j dd}| dkrtdd| dtjj dd}tdd| d tjj d d}td | d d dtd!| d"d#dtd$| d%d&dd'}W d n1 sT0 Y |j
fd(d)g g d* | dkrt jd d+d d d d,| d-}t t jd.| d/d} t jd0| d1d}
t jd d2| d3}| d4ks| d5kr*t jd | d6d3} | j
td7|| | g| | gd d8 |j
ttd9||| | g||
gd d8 |j
ttd:||| | g||
gd; W d n1 s0 Y n8t jd<| d} t jd.| d/d} t jd0| d}
g }
| d4krtjjj}
n| d5krtjjj}
| D ]2\}}t t j ||| d4kr&d4nd ||
d= q|| dkrH|n| | |
fW d W d S 1 sv0 Y W d n1 s0 Y d S )>Nc S s t j| s td| d d S t j| sFtd| dtjd d S tjj st j
| }t dkrrt
| nHt dkrtd|g n,d t jv rtd
|g ntd |g d S ) NzFolder "zH" does not exist. After you create an image, the folder will be created.z
WARNING
An open_folder request was made with an argument that is not a folder.
This could be an error or a malicious attempt to run code on your computer.
Requested path was: r )rm WindowsDarwinrQ zmicrosoft-standard-WSL2zwsl-openzxdg-open)rO rR existsprintisdirsysstderrr cmd_optshide_ui_dir_confignormpathplatformsystem startfilespPopenunamerelease)rw rR r r r open_folderm s& z(create_output_panel..open_folderpanel_results)variantelem_idZ_gallery_container)r OutputFZ_gallery T)label
show_labelr columnspreviewr6 Zimage_buttons_z
image-buttons)r elem_classesZ _open_folderz Open images output directory.)r rH tooltipextrasu 💾Zsave_z)Save the image to a dedicated directory (z).)r r u 🗃️Z save_zip_z7Save zip archive with images to a dedicated directory ()u 🖼️Z_send_to_img2imgz4Send image and generation parameters to img2img tab.u 🎨️Z_send_to_inpaintz r z%create_output_panel..fninputsoutputsmultipleZdownload_files_)
file_count interactiver rH r Z
html_info_infotextZ html_log_zhtml-logZgeneration_info_)rH r txt2imgr Z_generation_info_buttonz [x, y, false, selected_gallery_index()]z6(x, y, z, w) => [x, y, true, selected_gallery_index()])r r r r Z html_info_x_) paste_buttontabnamesource_tabnamesource_image_componentpaste_field_names) r
ColumnGroupGalleryr rJ gallery_heightRowr
folder_symbolr r rK clickrh HTMLTextboxButtonr r wrap_gradio_callrx rX scriptsscripts_txt2imgr scripts_img2imgr parameters_copypaste register_paste_params_button ParamBinding)r r Zresult_galleryr Zopen_folder_buttonsaveZsave_zipbuttonsZdownload_filesr html_logZgeneration_info_buttonZ html_info_xr Z
paste_tabnamer r r r create_output_panelk s B
&
(
r c sz t | tr| n| g d } D ] }t|dd }|d ur q:q fdd}tt||r^| dndd}|j|g d |S )Nr c sr t r n D ] \} }D ]}t|| | q,q tdkr^ fddD S tjf i pli S )Nr< c s g | ]}t jf i pi qS r r
r )r _argsr r
r z:create_refresh_button..refresh..)callabler r! r r
r )kvcompZrefresh_componentsrefresh_methodrefreshed_argsr r refresh s z&create_refresh_button..refreshz : refreshZRefresh)r% r r r )
isinstancelistgetattrr refresh_symbolr )Zrefresh_componentr r r r r r Zrefresh_buttonr r r create_refresh_button s
r ) button_closec C sF d|_ | jdd g |gdjdd|j d d |rB|jdd d dS )
zSets up the UI so that the dialog (gr.Box) is invisible, and is only shown when buttons_show is clicked, in a fullscreen modal window.Fc S s t jddS )NT)rH r r r r r r r z setup_dialog..r Nzfunction(){ popupId('z'); })r r Z
closePopup)rH r thenr )Z button_showZdialogr r r r setup_dialog s r )N)r r rO r r~ gradior
subprocessr rX r r 'modules.generation_parameters_copypaster modules.imagesmodules.ui_componentsr generation_parameters_copypaster r r r r rx r r r r r r r s&
Ex