2022-09-03 12:08:45 +03:00
import html
import json
2022-09-10 11:10:00 +03:00
import math
2022-09-03 12:08:45 +03:00
import mimetypes
import os
2022-10-22 14:07:00 +03:00
import platform
2022-09-05 23:08:06 +03:00
import random
2022-10-22 14:07:00 +03:00
import subprocess as sp
2022-09-03 12:08:45 +03:00
import sys
2022-10-15 13:11:28 +03:00
import tempfile
2022-09-03 12:08:45 +03:00
import time
import traceback
2022-10-14 20:04:47 +03:00
from functools import partial , reduce
2022-09-03 12:08:45 +03:00
2022-10-22 14:07:00 +03:00
import gradio as gr
import gradio . routes
import gradio . utils
2022-09-06 19:33:51 +03:00
import numpy as np
2022-09-28 17:05:23 +03:00
from PIL import Image , PngImagePlugin
2022-11-28 09:00:10 +03:00
from modules . call_queue import wrap_gradio_gpu_call , wrap_queued_call , wrap_gradio_call
2022-09-03 12:08:45 +03:00
2023-01-03 09:04:29 +03:00
from modules import sd_hijack , sd_models , localization , script_callbacks , ui_extensions , deepbooru
2023-01-07 09:56:37 +03:00
from modules . ui_components import FormRow , FormGroup , ToolButton , FormHTML
2022-09-03 12:08:45 +03:00
from modules . paths import script_path
2022-10-16 22:06:21 +03:00
2022-10-16 20:08:23 +03:00
from modules . shared import opts , cmd_opts , restricted_opts
2022-10-14 11:56:41 +03:00
2022-10-22 14:07:00 +03:00
import modules . codeformer_model
2022-10-27 08:36:11 +03:00
import modules . generation_parameters_copypaste as parameters_copypaste
2022-10-22 14:07:00 +03:00
import modules . gfpgan_model
import modules . hypernetworks . ui
2022-09-03 17:21:15 +03:00
import modules . scripts
2022-10-22 14:07:00 +03:00
import modules . shared as shared
2022-09-09 23:16:02 +03:00
import modules . styles
2022-10-22 14:07:00 +03:00
import modules . textual_inversion . ui
2022-10-05 23:16:27 +03:00
from modules import prompt_parser
2022-10-04 19:19:50 +03:00
from modules . images import save_image
2022-10-22 14:07:00 +03:00
from modules . sd_hijack import model_hijack
from modules . sd_samplers import samplers , samplers_for_img2img
2023-01-09 23:35:40 +03:00
from modules . textual_inversion import textual_inversion
2022-10-11 15:51:22 +03:00
import modules . hypernetworks . ui
2022-10-27 08:36:11 +03:00
from modules . generation_parameters_copypaste import image_from_url_text
2022-09-03 12:08:45 +03:00
2022-10-08 22:12:24 +03:00
# this is a fix for Windows users. Without it, javascript files will be served with text/html content-type and the browser will not show any UI
2022-09-03 12:08:45 +03:00
mimetypes . init ( )
mimetypes . add_type ( ' application/javascript ' , ' .js ' )
2022-09-05 19:37:11 +03:00
if not cmd_opts . share and not cmd_opts . listen :
2022-09-03 12:08:45 +03:00
# fix gradio phoning home
gradio . utils . version_check = lambda : None
gradio . utils . get_local_ip_address = lambda : ' 127.0.0.1 '
2022-12-14 21:59:33 +03:00
if cmd_opts . ngrok is not None :
2022-10-11 12:40:27 +03:00
import modules . ngrok as ngrok
print ( ' ngrok authtoken detected, trying to connect... ' )
2022-12-14 21:59:33 +03:00
ngrok . connect (
cmd_opts . ngrok ,
cmd_opts . port if cmd_opts . port is not None else 7860 ,
cmd_opts . ngrok_region
)
2022-10-11 12:40:27 +03:00
2022-09-03 12:08:45 +03:00
def gr_show ( visible = True ) :
return { " visible " : visible , " __type__ " : " update " }
sample_img2img = " assets/stable-samples/img2img/sketch-mountains-input.jpg "
sample_img2img = sample_img2img if os . path . exists ( sample_img2img ) else None
css_hide_progressbar = """
. wrap . m - 12 svg { display : none ! important ; }
2022-09-27 10:44:00 +03:00
. wrap . m - 12 : : before { content : " Loading... " }
2022-11-16 07:08:03 +03:00
. wrap . z - 20 svg { display : none ! important ; }
. wrap . z - 20 : : before { content : " Loading... " }
2022-09-03 12:08:45 +03:00
. progress - bar { display : none ! important ; }
. meta - text { display : none ! important ; }
2022-11-16 07:08:03 +03:00
. meta - text - center { display : none ! important ; }
2022-09-03 12:08:45 +03:00
"""
2022-09-16 22:20:56 +03:00
# Using constants for these since the variation selector isn't visible.
# Important that they exactly match script.js for tooltip to work.
random_symbol = ' \U0001f3b2 \ufe0f ' # 🎲️
reuse_symbol = ' \u267b \ufe0f ' # ♻️
2022-09-23 22:49:21 +03:00
paste_symbol = ' \u2199 \ufe0f ' # ↙
2022-10-02 22:08:23 +03:00
folder_symbol = ' \U0001f4c2 ' # 📂
2022-10-13 19:22:41 +03:00
refresh_symbol = ' \U0001f504 ' # 🔄
2022-10-15 14:22:30 +03:00
save_style_symbol = ' \U0001f4be ' # 💾
apply_style_symbol = ' \U0001f4cb ' # 📋
2022-10-21 23:32:26 +03:00
clear_prompt_symbol = ' \U0001F5D1 ' # 🗑️
2022-10-13 19:22:41 +03:00
2022-09-16 22:20:56 +03:00
2022-09-03 12:08:45 +03:00
def plaintext_to_html ( text ) :
2022-09-10 14:53:38 +03:00
text = " <p> " + " <br> \n " . join ( [ f " { html . escape ( x ) } " for x in text . split ( ' \n ' ) ] ) + " </p> "
2022-09-03 12:08:45 +03:00
return text
def send_gradio_gallery_to_image ( x ) :
if len ( x ) == 0 :
return None
return image_from_url_text ( x [ 0 ] )
2022-10-08 08:09:29 +03:00
def save_files ( js_data , images , do_make_zip , index ) :
2022-10-10 04:26:52 +03:00
import csv
2022-09-03 12:08:45 +03:00
filenames = [ ]
2022-10-08 08:09:29 +03:00
fullfns = [ ]
2022-09-03 12:08:45 +03:00
2022-10-08 22:12:24 +03:00
#quick dictionary to class object conversion. Its necessary due apply_filename_pattern requiring it
2022-09-28 11:53:40 +03:00
class MyObject :
def __init__ ( self , d = None ) :
if d is not None :
for key , value in d . items ( ) :
setattr ( self , key , value )
2022-09-03 12:08:45 +03:00
data = json . loads ( js_data )
2022-09-29 13:19:13 +03:00
2022-09-28 11:53:40 +03:00
p = MyObject ( data )
path = opts . outdir_save
2022-10-02 19:15:31 +03:00
save_to_dirs = opts . use_save_to_dirs_for_ui
2022-10-04 19:19:50 +03:00
extension : str = opts . samples_format
start_index = 0
2022-09-29 13:19:13 +03:00
2022-09-28 17:05:23 +03:00
if index > - 1 and opts . save_selected_only and ( index > = data [ " index_of_first_image " ] ) : # ensures we are looking at a specific non-grid picture, and we have save_selected_only
2022-09-29 13:19:13 +03:00
2022-09-14 05:11:46 +03:00
images = [ images [ index ] ]
2022-10-04 19:19:50 +03:00
start_index = index
2022-09-03 12:08:45 +03:00
2022-10-12 07:54:24 +03:00
os . makedirs ( opts . outdir_save , exist_ok = True )
2022-09-03 21:33:47 +03:00
with open ( os . path . join ( opts . outdir_save , " log.csv " ) , " a " , encoding = " utf8 " , newline = ' ' ) as file :
2022-09-03 12:08:45 +03:00
at_start = file . tell ( ) == 0
writer = csv . writer ( file )
if at_start :
2022-09-12 19:57:31 +03:00
writer . writerow ( [ " prompt " , " seed " , " width " , " height " , " sampler " , " cfgs " , " steps " , " filename " , " negative_prompt " ] )
2022-09-03 12:08:45 +03:00
2022-10-04 19:19:50 +03:00
for image_index , filedata in enumerate ( images , start_index ) :
2022-10-15 17:25:35 +03:00
image = image_from_url_text ( filedata )
2022-09-03 12:08:45 +03:00
2022-10-04 19:19:50 +03:00
is_grid = image_index < p . index_of_first_image
i = 0 if is_grid else ( image_index - p . index_of_first_image )
2022-10-13 09:09:30 +03:00
fullfn , txt_fullfn = save_image ( image , path , " " , seed = p . all_seeds [ i ] , prompt = p . all_prompts [ i ] , extension = extension , info = p . infotexts [ image_index ] , grid = is_grid , p = p , save_to_dirs = save_to_dirs )
2022-10-04 19:19:50 +03:00
filename = os . path . relpath ( fullfn , path )
2022-09-03 12:08:45 +03:00
filenames . append ( filename )
2022-10-08 08:09:29 +03:00
fullfns . append ( fullfn )
2022-10-09 08:01:10 +03:00
if txt_fullfn :
filenames . append ( os . path . basename ( txt_fullfn ) )
fullfns . append ( txt_fullfn )
2022-09-03 12:08:45 +03:00
2022-11-19 12:01:51 +03:00
writer . writerow ( [ data [ " prompt " ] , data [ " seed " ] , data [ " width " ] , data [ " height " ] , data [ " sampler_name " ] , data [ " cfg_scale " ] , data [ " steps " ] , filenames [ 0 ] , data [ " negative_prompt " ] ] )
2022-09-03 12:08:45 +03:00
2022-10-08 08:09:29 +03:00
# Make Zip
if do_make_zip :
zip_filepath = os . path . join ( path , " images.zip " )
from zipfile import ZipFile
with ZipFile ( zip_filepath , " w " ) as zip_file :
for i in range ( len ( fullfns ) ) :
with open ( fullfns [ i ] , mode = " rb " ) as f :
zip_file . writestr ( filenames [ i ] , f . read ( ) )
fullfns . insert ( 0 , zip_filepath )
2022-12-31 23:40:55 +03:00
return gr . File . update ( value = fullfns , visible = True ) , plaintext_to_html ( f " Saved: { filenames [ 0 ] } " )
2022-10-15 13:11:28 +03:00
2022-09-04 13:52:01 +03:00
def visit ( x , func , path = " " ) :
if hasattr ( x , ' children ' ) :
for c in x . children :
visit ( c , func , path )
elif x . label is not None :
func ( path + " / " + str ( x . label ) , x )
2022-09-03 12:08:45 +03:00
2022-09-11 17:35:12 +03:00
def add_style ( name : str , prompt : str , negative_prompt : str ) :
if name is None :
2022-10-15 14:22:30 +03:00
return [ gr_show ( ) for x in range ( 4 ) ]
2022-09-09 23:16:02 +03:00
2022-09-11 17:35:12 +03:00
style = modules . styles . PromptStyle ( name , prompt , negative_prompt )
2022-09-14 17:56:21 +03:00
shared . prompt_styles . styles [ style . name ] = style
2022-09-11 17:35:12 +03:00
# Save all loaded prompt styles: this allows us to update the storage format in the future more easily, because we
# reserialize all styles every time we save them
2022-09-14 17:56:21 +03:00
shared . prompt_styles . save_styles ( shared . styles_filename )
2022-09-09 23:16:02 +03:00
2022-09-23 20:46:02 +03:00
return [ gr . Dropdown . update ( visible = True , choices = list ( shared . prompt_styles . styles ) ) for _ in range ( 4 ) ]
2022-09-14 17:56:21 +03:00
2023-01-07 09:56:37 +03:00
def calc_resolution_hires ( enable , width , height , hr_scale , hr_resize_x , hr_resize_y ) :
from modules import processing , devices
if not enable :
return " "
p = processing . StableDiffusionProcessingTxt2Img ( width = width , height = height , enable_hr = True , hr_scale = hr_scale , hr_resize_x = hr_resize_x , hr_resize_y = hr_resize_y )
with devices . autocast ( ) :
p . init ( [ " " ] , [ 0 ] , [ 0 ] )
2023-01-09 14:57:47 +03:00
return f " resize: from <span class= ' resolution ' > { p . width } x { p . height } </span> to <span class= ' resolution ' > { p . hr_resize_x or p . hr_upscale_to_x } x { p . hr_resize_y or p . hr_upscale_to_y } </span> "
2023-01-07 09:56:37 +03:00
2022-09-14 17:56:21 +03:00
def apply_styles ( prompt , prompt_neg , style1_name , style2_name ) :
prompt = shared . prompt_styles . apply_styles_to_prompt ( prompt , [ style1_name , style2_name ] )
prompt_neg = shared . prompt_styles . apply_negative_styles_to_prompt ( prompt_neg , [ style1_name , style2_name ] )
return [ gr . Textbox . update ( value = prompt ) , gr . Textbox . update ( value = prompt_neg ) , gr . Dropdown . update ( value = " None " ) , gr . Dropdown . update ( value = " None " ) ]
2022-09-09 23:16:02 +03:00
2022-09-11 18:48:36 +03:00
def interrogate ( image ) :
2022-12-25 07:23:12 +03:00
prompt = shared . interrogator . interrogate ( image . convert ( " RGB " ) )
2022-09-11 18:48:36 +03:00
return gr_show ( True ) if prompt is None else prompt
2022-09-14 17:56:21 +03:00
2022-10-05 21:50:10 +03:00
def interrogate_deepbooru ( image ) :
2022-11-20 16:39:20 +03:00
prompt = deepbooru . model . tag ( image )
2022-10-05 21:50:10 +03:00
return gr_show ( True ) if prompt is None else prompt
2023-01-01 16:51:12 +03:00
def create_seed_inputs ( target_interface ) :
2023-01-03 09:04:29 +03:00
with FormRow ( elem_id = target_interface + ' _seed_row ' ) :
seed = ( gr . Textbox if cmd_opts . use_textbox_seed else gr . Number ) ( label = ' Seed ' , value = - 1 , elem_id = target_interface + ' _seed ' )
seed . style ( container = False )
random_seed = gr . Button ( random_symbol , elem_id = target_interface + ' _random_seed ' )
reuse_seed = gr . Button ( reuse_symbol , elem_id = target_interface + ' _reuse_seed ' )
2022-09-16 22:20:56 +03:00
2023-01-03 09:04:29 +03:00
with gr . Group ( elem_id = target_interface + ' _subseed_show_box ' ) :
2023-01-01 16:51:12 +03:00
seed_checkbox = gr . Checkbox ( label = ' Extra ' , elem_id = target_interface + ' _subseed_show ' , value = False )
2022-09-16 22:20:56 +03:00
# Components to show/hide based on the 'Extra' checkbox
seed_extras = [ ]
2023-01-03 09:04:29 +03:00
with FormRow ( visible = False , elem_id = target_interface + ' _subseed_row ' ) as seed_extra_row_1 :
2022-09-16 22:20:56 +03:00
seed_extras . append ( seed_extra_row_1 )
2023-01-03 09:04:29 +03:00
subseed = gr . Number ( label = ' Variation seed ' , value = - 1 , elem_id = target_interface + ' _subseed ' )
subseed . style ( container = False )
random_subseed = gr . Button ( random_symbol , elem_id = target_interface + ' _random_subseed ' )
reuse_subseed = gr . Button ( reuse_symbol , elem_id = target_interface + ' _reuse_subseed ' )
2023-01-01 16:51:12 +03:00
subseed_strength = gr . Slider ( label = ' Variation strength ' , value = 0.0 , minimum = 0 , maximum = 1 , step = 0.01 , elem_id = target_interface + ' _subseed_strength ' )
2022-09-16 22:20:56 +03:00
2023-01-03 09:04:29 +03:00
with FormRow ( visible = False ) as seed_extra_row_2 :
2022-09-16 22:20:56 +03:00
seed_extras . append ( seed_extra_row_2 )
2023-01-01 16:51:12 +03:00
seed_resize_from_w = gr . Slider ( minimum = 0 , maximum = 2048 , step = 8 , label = " Resize seed from width " , value = 0 , elem_id = target_interface + ' _seed_resize_from_w ' )
seed_resize_from_h = gr . Slider ( minimum = 0 , maximum = 2048 , step = 8 , label = " Resize seed from height " , value = 0 , elem_id = target_interface + ' _seed_resize_from_h ' )
2022-09-16 22:20:56 +03:00
random_seed . click ( fn = lambda : - 1 , show_progress = False , inputs = [ ] , outputs = [ seed ] )
random_subseed . click ( fn = lambda : - 1 , show_progress = False , inputs = [ ] , outputs = [ subseed ] )
def change_visibility ( show ) :
return { comp : gr_show ( show ) for comp in seed_extras }
seed_checkbox . change ( change_visibility , show_progress = False , inputs = [ seed_checkbox ] , outputs = seed_extras )
2022-09-21 13:34:10 +03:00
return seed , reuse_seed , subseed , reuse_subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox
2022-09-16 22:20:56 +03:00
2022-10-21 22:24:14 +03:00
2022-11-01 22:33:55 +03:00
def connect_clear_prompt ( button ) :
2022-10-21 23:32:26 +03:00
""" Given clear button, prompt, and token_counter objects, setup clear prompt button click event """
2022-10-21 22:24:14 +03:00
button . click (
2022-10-21 23:13:12 +03:00
_js = " clear_prompt " ,
2022-11-01 22:03:56 +03:00
fn = None ,
2022-11-01 22:33:55 +03:00
inputs = [ ] ,
outputs = [ ] ,
2022-10-21 22:24:14 +03:00
)
2022-10-20 09:08:24 +03:00
2022-09-19 09:02:10 +03:00
def connect_reuse_seed ( seed : gr . Number , reuse_seed : gr . Button , generation_info : gr . Textbox , dummy_component , is_subseed ) :
""" Connects a ' reuse (sub)seed ' button ' s click event so that it copies last used
( sub ) seed value from generation info the to the seed field . If copying subseed and subseed strength
2022-09-16 22:20:56 +03:00
was 0 , i . e . no variation seed was used , it copies the normal seed value instead . """
2022-09-19 09:02:10 +03:00
def copy_seed ( gen_info_string : str , index ) :
res = - 1
2022-09-16 22:20:56 +03:00
try :
gen_info = json . loads ( gen_info_string )
2022-09-19 09:02:10 +03:00
index - = gen_info . get ( ' index_of_first_image ' , 0 )
if is_subseed and gen_info . get ( ' subseed_strength ' , 0 ) > 0 :
all_subseeds = gen_info . get ( ' all_subseeds ' , [ - 1 ] )
res = all_subseeds [ index if 0 < = index < len ( all_subseeds ) else 0 ]
2022-09-16 22:20:56 +03:00
else :
2022-09-19 09:02:10 +03:00
all_seeds = gen_info . get ( ' all_seeds ' , [ - 1 ] )
res = all_seeds [ index if 0 < = index < len ( all_seeds ) else 0 ]
2022-09-16 22:20:56 +03:00
except json . decoder . JSONDecodeError as e :
if gen_info_string != ' ' :
print ( " Error parsing JSON generation info: " , file = sys . stderr )
print ( gen_info_string , file = sys . stderr )
2022-09-19 09:02:10 +03:00
return [ res , gr_show ( False ) ]
2022-09-16 22:20:56 +03:00
reuse_seed . click (
fn = copy_seed ,
2022-09-19 09:02:10 +03:00
_js = " (x, y) => [x, selected_gallery_index()] " ,
2022-09-16 22:20:56 +03:00
show_progress = False ,
2022-09-19 09:02:10 +03:00
inputs = [ generation_info , dummy_component ] ,
outputs = [ seed , dummy_component ]
2022-09-16 22:20:56 +03:00
)
2022-10-04 14:35:12 +03:00
2022-09-29 22:47:06 +03:00
def update_token_counter ( text , steps ) :
2022-10-04 14:35:12 +03:00
try :
2022-10-05 23:16:27 +03:00
_ , prompt_flat_list , _ = prompt_parser . get_multicond_prompt_list ( [ text ] )
prompt_schedules = prompt_parser . get_learned_conditioning_prompt_schedules ( prompt_flat_list , steps )
2022-10-04 14:35:12 +03:00
except Exception :
# a parsing error can happen here during typing, and we don't want to bother the user with
# messages related to it in console
prompt_schedules = [ [ [ steps , text ] ] ]
2022-09-29 22:47:06 +03:00
flat_prompts = reduce ( lambda list1 , list2 : list1 + list2 , prompt_schedules )
2022-10-04 14:35:12 +03:00
prompts = [ prompt_text for step , prompt_text in flat_prompts ]
2023-01-07 01:45:28 +03:00
token_count , max_length = max ( [ model_hijack . get_prompt_lengths ( prompt ) for prompt in prompts ] , key = lambda args : args [ 0 ] )
2022-09-28 02:29:53 +03:00
style_class = ' class= " red " ' if ( token_count > max_length ) else " "
return f " <span { style_class } > { token_count } / { max_length } </span> "
2022-09-19 16:42:56 +03:00
2022-10-04 14:35:12 +03:00
2022-09-14 17:56:21 +03:00
def create_toprow ( is_img2img ) :
2022-09-23 20:46:02 +03:00
id_part = " img2img " if is_img2img else " txt2img "
2022-09-14 17:56:21 +03:00
with gr . Row ( elem_id = " toprow " ) :
2022-10-15 14:22:30 +03:00
with gr . Column ( scale = 6 ) :
2022-09-14 17:56:21 +03:00
with gr . Row ( ) :
2022-09-23 20:54:17 +03:00
with gr . Column ( scale = 80 ) :
2022-09-14 17:56:21 +03:00
with gr . Row ( ) :
2022-10-19 17:30:33 +03:00
prompt = gr . Textbox ( label = " Prompt " , elem_id = f " { id_part } _prompt " , show_label = False , lines = 2 ,
2022-10-15 06:48:13 +03:00
placeholder = " Prompt (press Ctrl+Enter or Alt+Enter to generate) "
)
2022-09-14 17:56:21 +03:00
with gr . Row ( ) :
2022-10-15 14:22:30 +03:00
with gr . Column ( scale = 80 ) :
2022-10-11 10:08:45 +03:00
with gr . Row ( ) :
2022-10-19 17:30:33 +03:00
negative_prompt = gr . Textbox ( label = " Negative prompt " , elem_id = f " { id_part } _neg_prompt " , show_label = False , lines = 2 ,
2022-10-15 06:48:13 +03:00
placeholder = " Negative prompt (press Ctrl+Enter or Alt+Enter to generate) "
)
2022-10-15 14:22:30 +03:00
with gr . Column ( scale = 1 , elem_id = " roll_col " ) :
paste = gr . Button ( value = paste_symbol , elem_id = " paste " )
save_style = gr . Button ( value = save_style_symbol , elem_id = " style_create " )
prompt_style_apply = gr . Button ( value = apply_style_symbol , elem_id = " style_apply " )
2022-12-10 13:46:18 +03:00
clear_prompt_button = gr . Button ( value = clear_prompt_symbol , elem_id = f " { id_part } _clear_prompt " )
2022-10-15 14:22:30 +03:00
token_counter = gr . HTML ( value = " <span></span> " , elem_id = f " { id_part } _token_counter " )
token_button = gr . Button ( visible = False , elem_id = f " { id_part } _token_button " )
2022-12-10 13:46:18 +03:00
clear_prompt_button . click (
fn = lambda * x : x ,
_js = " confirm_clear_prompt " ,
inputs = [ prompt , negative_prompt ] ,
outputs = [ prompt , negative_prompt ] ,
)
2022-10-15 14:22:30 +03:00
button_interrogate = None
button_deepbooru = None
if is_img2img :
with gr . Column ( scale = 1 , elem_id = " interrogate_col " ) :
button_interrogate = gr . Button ( ' Interrogate \n CLIP ' , elem_id = " interrogate " )
2022-11-26 16:10:46 +03:00
button_deepbooru = gr . Button ( ' Interrogate \n DeepBooru ' , elem_id = " deepbooru " )
2022-09-14 17:56:21 +03:00
with gr . Column ( scale = 1 ) :
with gr . Row ( ) :
2022-10-05 06:56:30 +03:00
skip = gr . Button ( ' Skip ' , elem_id = f " { id_part } _skip " )
2022-09-23 20:46:02 +03:00
interrupt = gr . Button ( ' Interrupt ' , elem_id = f " { id_part } _interrupt " )
2022-09-30 23:31:00 +03:00
submit = gr . Button ( ' Generate ' , elem_id = f " { id_part } _generate " , variant = ' primary ' )
2022-09-22 04:12:39 +03:00
2022-10-05 06:56:30 +03:00
skip . click (
fn = lambda : shared . state . skip ( ) ,
inputs = [ ] ,
outputs = [ ] ,
)
2022-09-22 04:12:39 +03:00
interrupt . click (
fn = lambda : shared . state . interrupt ( ) ,
inputs = [ ] ,
outputs = [ ] ,
)
2022-09-14 17:56:21 +03:00
2022-10-15 14:22:30 +03:00
with gr . Row ( ) :
with gr . Column ( scale = 1 , elem_id = " style_pos_col " ) :
prompt_style = gr . Dropdown ( label = " Style 1 " , elem_id = f " { id_part } _style_index " , choices = [ k for k , v in shared . prompt_styles . styles . items ( ) ] , value = next ( iter ( shared . prompt_styles . styles . keys ( ) ) ) )
with gr . Column ( scale = 1 , elem_id = " style_neg_col " ) :
prompt_style2 = gr . Dropdown ( label = " Style 2 " , elem_id = f " { id_part } _style2_index " , choices = [ k for k , v in shared . prompt_styles . styles . items ( ) ] , value = next ( iter ( shared . prompt_styles . styles . keys ( ) ) ) )
2023-01-01 14:07:40 +03:00
return prompt , prompt_style , negative_prompt , prompt_style2 , submit , button_interrogate , button_deepbooru , prompt_style_apply , save_style , paste , token_counter , token_button
2022-09-14 17:56:21 +03:00
2023-01-10 12:29:45 +03:00
def setup_progressbar ( * args , * * kwargs ) :
import modules . ui_progress
2022-09-18 11:14:42 +03:00
2023-01-10 12:29:45 +03:00
modules . ui_progress . setup_progressbar ( * args , * * kwargs )
2022-09-14 17:56:21 +03:00
2022-10-14 19:30:28 +03:00
def apply_setting ( key , value ) :
if value is None :
return gr . update ( )
2022-10-22 22:05:22 +03:00
if shared . cmd_opts . freeze_settings :
return gr . update ( )
2022-10-17 17:58:21 +03:00
# dont allow model to be swapped when model hash exists in prompt
if key == " sd_model_checkpoint " and opts . disable_weights_auto_swap :
return gr . update ( )
2022-10-14 19:30:28 +03:00
if key == " sd_model_checkpoint " :
ckpt_info = sd_models . get_closet_checkpoint_match ( value )
if ckpt_info is not None :
value = ckpt_info . title
else :
return gr . update ( )
comp_args = opts . data_labels [ key ] . component_args
if comp_args and isinstance ( comp_args , dict ) and comp_args . get ( ' visible ' ) is False :
return
valtype = type ( opts . data_labels [ key ] . default )
2023-01-02 20:46:51 +03:00
oldval = opts . data . get ( key , None )
2022-10-14 19:30:28 +03:00
opts . data [ key ] = valtype ( value ) if valtype != type ( None ) else value
if oldval != value and opts . data_labels [ key ] . onchange is not None :
opts . data_labels [ key ] . onchange ( )
opts . save ( shared . config_filename )
return value
2022-11-09 23:24:31 +03:00
def update_generation_info ( args ) :
generation_info , html_info , img_index = args
try :
generation_info = json . loads ( generation_info )
2022-11-10 21:07:41 +03:00
if img_index < 0 or img_index > = len ( generation_info [ " infotexts " ] ) :
return html_info
2022-11-09 23:24:31 +03:00
return plaintext_to_html ( generation_info [ " infotexts " ] [ img_index ] )
except Exception :
pass
# if the json parse or anything else fails, just return the old html_info
return html_info
2022-10-02 15:03:39 +03:00
2022-10-16 07:42:52 +03:00
2022-10-21 16:10:51 +03:00
def create_refresh_button ( refresh_component , refresh_method , refreshed_args , elem_id ) :
def refresh ( ) :
refresh_method ( )
args = refreshed_args ( ) if callable ( refreshed_args ) else refreshed_args
2022-10-02 15:03:39 +03:00
2022-10-21 16:10:51 +03:00
for k , v in args . items ( ) :
setattr ( refresh_component , k , v )
2022-10-16 07:42:52 +03:00
2022-10-21 16:10:51 +03:00
return gr . update ( * * ( args or { } ) )
2022-10-16 07:42:52 +03:00
2023-01-03 09:04:29 +03:00
refresh_button = ToolButton ( value = refresh_symbol , elem_id = elem_id )
2022-10-21 16:10:51 +03:00
refresh_button . click (
fn = refresh ,
inputs = [ ] ,
outputs = [ refresh_component ]
)
return refresh_button
2022-10-16 07:42:52 +03:00
2022-10-29 08:28:48 +03:00
def create_output_panel ( tabname , outdir ) :
2022-10-27 08:36:11 +03:00
def open_folder ( f ) :
if not os . path . exists ( f ) :
print ( f ' Folder " { f } " does not exist. After you create an image, the folder will be created. ' )
return
elif not os . path . isdir ( f ) :
print ( f """
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 : { f }
""" , file=sys.stderr)
return
if not shared . cmd_opts . hide_ui_dir_config :
path = os . path . normpath ( f )
if platform . system ( ) == " Windows " :
os . startfile ( path )
elif platform . system ( ) == " Darwin " :
sp . Popen ( [ " open " , path ] )
2023-01-06 04:58:22 +03:00
elif " microsoft-standard-WSL2 " in platform . uname ( ) . release :
sp . Popen ( [ " wsl-open " , path ] )
2022-10-27 08:36:11 +03:00
else :
sp . Popen ( [ " xdg-open " , path ] )
2022-10-29 08:28:48 +03:00
with gr . Column ( variant = ' panel ' ) :
with gr . Group ( ) :
2022-10-27 08:36:11 +03:00
result_gallery = gr . Gallery ( label = ' Output ' , show_label = False , elem_id = f " { tabname } _gallery " ) . style ( grid = 4 )
2022-10-29 08:28:48 +03:00
2022-10-27 08:36:11 +03:00
generation_info = None
with gr . Column ( ) :
2023-01-01 00:38:58 +03:00
with gr . Row ( elem_id = f " image_buttons_ { tabname } " ) :
2023-01-05 00:03:32 +03:00
open_folder_button = gr . Button ( folder_symbol , elem_id = " hidden_element " if shared . cmd_opts . hide_ui_dir_config else f ' open_folder_ { tabname } ' )
2022-09-19 16:42:56 +03:00
2022-10-27 08:36:11 +03:00
if tabname != " extras " :
2022-10-29 09:13:36 +03:00
save = gr . Button ( ' Save ' , elem_id = f ' save_ { tabname } ' )
2023-01-01 00:38:58 +03:00
save_zip = gr . Button ( ' Zip ' , elem_id = f ' save_zip_ { tabname } ' )
2022-10-27 08:36:11 +03:00
buttons = parameters_copypaste . create_buttons ( [ " img2img " , " inpaint " , " extras " ] )
2022-10-29 08:26:12 +03:00
open_folder_button . click (
2022-10-27 08:36:11 +03:00
fn = lambda : open_folder ( opts . outdir_samples or outdir ) ,
inputs = [ ] ,
outputs = [ ] ,
)
2022-10-29 08:28:48 +03:00
if tabname != " extras " :
2022-10-27 08:36:11 +03:00
with gr . Row ( ) :
2023-01-05 00:03:32 +03:00
download_files = gr . File ( None , file_count = " multiple " , interactive = False , show_label = False , visible = False , elem_id = f ' download_files_ { tabname } ' )
2022-10-27 08:36:11 +03:00
with gr . Group ( ) :
2023-01-05 00:03:32 +03:00
html_info = gr . HTML ( elem_id = f ' html_info_ { tabname } ' )
html_log = gr . HTML ( elem_id = f ' html_log_ { tabname } ' )
2022-09-03 12:08:45 +03:00
2023-01-05 00:03:32 +03:00
generation_info = gr . Textbox ( visible = False , elem_id = f ' generation_info_ { tabname } ' )
2022-11-09 23:24:31 +03:00
if tabname == ' txt2img ' or tabname == ' img2img ' :
generation_info_button = gr . Button ( visible = False , elem_id = f " { tabname } _generation_info_button " )
generation_info_button . click (
fn = update_generation_info ,
_js = " (x, y) => [x, y, selected_gallery_index()] " ,
inputs = [ generation_info , html_info ] ,
outputs = [ html_info ] ,
preprocess = False
)
2022-10-27 08:36:11 +03:00
save . click (
fn = wrap_gradio_call ( save_files ) ,
2023-01-01 00:38:58 +03:00
_js = " (x, y, z, w) => [x, y, false, selected_gallery_index()] " ,
2022-10-27 08:36:11 +03:00
inputs = [
generation_info ,
result_gallery ,
2023-01-01 00:38:58 +03:00
html_info ,
2022-10-27 08:36:11 +03:00
html_info ,
] ,
outputs = [
download_files ,
2023-01-01 00:38:58 +03:00
html_log ,
]
)
2022-09-17 12:38:15 +03:00
2023-01-01 00:38:58 +03:00
save_zip . click (
fn = wrap_gradio_call ( save_files ) ,
_js = " (x, y, z, w) => [x, y, true, selected_gallery_index()] " ,
2022-10-27 08:36:11 +03:00
inputs = [
generation_info ,
result_gallery ,
html_info ,
html_info ,
] ,
outputs = [
download_files ,
2022-12-31 23:40:55 +03:00
html_log ,
2022-10-27 08:36:11 +03:00
]
)
2022-09-03 12:08:45 +03:00
2022-10-27 08:36:11 +03:00
else :
2023-01-05 00:03:32 +03:00
html_info_x = gr . HTML ( elem_id = f ' html_info_x_ { tabname } ' )
html_info = gr . HTML ( elem_id = f ' html_info_ { tabname } ' )
html_log = gr . HTML ( elem_id = f ' html_log_ { tabname } ' )
2022-09-03 12:08:45 +03:00
2022-10-27 08:36:11 +03:00
parameters_copypaste . bind_buttons ( buttons , result_gallery , " txt2img " if tabname == " txt2img " else None )
2022-12-31 23:40:55 +03:00
return result_gallery , generation_info if tabname != " extras " else html_info_x , html_info , html_log
2022-10-10 04:26:52 +03:00
2022-10-08 08:09:29 +03:00
2023-01-01 01:19:10 +03:00
def create_sampler_and_steps_selection ( choices , tabname ) :
if opts . samplers_in_dropdown :
2023-01-03 09:04:29 +03:00
with FormRow ( elem_id = f " sampler_selection_ { tabname } " ) :
2023-01-01 01:19:10 +03:00
sampler_index = gr . Dropdown ( label = ' Sampling method ' , elem_id = f " { tabname } _sampling " , choices = [ x . name for x in choices ] , value = choices [ 0 ] . name , type = " index " )
2023-01-04 22:04:40 +03:00
steps = gr . Slider ( minimum = 1 , maximum = 150 , step = 1 , elem_id = f " { tabname } _steps " , label = " Sampling steps " , value = 20 )
2023-01-01 01:19:10 +03:00
else :
2023-01-03 09:04:29 +03:00
with FormGroup ( elem_id = f " sampler_selection_ { tabname } " ) :
2023-01-04 22:04:40 +03:00
steps = gr . Slider ( minimum = 1 , maximum = 150 , step = 1 , elem_id = f " { tabname } _steps " , label = " Sampling steps " , value = 20 )
2023-01-01 01:19:10 +03:00
sampler_index = gr . Radio ( label = ' Sampling method ' , elem_id = f " { tabname } _sampling " , choices = [ x . name for x in choices ] , value = choices [ 0 ] . name , type = " index " )
return steps , sampler_index
2022-10-29 08:28:48 +03:00
2022-10-16 07:42:52 +03:00
2023-01-03 10:39:21 +03:00
def ordered_ui_categories ( ) :
user_order = { x . strip ( ) : i for i , x in enumerate ( shared . opts . ui_reorder . split ( " , " ) ) }
for i , category in sorted ( enumerate ( shared . ui_reorder_categories ) , key = lambda x : user_order . get ( x [ 1 ] , x [ 0 ] + 1000 ) ) :
yield category
2022-11-28 09:00:10 +03:00
def create_ui ( ) :
2022-10-21 16:10:51 +03:00
import modules . img2img
import modules . txt2img
2022-10-16 07:42:52 +03:00
2022-11-02 07:26:31 +03:00
reload_javascript ( )
2022-10-31 17:36:45 +03:00
parameters_copypaste . reset ( )
2022-10-16 07:42:52 +03:00
2022-11-19 19:10:17 +03:00
modules . scripts . scripts_current = modules . scripts . scripts_txt2img
modules . scripts . scripts_txt2img . initialize_scripts ( is_img2img = False )
2022-09-03 12:08:45 +03:00
with gr . Blocks ( analytics_enabled = False ) as txt2img_interface :
2023-01-01 14:07:40 +03:00
txt2img_prompt , txt2img_prompt_style , txt2img_negative_prompt , txt2img_prompt_style2 , submit , _ , _ , txt2img_prompt_style_apply , txt2img_save_style , txt2img_paste , token_counter , token_button = create_toprow ( is_img2img = False )
2022-10-20 05:23:57 +03:00
2022-09-19 09:02:10 +03:00
dummy_component = gr . Label ( visible = False )
2022-10-14 18:15:03 +03:00
txt_prompt_img = gr . File ( label = " " , elem_id = " txt2img_prompt_image " , file_count = " single " , type = " bytes " , visible = False )
2022-09-03 12:08:45 +03:00
2022-09-23 20:46:02 +03:00
with gr . Row ( elem_id = ' txt2img_progress_row ' ) :
with gr . Column ( scale = 1 ) :
pass
2022-09-22 04:12:39 +03:00
2022-09-23 20:46:02 +03:00
with gr . Column ( scale = 1 ) :
progressbar = gr . HTML ( elem_id = " txt2img_progressbar " )
2022-09-22 04:12:39 +03:00
txt2img_preview = gr . Image ( elem_id = ' txt2img_preview ' , visible = False )
2022-09-23 20:46:02 +03:00
setup_progressbar ( progressbar , txt2img_preview , ' txt2img ' )
2022-09-22 04:12:39 +03:00
2022-09-03 12:08:45 +03:00
with gr . Row ( ) . style ( equal_height = False ) :
2023-01-14 13:38:10 +03:00
with gr . Column ( variant = ' compact ' , elem_id = " txt2img_settings " ) :
2023-01-03 10:39:21 +03:00
for category in ordered_ui_categories ( ) :
if category == " sampler " :
steps , sampler_index = create_sampler_and_steps_selection ( samplers , " txt2img " )
2022-09-19 16:42:56 +03:00
2023-01-03 10:39:21 +03:00
elif category == " dimensions " :
with FormRow ( ) :
with gr . Column ( elem_id = " txt2img_column_size " , scale = 4 ) :
width = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Width " , value = 512 , elem_id = " txt2img_width " )
height = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Height " , value = 512 , elem_id = " txt2img_height " )
if opts . dimensions_and_batch_together :
with gr . Column ( elem_id = " txt2img_column_batch " ) :
batch_count = gr . Slider ( minimum = 1 , step = 1 , label = ' Batch count ' , value = 1 , elem_id = " txt2img_batch_count " )
batch_size = gr . Slider ( minimum = 1 , maximum = 8 , step = 1 , label = ' Batch size ' , value = 1 , elem_id = " txt2img_batch_size " )
elif category == " cfg " :
cfg_scale = gr . Slider ( minimum = 1.0 , maximum = 30.0 , step = 0.5 , label = ' CFG Scale ' , value = 7.0 , elem_id = " txt2img_cfg_scale " )
elif category == " seed " :
seed , reuse_seed , subseed , reuse_subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox = create_seed_inputs ( ' txt2img ' )
elif category == " checkboxes " :
with FormRow ( elem_id = " txt2img_checkboxes " ) :
restore_faces = gr . Checkbox ( label = ' Restore faces ' , value = False , visible = len ( shared . face_restorers ) > 1 , elem_id = " txt2img_restore_faces " )
tiling = gr . Checkbox ( label = ' Tiling ' , value = False , elem_id = " txt2img_tiling " )
enable_hr = gr . Checkbox ( label = ' Hires. fix ' , value = False , elem_id = " txt2img_enable_hr " )
2023-01-07 09:56:37 +03:00
hr_final_resolution = FormHTML ( value = " " , elem_id = " txtimg_hr_finalres " , label = " Upscaled resolution " , interactive = False )
2023-01-03 10:39:21 +03:00
elif category == " hires_fix " :
2023-01-04 22:04:40 +03:00
with FormGroup ( visible = False , elem_id = " txt2img_hires_fix " ) as hr_options :
with FormRow ( elem_id = " txt2img_hires_fix_row1 " ) :
hr_upscaler = gr . Dropdown ( label = " Upscaler " , elem_id = " txt2img_hr_upscaler " , choices = [ * shared . latent_upscale_modes , * [ x . name for x in shared . sd_upscalers ] ] , value = shared . latent_upscale_default_mode )
hr_second_pass_steps = gr . Slider ( minimum = 0 , maximum = 150 , step = 1 , label = ' Hires steps ' , value = 0 , elem_id = " txt2img_hires_steps " )
denoising_strength = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.01 , label = ' Denoising strength ' , value = 0.7 , elem_id = " txt2img_denoising_strength " )
with FormRow ( elem_id = " txt2img_hires_fix_row2 " ) :
hr_scale = gr . Slider ( minimum = 1.0 , maximum = 4.0 , step = 0.05 , label = " Upscale by " , value = 2.0 , elem_id = " txt2img_hr_scale " )
hr_resize_x = gr . Slider ( minimum = 0 , maximum = 2048 , step = 8 , label = " Resize width to " , value = 0 , elem_id = " txt2img_hr_resize_x " )
hr_resize_y = gr . Slider ( minimum = 0 , maximum = 2048 , step = 8 , label = " Resize height to " , value = 0 , elem_id = " txt2img_hr_resize_y " )
2023-01-03 10:39:21 +03:00
elif category == " batch " :
if not opts . dimensions_and_batch_together :
with FormRow ( elem_id = " txt2img_column_batch " ) :
batch_count = gr . Slider ( minimum = 1 , step = 1 , label = ' Batch count ' , value = 1 , elem_id = " txt2img_batch_count " )
batch_size = gr . Slider ( minimum = 1 , maximum = 8 , step = 1 , label = ' Batch size ' , value = 1 , elem_id = " txt2img_batch_size " )
elif category == " scripts " :
with FormGroup ( elem_id = " txt2img_script_container " ) :
custom_inputs = modules . scripts . scripts_txt2img . setup_ui ( )
2022-09-03 12:08:45 +03:00
2023-01-07 09:56:37 +03:00
hr_resolution_preview_inputs = [ enable_hr , width , height , hr_scale , hr_resize_x , hr_resize_y ]
for input in hr_resolution_preview_inputs :
2023-01-09 14:57:47 +03:00
input . change (
fn = calc_resolution_hires ,
inputs = hr_resolution_preview_inputs ,
outputs = [ hr_final_resolution ] ,
show_progress = False ,
)
input . change (
None ,
_js = " onCalcResolutionHires " ,
inputs = hr_resolution_preview_inputs ,
outputs = [ ] ,
show_progress = False ,
)
2023-01-07 08:53:53 +03:00
2022-12-31 23:40:55 +03:00
txt2img_gallery , generation_info , html_info , html_log = create_output_panel ( " txt2img " , opts . outdir_txt2img_samples )
2022-10-29 10:56:19 +03:00
parameters_copypaste . bind_buttons ( { " txt2img " : txt2img_paste } , None , txt2img_prompt )
2022-09-03 12:08:45 +03:00
2022-09-19 09:02:10 +03:00
connect_reuse_seed ( seed , reuse_seed , generation_info , dummy_component , is_subseed = False )
connect_reuse_seed ( subseed , reuse_subseed , generation_info , dummy_component , is_subseed = True )
2022-09-16 22:20:56 +03:00
2022-09-03 12:08:45 +03:00
txt2img_args = dict (
2022-12-31 23:40:55 +03:00
fn = wrap_gradio_gpu_call ( modules . txt2img . txt2img , extra_outputs = [ None , ' ' , ' ' ] ) ,
2022-09-06 02:09:01 +03:00
_js = " submit " ,
2022-09-03 12:08:45 +03:00
inputs = [
2022-09-09 23:16:02 +03:00
txt2img_prompt ,
2022-09-11 17:35:12 +03:00
txt2img_negative_prompt ,
2022-09-09 23:16:02 +03:00
txt2img_prompt_style ,
2022-09-14 17:56:21 +03:00
txt2img_prompt_style2 ,
2022-09-03 12:08:45 +03:00
steps ,
sampler_index ,
2022-09-07 12:32:28 +03:00
restore_faces ,
2022-09-05 03:25:37 +03:00
tiling ,
2022-09-03 12:08:45 +03:00
batch_count ,
batch_size ,
cfg_scale ,
seed ,
2022-09-21 13:34:10 +03:00
subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox ,
2022-09-03 12:08:45 +03:00
height ,
width ,
2022-09-19 16:42:56 +03:00
enable_hr ,
denoising_strength ,
2023-01-02 19:42:10 +03:00
hr_scale ,
hr_upscaler ,
2023-01-04 22:04:40 +03:00
hr_second_pass_steps ,
hr_resize_x ,
hr_resize_y ,
2022-09-03 17:21:15 +03:00
] + custom_inputs ,
2022-10-15 17:20:17 +03:00
2022-09-03 12:08:45 +03:00
outputs = [
txt2img_gallery ,
generation_info ,
2022-12-31 23:40:55 +03:00
html_info ,
html_log ,
2022-09-18 11:14:42 +03:00
] ,
show_progress = False ,
2022-09-03 12:08:45 +03:00
)
2022-09-09 23:16:02 +03:00
txt2img_prompt . submit ( * * txt2img_args )
2022-09-03 12:08:45 +03:00
submit . click ( * * txt2img_args )
2022-10-13 02:17:26 +03:00
txt_prompt_img . change (
fn = modules . images . image_data ,
inputs = [
txt_prompt_img
] ,
outputs = [
txt2img_prompt ,
txt_prompt_img
]
)
2022-09-19 16:42:56 +03:00
enable_hr . change (
fn = lambda x : gr_show ( x ) ,
inputs = [ enable_hr ] ,
outputs = [ hr_options ] ,
2023-01-07 09:56:37 +03:00
show_progress = False ,
2022-09-19 16:42:56 +03:00
)
2022-09-25 09:25:28 +03:00
txt2img_paste_fields = [
( txt2img_prompt , " Prompt " ) ,
( txt2img_negative_prompt , " Negative prompt " ) ,
( steps , " Steps " ) ,
( sampler_index , " Sampler " ) ,
( restore_faces , " Face restoration " ) ,
( cfg_scale , " CFG scale " ) ,
( seed , " Seed " ) ,
( width , " Size-1 " ) ,
( height , " Size-2 " ) ,
( batch_size , " Batch size " ) ,
( subseed , " Variation seed " ) ,
( subseed_strength , " Variation seed strength " ) ,
( seed_resize_from_w , " Seed resize from-1 " ) ,
( seed_resize_from_h , " Seed resize from-2 " ) ,
( denoising_strength , " Denoising strength " ) ,
( enable_hr , lambda d : " Denoising strength " in d ) ,
( hr_options , lambda d : gr . Row . update ( visible = " Denoising strength " in d ) ) ,
2023-01-02 19:42:10 +03:00
( hr_scale , " Hires upscale " ) ,
( hr_upscaler , " Hires upscaler " ) ,
2023-01-04 22:04:40 +03:00
( hr_second_pass_steps , " Hires steps " ) ,
( hr_resize_x , " Hires resize-1 " ) ,
( hr_resize_y , " Hires resize-2 " ) ,
2022-10-22 12:23:45 +03:00
* modules . scripts . scripts_txt2img . infotext_fields
2022-09-25 09:25:28 +03:00
]
2022-10-29 10:56:19 +03:00
parameters_copypaste . add_paste_fields ( " txt2img " , None , txt2img_paste_fields )
2022-10-14 20:31:49 +03:00
txt2img_preview_params = [
txt2img_prompt ,
txt2img_negative_prompt ,
steps ,
sampler_index ,
cfg_scale ,
seed ,
width ,
height ,
]
2022-11-28 09:00:10 +03:00
token_button . click ( fn = wrap_queued_call ( update_token_counter ) , inputs = [ txt2img_prompt , steps ] , outputs = [ token_counter ] )
2022-09-23 22:49:21 +03:00
2022-11-19 19:10:17 +03:00
modules . scripts . scripts_current = modules . scripts . scripts_img2img
modules . scripts . scripts_img2img . initialize_scripts ( is_img2img = True )
2022-09-23 22:49:21 +03:00
2022-09-03 12:08:45 +03:00
with gr . Blocks ( analytics_enabled = False ) as img2img_interface :
2023-01-01 14:07:40 +03:00
img2img_prompt , img2img_prompt_style , img2img_negative_prompt , img2img_prompt_style2 , submit , img2img_interrogate , img2img_deepbooru , img2img_prompt_style_apply , img2img_save_style , img2img_paste , token_counter , token_button = create_toprow ( is_img2img = True )
2022-09-03 12:08:45 +03:00
2022-09-23 20:46:02 +03:00
with gr . Row ( elem_id = ' img2img_progress_row ' ) :
2022-10-14 18:15:03 +03:00
img2img_prompt_img = gr . File ( label = " " , elem_id = " img2img_prompt_image " , file_count = " single " , type = " bytes " , visible = False )
2022-10-13 02:17:26 +03:00
2022-09-23 20:46:02 +03:00
with gr . Column ( scale = 1 ) :
pass
2022-09-22 04:12:39 +03:00
2022-09-23 20:46:02 +03:00
with gr . Column ( scale = 1 ) :
progressbar = gr . HTML ( elem_id = " img2img_progressbar " )
2022-09-22 04:12:39 +03:00
img2img_preview = gr . Image ( elem_id = ' img2img_preview ' , visible = False )
2022-09-23 20:46:02 +03:00
setup_progressbar ( progressbar , img2img_preview , ' img2img ' )
2022-09-22 12:11:48 +03:00
2023-01-03 09:04:29 +03:00
with FormRow ( ) . style ( equal_height = False ) :
2023-01-14 13:38:10 +03:00
with gr . Column ( variant = ' compact ' , elem_id = " img2img_settings " ) :
2023-01-11 20:33:24 +03:00
with gr . Tabs ( elem_id = " mode_img2img " ) :
with gr . TabItem ( ' img2img ' , id = ' img2img ' , elem_id = " img2img_img2img_tab " ) as tab_img2img :
init_img = gr . Image ( label = " Image for img2img " , elem_id = " img2img_image " , show_label = False , source = " upload " , interactive = True , type = " pil " , tool = " editor " , image_mode = " RGBA " ) . style ( height = 480 )
2022-09-09 19:43:16 +03:00
2023-01-11 20:33:24 +03:00
with gr . TabItem ( ' Sketch ' , id = ' img2img_sketch ' , elem_id = " img2img_img2img_sketch_tab " ) as tab_sketch :
sketch = gr . Image ( label = " Image for img2img " , elem_id = " img2img_sketch " , show_label = False , source = " upload " , interactive = True , type = " pil " , tool = " color-sketch " , image_mode = " RGBA " ) . style ( height = 480 )
2022-09-03 12:08:45 +03:00
2023-01-11 20:33:24 +03:00
with gr . TabItem ( ' Inpaint ' , id = ' inpaint ' , elem_id = " img2img_inpaint_tab " ) as tab_inpaint :
init_img_with_mask = gr . Image ( label = " Image for inpainting with mask " , show_label = False , elem_id = " img2maskimg " , source = " upload " , interactive = True , type = " pil " , tool = " sketch " , image_mode = " RGBA " ) . style ( height = 480 )
2022-09-22 12:11:48 +03:00
2023-01-11 20:33:24 +03:00
with gr . TabItem ( ' Inpaint sketch ' , id = ' inpaint_sketch ' , elem_id = " img2img_inpaint_sketch_tab " ) as tab_inpaint_color :
inpaint_color_sketch = gr . Image ( label = " Color sketch inpainting " , show_label = False , elem_id = " inpaint_sketch " , source = " upload " , interactive = True , type = " pil " , tool = " color-sketch " , image_mode = " RGBA " ) . style ( height = 480 )
inpaint_color_sketch_orig = gr . State ( None )
2022-09-03 12:08:45 +03:00
2023-01-11 20:33:24 +03:00
def update_orig ( image , state ) :
if image is not None :
same_size = state is not None and state . size == image . size
has_exact_match = np . any ( np . all ( np . array ( image ) == np . array ( state ) , axis = - 1 ) )
edited = same_size and has_exact_match
return image if not edited or state is None else state
2022-09-22 12:11:48 +03:00
2023-01-11 20:33:24 +03:00
inpaint_color_sketch . change ( update_orig , [ inpaint_color_sketch , inpaint_color_sketch_orig ] , inpaint_color_sketch_orig )
2022-09-22 12:11:48 +03:00
2023-01-11 20:33:24 +03:00
with gr . TabItem ( ' Inpaint upload ' , id = ' inpaint_upload ' , elem_id = " img2img_inpaint_upload_tab " ) as tab_inpaint_upload :
init_img_inpaint = gr . Image ( label = " Image for img2img " , show_label = False , source = " upload " , interactive = True , type = " pil " , elem_id = " img_inpaint_base " )
init_mask_inpaint = gr . Image ( label = " Mask " , source = " upload " , interactive = True , type = " pil " , elem_id = " img_inpaint_mask " )
2022-09-22 12:11:48 +03:00
2023-01-11 20:33:24 +03:00
with gr . TabItem ( ' Batch ' , id = ' batch ' , elem_id = " img2img_batch_tab " ) as tab_batch :
2022-09-24 16:29:20 +03:00
hidden = ' <br>Disabled when launched with --hide-ui-dir-config. ' if shared . cmd_opts . hide_ui_dir_config else ' '
2022-10-03 11:48:19 +03:00
gr . HTML ( f " <p class= \" text-gray-500 \" >Process images in a directory on the same machine where the server is running.<br>Use an empty output directory to save pictures normally instead of writing to the output directory. { hidden } </p> " )
2023-01-01 16:51:12 +03:00
img2img_batch_input_dir = gr . Textbox ( label = " Input directory " , * * shared . hide_dirs , elem_id = " img2img_batch_input_dir " )
img2img_batch_output_dir = gr . Textbox ( label = " Output directory " , * * shared . hide_dirs , elem_id = " img2img_batch_output_dir " )
2022-09-03 12:08:45 +03:00
2023-01-11 20:33:24 +03:00
with FormGroup ( elem_id = " inpaint_controls " , visible = False ) as inpaint_controls :
with FormRow ( ) :
mask_blur = gr . Slider ( label = ' Mask blur ' , minimum = 0 , maximum = 64 , step = 1 , value = 4 , elem_id = " img2img_mask_blur " )
mask_alpha = gr . Slider ( label = " Mask transparency " , visible = False , elem_id = " img2img_mask_alpha " )
with FormRow ( ) :
inpainting_mask_invert = gr . Radio ( label = ' Mask mode ' , choices = [ ' Inpaint masked ' , ' Inpaint not masked ' ] , value = ' Inpaint masked ' , type = " index " , elem_id = " img2img_mask_mode " )
with FormRow ( ) :
inpainting_fill = gr . Radio ( label = ' Masked content ' , choices = [ ' fill ' , ' original ' , ' latent noise ' , ' latent nothing ' ] , value = ' original ' , type = " index " , elem_id = " img2img_inpainting_fill " )
with FormRow ( ) :
with gr . Column ( ) :
inpaint_full_res = gr . Radio ( label = " Inpaint area " , choices = [ " Whole picture " , " Only masked " ] , type = " index " , value = " Whole picture " , elem_id = " img2img_inpaint_full_res " )
with gr . Column ( scale = 4 ) :
inpaint_full_res_padding = gr . Slider ( label = ' Only masked padding, pixels ' , minimum = 0 , maximum = 256 , step = 4 , value = 32 , elem_id = " img2img_inpaint_full_res_padding " )
def select_img2img_tab ( tab ) :
return gr . update ( visible = tab in [ 2 , 3 , 4 ] ) , gr . update ( visible = tab == 3 ) ,
for i , elem in enumerate ( [ tab_img2img , tab_sketch , tab_inpaint , tab_inpaint_color , tab_inpaint_upload , tab_batch ] ) :
elem . select (
fn = lambda tab = i : select_img2img_tab ( tab ) ,
inputs = [ ] ,
outputs = [ inpaint_controls , mask_alpha ] ,
)
2023-01-03 09:04:29 +03:00
with FormRow ( ) :
resize_mode = gr . Radio ( label = " Resize mode " , elem_id = " resize_mode " , choices = [ " Just resize " , " Crop and resize " , " Resize and fill " , " Just resize (latent upscale) " ] , type = " index " , value = " Just resize " )
2022-09-22 12:11:48 +03:00
2023-01-03 10:39:21 +03:00
for category in ordered_ui_categories ( ) :
if category == " sampler " :
steps , sampler_index = create_sampler_and_steps_selection ( samplers_for_img2img , " img2img " )
2022-09-03 21:02:38 +03:00
2023-01-03 10:39:21 +03:00
elif category == " dimensions " :
with FormRow ( ) :
with gr . Column ( elem_id = " img2img_column_size " , scale = 4 ) :
width = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Width " , value = 512 , elem_id = " img2img_width " )
height = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Height " , value = 512 , elem_id = " img2img_height " )
2022-09-25 08:40:29 +03:00
2023-01-03 10:39:21 +03:00
if opts . dimensions_and_batch_together :
with gr . Column ( elem_id = " img2img_column_batch " ) :
batch_count = gr . Slider ( minimum = 1 , step = 1 , label = ' Batch count ' , value = 1 , elem_id = " img2img_batch_count " )
batch_size = gr . Slider ( minimum = 1 , maximum = 8 , step = 1 , label = ' Batch size ' , value = 1 , elem_id = " img2img_batch_size " )
2022-09-03 12:08:45 +03:00
2023-01-03 10:39:21 +03:00
elif category == " cfg " :
with FormGroup ( ) :
cfg_scale = gr . Slider ( minimum = 1.0 , maximum = 30.0 , step = 0.5 , label = ' CFG Scale ' , value = 7.0 , elem_id = " img2img_cfg_scale " )
denoising_strength = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.01 , label = ' Denoising strength ' , value = 0.75 , elem_id = " img2img_denoising_strength " )
2022-09-03 12:08:45 +03:00
2023-01-03 10:39:21 +03:00
elif category == " seed " :
seed , reuse_seed , subseed , reuse_subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox = create_seed_inputs ( ' img2img ' )
2022-09-03 12:08:45 +03:00
2023-01-03 10:39:21 +03:00
elif category == " checkboxes " :
with FormRow ( elem_id = " img2img_checkboxes " ) :
restore_faces = gr . Checkbox ( label = ' Restore faces ' , value = False , visible = len ( shared . face_restorers ) > 1 , elem_id = " img2img_restore_faces " )
tiling = gr . Checkbox ( label = ' Tiling ' , value = False , elem_id = " img2img_tiling " )
2022-09-03 12:08:45 +03:00
2023-01-03 10:39:21 +03:00
elif category == " batch " :
if not opts . dimensions_and_batch_together :
with FormRow ( elem_id = " img2img_column_batch " ) :
batch_count = gr . Slider ( minimum = 1 , step = 1 , label = ' Batch count ' , value = 1 , elem_id = " img2img_batch_count " )
batch_size = gr . Slider ( minimum = 1 , maximum = 8 , step = 1 , label = ' Batch size ' , value = 1 , elem_id = " img2img_batch_size " )
2022-09-03 17:21:15 +03:00
2023-01-03 10:39:21 +03:00
elif category == " scripts " :
with FormGroup ( elem_id = " img2img_script_container " ) :
custom_inputs = modules . scripts . scripts_img2img . setup_ui ( )
2022-09-17 12:38:15 +03:00
2022-12-31 23:40:55 +03:00
img2img_gallery , generation_info , html_info , html_log = create_output_panel ( " img2img " , opts . outdir_img2img_samples )
2022-10-29 10:56:19 +03:00
parameters_copypaste . bind_buttons ( { " img2img " : img2img_paste } , None , img2img_prompt )
2022-09-03 12:08:45 +03:00
2022-09-19 09:02:10 +03:00
connect_reuse_seed ( seed , reuse_seed , generation_info , dummy_component , is_subseed = False )
connect_reuse_seed ( subseed , reuse_subseed , generation_info , dummy_component , is_subseed = True )
2022-09-16 22:20:56 +03:00
2022-10-13 02:17:26 +03:00
img2img_prompt_img . change (
fn = modules . images . image_data ,
inputs = [
2022-10-14 18:15:03 +03:00
img2img_prompt_img
2022-10-13 02:17:26 +03:00
] ,
outputs = [
img2img_prompt ,
img2img_prompt_img
]
)
2022-09-03 12:08:45 +03:00
img2img_args = dict (
2022-12-31 23:40:55 +03:00
fn = wrap_gradio_gpu_call ( modules . img2img . img2img , extra_outputs = [ None , ' ' , ' ' ] ) ,
2022-09-22 12:11:48 +03:00
_js = " submit_img2img " ,
2022-09-03 12:08:45 +03:00
inputs = [
2022-09-22 12:11:48 +03:00
dummy_component ,
2022-09-09 23:16:02 +03:00
img2img_prompt ,
2022-09-11 17:35:12 +03:00
img2img_negative_prompt ,
2022-09-09 23:16:02 +03:00
img2img_prompt_style ,
2022-09-14 17:56:21 +03:00
img2img_prompt_style2 ,
2022-09-03 12:08:45 +03:00
init_img ,
2023-01-11 20:33:24 +03:00
sketch ,
2022-09-03 12:08:45 +03:00
init_img_with_mask ,
2023-01-11 20:33:24 +03:00
inpaint_color_sketch ,
inpaint_color_sketch_orig ,
2022-09-22 12:11:48 +03:00
init_img_inpaint ,
init_mask_inpaint ,
2022-09-03 12:08:45 +03:00
steps ,
sampler_index ,
mask_blur ,
2022-11-09 06:06:29 +03:00
mask_alpha ,
2022-09-03 12:08:45 +03:00
inpainting_fill ,
2022-09-07 12:32:28 +03:00
restore_faces ,
2022-09-05 03:25:37 +03:00
tiling ,
2022-09-03 12:08:45 +03:00
batch_count ,
batch_size ,
cfg_scale ,
denoising_strength ,
seed ,
2022-09-21 13:34:10 +03:00
subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox ,
2022-09-03 12:08:45 +03:00
height ,
width ,
resize_mode ,
inpaint_full_res ,
2022-09-22 12:11:48 +03:00
inpaint_full_res_padding ,
2022-09-03 21:02:38 +03:00
inpainting_mask_invert ,
2022-09-22 12:11:48 +03:00
img2img_batch_input_dir ,
img2img_batch_output_dir ,
2022-09-03 17:21:15 +03:00
] + custom_inputs ,
2022-09-03 12:08:45 +03:00
outputs = [
img2img_gallery ,
generation_info ,
2022-12-31 23:40:55 +03:00
html_info ,
html_log ,
2022-09-18 11:14:42 +03:00
] ,
show_progress = False ,
2022-09-03 12:08:45 +03:00
)
2022-09-09 23:16:02 +03:00
img2img_prompt . submit ( * * img2img_args )
2022-09-03 12:08:45 +03:00
submit . click ( * * img2img_args )
2022-09-11 18:48:36 +03:00
img2img_interrogate . click (
fn = interrogate ,
inputs = [ init_img ] ,
outputs = [ img2img_prompt ] ,
)
2022-11-26 16:10:46 +03:00
img2img_deepbooru . click (
fn = interrogate_deepbooru ,
inputs = [ init_img ] ,
outputs = [ img2img_prompt ] ,
2022-09-14 17:56:21 +03:00
)
prompts = [ ( txt2img_prompt , txt2img_negative_prompt ) , ( img2img_prompt , img2img_negative_prompt ) ]
style_dropdowns = [ ( txt2img_prompt_style , txt2img_prompt_style2 ) , ( img2img_prompt_style , img2img_prompt_style2 ) ]
2022-09-30 19:12:44 +03:00
style_js_funcs = [ " update_txt2img_tokens " , " update_img2img_tokens " ]
2022-09-14 17:56:21 +03:00
for button , ( prompt , negative_prompt ) in zip ( [ txt2img_save_style , img2img_save_style ] , prompts ) :
2022-09-09 23:16:02 +03:00
button . click (
fn = add_style ,
_js = " ask_for_style_name " ,
2022-09-11 17:35:12 +03:00
# Have to pass empty dummy component here, because the JavaScript and Python function have to accept
# the same number of parameters, but we only know the style-name after the JavaScript prompt
inputs = [ dummy_component , prompt , negative_prompt ] ,
2022-09-14 17:56:21 +03:00
outputs = [ txt2img_prompt_style , img2img_prompt_style , txt2img_prompt_style2 , img2img_prompt_style2 ] ,
)
2022-09-29 21:40:47 +03:00
for button , ( prompt , negative_prompt ) , ( style1 , style2 ) , js_func in zip ( [ txt2img_prompt_style_apply , img2img_prompt_style_apply ] , prompts , style_dropdowns , style_js_funcs ) :
2022-09-14 17:56:21 +03:00
button . click (
fn = apply_styles ,
2022-09-29 21:40:47 +03:00
_js = js_func ,
2022-09-14 17:56:21 +03:00
inputs = [ prompt , negative_prompt , style1 , style2 ] ,
outputs = [ prompt , negative_prompt , style1 , style2 ] ,
2022-09-09 23:16:02 +03:00
)
2022-10-27 08:36:11 +03:00
token_button . click ( fn = update_token_counter , inputs = [ img2img_prompt , steps ] , outputs = [ token_counter ] )
2022-09-25 09:25:28 +03:00
img2img_paste_fields = [
( img2img_prompt , " Prompt " ) ,
( img2img_negative_prompt , " Negative prompt " ) ,
( steps , " Steps " ) ,
( sampler_index , " Sampler " ) ,
( restore_faces , " Face restoration " ) ,
( cfg_scale , " CFG scale " ) ,
( seed , " Seed " ) ,
( width , " Size-1 " ) ,
( height , " Size-2 " ) ,
( batch_size , " Batch size " ) ,
( subseed , " Variation seed " ) ,
( subseed_strength , " Variation seed strength " ) ,
( seed_resize_from_w , " Seed resize from-1 " ) ,
( seed_resize_from_h , " Seed resize from-2 " ) ,
( denoising_strength , " Denoising strength " ) ,
2022-11-27 16:35:35 +03:00
( mask_blur , " Mask blur " ) ,
2022-10-22 12:23:45 +03:00
* modules . scripts . scripts_img2img . infotext_fields
2022-09-25 09:25:28 +03:00
]
2022-10-29 08:28:48 +03:00
parameters_copypaste . add_paste_fields ( " img2img " , init_img , img2img_paste_fields )
2022-10-27 08:36:11 +03:00
parameters_copypaste . add_paste_fields ( " inpaint " , init_img_with_mask , img2img_paste_fields )
2022-09-23 22:49:21 +03:00
2022-11-19 19:10:17 +03:00
modules . scripts . scripts_current = None
2022-09-23 22:49:21 +03:00
2022-09-03 12:08:45 +03:00
with gr . Blocks ( analytics_enabled = False ) as extras_interface :
with gr . Row ( ) . style ( equal_height = False ) :
2023-01-14 13:38:10 +03:00
with gr . Column ( variant = ' compact ' ) :
2022-09-22 12:11:48 +03:00
with gr . Tabs ( elem_id = " mode_extras " ) :
2023-01-01 16:51:12 +03:00
with gr . TabItem ( ' Single Image ' , elem_id = " extras_single_tab " ) :
extras_image = gr . Image ( label = " Source " , source = " upload " , interactive = True , type = " pil " , elem_id = " extras_image " )
2022-09-16 06:23:37 +03:00
2023-01-01 16:51:12 +03:00
with gr . TabItem ( ' Batch Process ' , elem_id = " extras_batch_process_tab " ) :
image_batch = gr . File ( label = " Batch Process " , file_count = " multiple " , interactive = True , type = " file " , elem_id = " extras_image_batch " )
2022-09-04 18:54:12 +03:00
2023-01-01 16:51:12 +03:00
with gr . TabItem ( ' Batch from Directory ' , elem_id = " extras_batch_directory_tab " ) :
extras_batch_input_dir = gr . Textbox ( label = " Input directory " , * * shared . hide_dirs , placeholder = " A directory on the same machine where the server is running. " , elem_id = " extras_batch_input_dir " )
extras_batch_output_dir = gr . Textbox ( label = " Output directory " , * * shared . hide_dirs , placeholder = " Leave blank to save images to the default path. " , elem_id = " extras_batch_output_dir " )
show_extras_results = gr . Checkbox ( label = ' Show result images ' , value = True , elem_id = " extras_show_extras_results " )
2022-10-16 07:50:55 +03:00
2022-11-04 08:28:22 +03:00
submit = gr . Button ( ' Generate ' , elem_id = " extras_generate " , variant = ' primary ' )
2022-10-10 04:26:52 +03:00
with gr . Tabs ( elem_id = " extras_resize_mode " ) :
2023-01-01 16:51:12 +03:00
with gr . TabItem ( ' Scale by ' , elem_id = " extras_scale_by_tab " ) :
upscaling_resize = gr . Slider ( minimum = 1.0 , maximum = 8.0 , step = 0.05 , label = " Resize " , value = 4 , elem_id = " extras_upscaling_resize " )
with gr . TabItem ( ' Scale to ' , elem_id = " extras_scale_to_tab " ) :
2022-10-10 04:26:52 +03:00
with gr . Group ( ) :
with gr . Row ( ) :
2023-01-01 16:51:12 +03:00
upscaling_resize_w = gr . Number ( label = " Width " , value = 512 , precision = 0 , elem_id = " extras_upscaling_resize_w " )
upscaling_resize_h = gr . Number ( label = " Height " , value = 512 , precision = 0 , elem_id = " extras_upscaling_resize_h " )
upscaling_crop = gr . Checkbox ( label = ' Crop to fit ' , value = True , elem_id = " extras_upscaling_crop " )
2022-10-29 08:28:48 +03:00
2022-09-04 18:54:12 +03:00
with gr . Group ( ) :
2022-10-29 08:28:48 +03:00
extras_upscaler_1 = gr . Radio ( label = ' Upscaler 1 ' , elem_id = " extras_upscaler_1 " , choices = [ x . name for x in shared . sd_upscalers ] , value = shared . sd_upscalers [ 0 ] . name , type = " index " )
2022-09-04 18:54:12 +03:00
with gr . Group ( ) :
2022-10-18 08:15:38 +03:00
extras_upscaler_2 = gr . Radio ( label = ' Upscaler 2 ' , elem_id = " extras_upscaler_2 " , choices = [ x . name for x in shared . sd_upscalers ] , value = shared . sd_upscalers [ 0 ] . name , type = " index " )
2023-01-01 16:51:12 +03:00
extras_upscaler_2_visibility = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.001 , label = " Upscaler 2 visibility " , value = 1 , elem_id = " extras_upscaler_2_visibility " )
2022-09-04 18:54:12 +03:00
with gr . Group ( ) :
2023-01-01 16:51:12 +03:00
gfpgan_visibility = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.001 , label = " GFPGAN visibility " , value = 0 , interactive = modules . gfpgan_model . have_gfpgan , elem_id = " extras_gfpgan_visibility " )
2022-09-07 13:35:02 +03:00
with gr . Group ( ) :
2023-01-01 16:51:12 +03:00
codeformer_visibility = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.001 , label = " CodeFormer visibility " , value = 0 , interactive = modules . codeformer_model . have_codeformer , elem_id = " extras_codeformer_visibility " )
codeformer_weight = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.001 , label = " CodeFormer weight (0 = maximum effect, 1 = minimum effect) " , value = 0 , interactive = modules . codeformer_model . have_codeformer , elem_id = " extras_codeformer_weight " )
2022-09-03 12:08:45 +03:00
2022-10-28 21:24:11 +03:00
with gr . Group ( ) :
2023-01-01 16:51:12 +03:00
upscale_before_face_fix = gr . Checkbox ( label = ' Upscale Before Restoring Faces ' , value = False , elem_id = " extras_upscale_before_face_fix " )
2022-09-03 12:08:45 +03:00
2022-12-31 23:40:55 +03:00
result_images , html_info_x , html_info , html_log = create_output_panel ( " extras " , opts . outdir_extras_samples )
2022-10-13 02:17:26 +03:00
2022-09-22 12:11:48 +03:00
submit . click (
2022-12-31 23:40:55 +03:00
fn = wrap_gradio_gpu_call ( modules . extras . run_extras , extra_outputs = [ None , ' ' ] ) ,
2022-09-22 12:11:48 +03:00
_js = " get_extras_tab_index " ,
2022-09-03 12:08:45 +03:00
inputs = [
2022-10-10 04:26:52 +03:00
dummy_component ,
2022-09-22 12:11:48 +03:00
dummy_component ,
2022-09-23 22:49:21 +03:00
extras_image ,
2022-09-16 06:23:37 +03:00
image_batch ,
2022-10-16 07:50:55 +03:00
extras_batch_input_dir ,
extras_batch_output_dir ,
show_extras_results ,
2022-09-07 13:35:02 +03:00
gfpgan_visibility ,
codeformer_visibility ,
codeformer_weight ,
2022-09-04 18:54:12 +03:00
upscaling_resize ,
2022-10-10 04:26:52 +03:00
upscaling_resize_w ,
upscaling_resize_h ,
upscaling_crop ,
2022-09-04 18:54:12 +03:00
extras_upscaler_1 ,
extras_upscaler_2 ,
extras_upscaler_2_visibility ,
2022-10-28 21:24:11 +03:00
upscale_before_face_fix ,
2022-09-03 12:08:45 +03:00
] ,
outputs = [
2022-09-16 06:23:37 +03:00
result_images ,
2022-09-03 12:08:45 +03:00
html_info_x ,
html_info ,
]
)
2022-10-29 08:28:48 +03:00
parameters_copypaste . add_paste_fields ( " extras " , extras_image , None )
2022-10-10 04:26:52 +03:00
2022-10-29 00:11:16 +03:00
extras_image . change (
fn = modules . extras . clear_cache ,
inputs = [ ] , outputs = [ ]
2022-09-23 14:17:23 +03:00
)
2022-09-03 12:08:45 +03:00
2022-09-23 22:49:21 +03:00
with gr . Blocks ( analytics_enabled = False ) as pnginfo_interface :
with gr . Row ( ) . style ( equal_height = False ) :
with gr . Column ( variant = ' panel ' ) :
image = gr . Image ( elem_id = " pnginfo_image " , label = " Source " , source = " upload " , interactive = True , type = " pil " )
with gr . Column ( variant = ' panel ' ) :
html = gr . HTML ( )
2023-01-01 16:51:12 +03:00
generation_info = gr . Textbox ( visible = False , elem_id = " pnginfo_generation_info " )
2022-09-23 22:49:21 +03:00
html2 = gr . HTML ( )
with gr . Row ( ) :
2022-10-27 08:36:11 +03:00
buttons = parameters_copypaste . create_buttons ( [ " txt2img " , " img2img " , " inpaint " , " extras " ] )
2022-10-29 08:28:48 +03:00
parameters_copypaste . bind_buttons ( buttons , image , generation_info )
2022-09-23 22:49:21 +03:00
image . change (
2022-10-02 15:03:39 +03:00
fn = wrap_gradio_call ( modules . extras . run_pnginfo ) ,
2022-09-23 22:49:21 +03:00
inputs = [ image ] ,
outputs = [ html , generation_info , html2 ] ,
)
2022-10-29 08:28:48 +03:00
2022-11-06 14:39:41 +03:00
with gr . Blocks ( analytics_enabled = False ) as modelmerger_interface :
2022-09-26 02:22:12 +03:00
with gr . Row ( ) . style ( equal_height = False ) :
2023-01-14 13:38:10 +03:00
with gr . Column ( variant = ' compact ' ) :
gr . HTML ( value = " <p style= ' margin-bottom: 2.5em ' >A merger of the two checkpoints will be generated in your <b>checkpoint</b> directory.</p> " )
2022-10-02 15:03:39 +03:00
2023-01-11 09:10:07 +03:00
with FormRow ( ) :
2022-10-14 09:05:06 +03:00
primary_model_name = gr . Dropdown ( modules . sd_models . checkpoint_tiles ( ) , elem_id = " modelmerger_primary_model_name " , label = " Primary model (A) " )
2023-01-01 10:35:38 +03:00
create_refresh_button ( primary_model_name , modules . sd_models . list_models , lambda : { " choices " : modules . sd_models . checkpoint_tiles ( ) } , " refresh_checkpoint_A " )
2022-10-14 09:05:06 +03:00
secondary_model_name = gr . Dropdown ( modules . sd_models . checkpoint_tiles ( ) , elem_id = " modelmerger_secondary_model_name " , label = " Secondary model (B) " )
2023-01-01 10:35:38 +03:00
create_refresh_button ( secondary_model_name , modules . sd_models . list_models , lambda : { " choices " : modules . sd_models . checkpoint_tiles ( ) } , " refresh_checkpoint_B " )
2022-10-14 09:05:06 +03:00
tertiary_model_name = gr . Dropdown ( modules . sd_models . checkpoint_tiles ( ) , elem_id = " modelmerger_tertiary_model_name " , label = " Tertiary model (C) " )
2023-01-01 10:35:38 +03:00
create_refresh_button ( tertiary_model_name , modules . sd_models . list_models , lambda : { " choices " : modules . sd_models . checkpoint_tiles ( ) } , " refresh_checkpoint_C " )
2023-01-01 16:51:12 +03:00
custom_name = gr . Textbox ( label = " Custom Name (Optional) " , elem_id = " modelmerger_custom_name " )
interp_amount = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.05 , label = ' Multiplier (M) - set to 0 to get model A ' , value = 0.3 , elem_id = " modelmerger_interp_amount " )
interp_method = gr . Radio ( choices = [ " Weighted sum " , " Add difference " ] , value = " Weighted sum " , label = " Interpolation Method " , elem_id = " modelmerger_interp_method " )
2022-11-27 15:51:29 +03:00
2023-01-11 09:10:07 +03:00
with FormRow ( ) :
2023-01-01 16:51:12 +03:00
checkpoint_format = gr . Radio ( choices = [ " ckpt " , " safetensors " ] , value = " ckpt " , label = " Checkpoint format " , elem_id = " modelmerger_checkpoint_format " )
save_as_half = gr . Checkbox ( value = False , label = " Save as float16 " , elem_id = " modelmerger_save_as_half " )
2022-11-27 15:51:29 +03:00
2023-01-11 09:10:07 +03:00
config_source = gr . Radio ( choices = [ " A, B or C " , " B " , " C " , " Don ' t " ] , value = " A, B or C " , label = " Copy config from " , type = " index " , elem_id = " modelmerger_config_method " )
2023-01-14 13:38:10 +03:00
with gr . Row ( ) :
modelmerger_merge = gr . Button ( elem_id = " modelmerger_merge " , value = " Merge " , variant = ' primary ' )
2022-10-02 15:03:39 +03:00
2022-09-26 02:22:12 +03:00
with gr . Column ( variant = ' panel ' ) :
2022-09-27 10:44:00 +03:00
submit_result = gr . Textbox ( elem_id = " modelmerger_result " , show_label = False )
2022-09-26 02:22:12 +03:00
2022-11-06 14:39:41 +03:00
with gr . Blocks ( analytics_enabled = False ) as train_interface :
2022-10-02 15:03:39 +03:00
with gr . Row ( ) . style ( equal_height = False ) :
2022-10-12 11:05:57 +03:00
gr . HTML ( value = " <p style= ' margin-bottom: 0.7em ' >See <b><a href= \" https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Textual-Inversion \" >wiki</a></b> for detailed explanation.</p> " )
2022-10-02 22:41:21 +03:00
2022-10-12 11:05:57 +03:00
with gr . Row ( ) . style ( equal_height = False ) :
with gr . Tabs ( elem_id = " train_tabs " ) :
2022-10-02 15:03:39 +03:00
2022-10-12 11:05:57 +03:00
with gr . Tab ( label = " Create embedding " ) :
2023-01-01 16:51:12 +03:00
new_embedding_name = gr . Textbox ( label = " Name " , elem_id = " train_new_embedding_name " )
initialization_text = gr . Textbox ( label = " Initialization text " , value = " * " , elem_id = " train_initialization_text " )
nvpt = gr . Slider ( label = " Number of vectors per token " , minimum = 1 , maximum = 75 , step = 1 , value = 1 , elem_id = " train_nvpt " )
overwrite_old_embedding = gr . Checkbox ( value = False , label = " Overwrite Old Embedding " , elem_id = " train_overwrite_old_embedding " )
2022-10-02 15:03:39 +03:00
with gr . Row ( ) :
with gr . Column ( scale = 3 ) :
gr . HTML ( value = " " )
with gr . Column ( ) :
2023-01-01 16:51:12 +03:00
create_embedding = gr . Button ( value = " Create embedding " , variant = ' primary ' , elem_id = " train_create_embedding " )
2022-10-02 15:03:39 +03:00
2022-10-12 11:05:57 +03:00
with gr . Tab ( label = " Create hypernetwork " ) :
2023-01-01 16:51:12 +03:00
new_hypernetwork_name = gr . Textbox ( label = " Name " , elem_id = " train_new_hypernetwork_name " )
new_hypernetwork_sizes = gr . CheckboxGroup ( label = " Modules " , value = [ " 768 " , " 320 " , " 640 " , " 1280 " ] , choices = [ " 768 " , " 1024 " , " 320 " , " 640 " , " 1280 " ] , elem_id = " train_new_hypernetwork_sizes " )
new_hypernetwork_layer_structure = gr . Textbox ( " 1, 2, 1 " , label = " Enter hypernetwork layer structure " , placeholder = " 1st and last digit must be 1. ex: ' 1, 2, 1 ' " , elem_id = " train_new_hypernetwork_layer_structure " )
new_hypernetwork_activation_func = gr . Dropdown ( value = " linear " , label = " Select activation function of hypernetwork. Recommended : Swish / Linear(none) " , choices = modules . hypernetworks . ui . keys , elem_id = " train_new_hypernetwork_activation_func " )
new_hypernetwork_initialization_option = gr . Dropdown ( value = " Normal " , label = " Select Layer weights initialization. Recommended: Kaiming for relu-like, Xavier for sigmoid-like, Normal otherwise " , choices = [ " Normal " , " KaimingUniform " , " KaimingNormal " , " XavierUniform " , " XavierNormal " ] , elem_id = " train_new_hypernetwork_initialization_option " )
new_hypernetwork_add_layer_norm = gr . Checkbox ( label = " Add layer normalization " , elem_id = " train_new_hypernetwork_add_layer_norm " )
new_hypernetwork_use_dropout = gr . Checkbox ( label = " Use dropout " , elem_id = " train_new_hypernetwork_use_dropout " )
2023-01-10 08:56:57 +03:00
new_hypernetwork_dropout_structure = gr . Textbox ( " 0, 0, 0 " , label = " Enter hypernetwork Dropout structure (or empty). Recommended : 0~0.35 incrementing sequence: 0, 0.05, 0.15 " , placeholder = " 1st and last digit must be 0 and values should be between 0 and 1. ex: ' 0, 0.01, 0 ' " )
2023-01-01 16:51:12 +03:00
overwrite_old_hypernetwork = gr . Checkbox ( value = False , label = " Overwrite Old Hypernetwork " , elem_id = " train_overwrite_old_hypernetwork " )
2022-10-07 23:22:22 +03:00
with gr . Row ( ) :
with gr . Column ( scale = 3 ) :
gr . HTML ( value = " " )
with gr . Column ( ) :
2023-01-01 16:51:12 +03:00
create_hypernetwork = gr . Button ( value = " Create hypernetwork " , variant = ' primary ' , elem_id = " train_create_hypernetwork " )
2022-10-02 15:03:39 +03:00
2022-10-12 11:05:57 +03:00
with gr . Tab ( label = " Preprocess images " ) :
2023-01-01 16:51:12 +03:00
process_src = gr . Textbox ( label = ' Source directory ' , elem_id = " train_process_src " )
process_dst = gr . Textbox ( label = ' Destination directory ' , elem_id = " train_process_dst " )
process_width = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Width " , value = 512 , elem_id = " train_process_width " )
process_height = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Height " , value = 512 , elem_id = " train_process_height " )
preprocess_txt_action = gr . Dropdown ( label = ' Existing Caption txt Action ' , value = " ignore " , choices = [ " ignore " , " copy " , " prepend " , " append " ] , elem_id = " train_preprocess_txt_action " )
2022-10-02 22:41:21 +03:00
with gr . Row ( ) :
2023-01-01 16:51:12 +03:00
process_flip = gr . Checkbox ( label = ' Create flipped copies ' , elem_id = " train_process_flip " )
process_split = gr . Checkbox ( label = ' Split oversized images ' , elem_id = " train_process_split " )
process_focal_crop = gr . Checkbox ( label = ' Auto focal point crop ' , elem_id = " train_process_focal_crop " )
process_caption = gr . Checkbox ( label = ' Use BLIP for caption ' , elem_id = " train_process_caption " )
process_caption_deepbooru = gr . Checkbox ( label = ' Use deepbooru for caption ' , visible = True , elem_id = " train_process_caption_deepbooru " )
2022-10-02 22:41:21 +03:00
2022-10-20 16:56:45 +03:00
with gr . Row ( visible = False ) as process_split_extra_row :
2023-01-01 16:51:12 +03:00
process_split_threshold = gr . Slider ( label = ' Split image threshold ' , value = 0.5 , minimum = 0.0 , maximum = 1.0 , step = 0.05 , elem_id = " train_process_split_threshold " )
process_overlap_ratio = gr . Slider ( label = ' Split image overlap ratio ' , value = 0.2 , minimum = 0.0 , maximum = 0.9 , step = 0.05 , elem_id = " train_process_overlap_ratio " )
2022-10-20 16:56:45 +03:00
2022-10-26 01:22:29 +03:00
with gr . Row ( visible = False ) as process_focal_crop_row :
2023-01-01 16:51:12 +03:00
process_focal_crop_face_weight = gr . Slider ( label = ' Focal point face weight ' , value = 0.9 , minimum = 0.0 , maximum = 1.0 , step = 0.05 , elem_id = " train_process_focal_crop_face_weight " )
process_focal_crop_entropy_weight = gr . Slider ( label = ' Focal point entropy weight ' , value = 0.15 , minimum = 0.0 , maximum = 1.0 , step = 0.05 , elem_id = " train_process_focal_crop_entropy_weight " )
process_focal_crop_edges_weight = gr . Slider ( label = ' Focal point edges weight ' , value = 0.5 , minimum = 0.0 , maximum = 1.0 , step = 0.05 , elem_id = " train_process_focal_crop_edges_weight " )
process_focal_crop_debug = gr . Checkbox ( label = ' Create debug image ' , elem_id = " train_process_focal_crop_debug " )
2022-10-26 01:22:29 +03:00
2022-10-02 22:41:21 +03:00
with gr . Row ( ) :
with gr . Column ( scale = 3 ) :
gr . HTML ( value = " " )
with gr . Column ( ) :
2022-11-18 05:03:57 +03:00
with gr . Row ( ) :
2023-01-01 16:51:12 +03:00
interrupt_preprocessing = gr . Button ( " Interrupt " , elem_id = " train_interrupt_preprocessing " )
run_preprocess = gr . Button ( value = " Preprocess " , variant = ' primary ' , elem_id = " train_run_preprocess " )
2022-10-02 22:41:21 +03:00
2022-10-20 16:56:45 +03:00
process_split . change (
fn = lambda show : gr_show ( show ) ,
inputs = [ process_split ] ,
outputs = [ process_split_extra_row ] ,
)
2022-10-26 01:22:29 +03:00
process_focal_crop . change (
fn = lambda show : gr_show ( show ) ,
inputs = [ process_focal_crop ] ,
outputs = [ process_focal_crop_row ] ,
)
2023-01-09 23:35:40 +03:00
def get_textual_inversion_template_names ( ) :
return sorted ( [ x for x in textual_inversion . textual_inversion_templates ] )
2022-10-12 11:05:57 +03:00
with gr . Tab ( label = " Train " ) :
2022-10-19 22:33:18 +03:00
gr . HTML ( value = " <p style= ' margin-bottom: 0.7em ' >Train an embedding or Hypernetwork; you must specify a directory with a set of 1:1 ratio images <a href= \" https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Textual-Inversion \" style= \" font-weight:bold; \" >[wiki]</a></p> " )
2023-01-04 20:10:40 +03:00
with FormRow ( ) :
2022-10-17 21:15:32 +03:00
train_embedding_name = gr . Dropdown ( label = ' Embedding ' , elem_id = " train_embedding " , choices = sorted ( sd_hijack . model_hijack . embedding_db . word_embeddings . keys ( ) ) )
2022-10-16 07:42:52 +03:00
create_refresh_button ( train_embedding_name , sd_hijack . model_hijack . embedding_db . load_textual_inversion_embeddings , lambda : { " choices " : sorted ( sd_hijack . model_hijack . embedding_db . word_embeddings . keys ( ) ) } , " refresh_train_embedding_name " )
2023-01-04 20:10:40 +03:00
2022-10-17 21:15:32 +03:00
train_hypernetwork_name = gr . Dropdown ( label = ' Hypernetwork ' , elem_id = " train_hypernetwork " , choices = [ x for x in shared . hypernetworks . keys ( ) ] )
2022-10-16 07:42:52 +03:00
create_refresh_button ( train_hypernetwork_name , shared . reload_hypernetworks , lambda : { " choices " : sorted ( [ x for x in shared . hypernetworks . keys ( ) ] ) } , " refresh_train_hypernetwork_name " )
2023-01-04 20:10:40 +03:00
with FormRow ( ) :
2023-01-01 16:51:12 +03:00
embedding_learn_rate = gr . Textbox ( label = ' Embedding Learning rate ' , placeholder = " Embedding Learning rate " , value = " 0.005 " , elem_id = " train_embedding_learn_rate " )
hypernetwork_learn_rate = gr . Textbox ( label = ' Hypernetwork Learning rate ' , placeholder = " Hypernetwork Learning rate " , value = " 0.00001 " , elem_id = " train_hypernetwork_learn_rate " )
2023-01-04 19:56:35 +03:00
2023-01-04 20:10:40 +03:00
with FormRow ( ) :
2022-10-28 13:16:23 +03:00
clip_grad_mode = gr . Dropdown ( value = " disabled " , label = " Gradient Clipping " , choices = [ " disabled " , " value " , " norm " ] )
2022-10-31 09:49:24 +03:00
clip_grad_value = gr . Textbox ( placeholder = " Gradient clip value " , value = " 0.1 " , show_label = False )
2023-01-01 16:51:12 +03:00
2023-01-04 20:10:40 +03:00
with FormRow ( ) :
batch_size = gr . Number ( label = ' Batch size ' , value = 1 , precision = 0 , elem_id = " train_batch_size " )
gradient_step = gr . Number ( label = ' Gradient accumulation steps ' , value = 1 , precision = 0 , elem_id = " train_gradient_step " )
2023-01-01 16:51:12 +03:00
dataset_directory = gr . Textbox ( label = ' Dataset directory ' , placeholder = " Path to directory with input images " , elem_id = " train_dataset_directory " )
log_directory = gr . Textbox ( label = ' Log directory ' , placeholder = " Path to directory where to write outputs " , value = " textual_inversion " , elem_id = " train_log_directory " )
2023-01-09 23:35:40 +03:00
with FormRow ( ) :
template_file = gr . Dropdown ( label = ' Prompt template ' , value = " style_filewords.txt " , elem_id = " train_template_file " , choices = get_textual_inversion_template_names ( ) )
create_refresh_button ( template_file , textual_inversion . list_textual_inversion_templates , lambda : { " choices " : get_textual_inversion_template_names ( ) } , " refrsh_train_template_file " )
2023-01-01 16:51:12 +03:00
training_width = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Width " , value = 512 , elem_id = " train_training_width " )
training_height = gr . Slider ( minimum = 64 , maximum = 2048 , step = 8 , label = " Height " , value = 512 , elem_id = " train_training_height " )
2023-01-09 22:52:23 +03:00
varsize = gr . Checkbox ( label = " Do not resize images " , value = False , elem_id = " train_varsize " )
2023-01-01 16:51:12 +03:00
steps = gr . Number ( label = ' Max steps ' , value = 100000 , precision = 0 , elem_id = " train_steps " )
2023-01-04 20:10:40 +03:00
with FormRow ( ) :
create_image_every = gr . Number ( label = ' Save an image to log directory every N steps, 0 to disable ' , value = 500 , precision = 0 , elem_id = " train_create_image_every " )
save_embedding_every = gr . Number ( label = ' Save a copy of embedding to log directory every N steps, 0 to disable ' , value = 500 , precision = 0 , elem_id = " train_save_embedding_every " )
2023-01-01 16:51:12 +03:00
save_image_with_stored_embedding = gr . Checkbox ( label = ' Save images with embedding in PNG chunks ' , value = True , elem_id = " train_save_image_with_stored_embedding " )
preview_from_txt2img = gr . Checkbox ( label = ' Read parameters (prompt, etc...) from txt2img tab when making previews ' , value = False , elem_id = " train_preview_from_txt2img " )
2023-01-04 20:10:40 +03:00
shuffle_tags = gr . Checkbox ( label = " Shuffle tags by ' , ' when creating prompts. " , value = False , elem_id = " train_shuffle_tags " )
tag_drop_out = gr . Slider ( minimum = 0 , maximum = 1 , step = 0.1 , label = " Drop out tags when creating prompts. " , value = 0 , elem_id = " train_tag_drop_out " )
latent_sampling_method = gr . Radio ( label = ' Choose latent sampling method ' , value = " once " , choices = [ ' once ' , ' deterministic ' , ' random ' ] , elem_id = " train_latent_sampling_method " )
2022-10-02 15:03:39 +03:00
with gr . Row ( ) :
2023-01-04 20:10:40 +03:00
train_embedding = gr . Button ( value = " Train Embedding " , variant = ' primary ' , elem_id = " train_train_embedding " )
2023-01-01 16:51:12 +03:00
interrupt_training = gr . Button ( value = " Interrupt " , elem_id = " train_interrupt_training " )
train_hypernetwork = gr . Button ( value = " Train Hypernetwork " , variant = ' primary ' , elem_id = " train_train_hypernetwork " )
2022-10-02 15:03:39 +03:00
2022-11-08 08:38:10 +03:00
params = script_callbacks . UiTrainTabParams ( txt2img_preview_params )
script_callbacks . ui_train_tabs_callback ( params )
2022-10-02 15:03:39 +03:00
with gr . Column ( ) :
progressbar = gr . HTML ( elem_id = " ti_progressbar " )
ti_output = gr . Text ( elem_id = " ti_output " , value = " " , show_label = False )
ti_gallery = gr . Gallery ( label = ' Output ' , show_label = False , elem_id = ' ti_gallery ' ) . style ( grid = 4 )
ti_preview = gr . Image ( elem_id = ' ti_preview ' , visible = False )
ti_progress = gr . HTML ( elem_id = " ti_progress " , value = " " )
ti_outcome = gr . HTML ( elem_id = " ti_error " , value = " " )
setup_progressbar ( progressbar , ti_preview , ' ti ' , textinfo = ti_progress )
create_embedding . click (
fn = modules . textual_inversion . ui . create_embedding ,
inputs = [
new_embedding_name ,
2022-10-02 19:40:51 +03:00
initialization_text ,
2022-10-02 15:03:39 +03:00
nvpt ,
2022-10-19 22:33:18 +03:00
overwrite_old_embedding ,
2022-10-02 15:03:39 +03:00
] ,
outputs = [
train_embedding_name ,
ti_output ,
ti_outcome ,
]
)
2022-10-07 23:22:22 +03:00
create_hypernetwork . click (
2022-10-11 15:54:34 +03:00
fn = modules . hypernetworks . ui . create_hypernetwork ,
2022-10-07 23:22:22 +03:00
inputs = [
new_hypernetwork_name ,
2022-10-11 18:04:47 +03:00
new_hypernetwork_sizes ,
2022-10-20 02:27:16 +03:00
overwrite_old_hypernetwork ,
2022-10-19 17:30:33 +03:00
new_hypernetwork_layer_structure ,
2022-10-20 03:10:45 +03:00
new_hypernetwork_activation_func ,
2022-10-25 08:48:49 +03:00
new_hypernetwork_initialization_option ,
2022-10-19 17:30:33 +03:00
new_hypernetwork_add_layer_norm ,
2023-01-10 08:56:57 +03:00
new_hypernetwork_use_dropout ,
new_hypernetwork_dropout_structure
2022-10-07 23:22:22 +03:00
] ,
outputs = [
train_hypernetwork_name ,
ti_output ,
ti_outcome ,
]
)
2022-10-02 22:41:21 +03:00
run_preprocess . click (
fn = wrap_gradio_gpu_call ( modules . textual_inversion . ui . preprocess , extra_outputs = [ gr . update ( ) ] ) ,
_js = " start_training_textual_inversion " ,
inputs = [
process_src ,
process_dst ,
2022-10-10 16:35:35 +03:00
process_width ,
process_height ,
2022-10-20 02:48:07 +03:00
preprocess_txt_action ,
2022-10-02 22:41:21 +03:00
process_flip ,
process_split ,
process_caption ,
2022-10-20 16:56:45 +03:00
process_caption_deepbooru ,
process_split_threshold ,
process_overlap_ratio ,
2022-10-26 01:22:29 +03:00
process_focal_crop ,
process_focal_crop_face_weight ,
process_focal_crop_entropy_weight ,
process_focal_crop_edges_weight ,
process_focal_crop_debug ,
2022-10-02 22:41:21 +03:00
] ,
outputs = [
ti_output ,
ti_outcome ,
] ,
)
2022-10-02 15:03:39 +03:00
train_embedding . click (
fn = wrap_gradio_gpu_call ( modules . textual_inversion . ui . train_embedding , extra_outputs = [ gr . update ( ) ] ) ,
_js = " start_training_textual_inversion " ,
inputs = [
train_embedding_name ,
2022-10-20 02:19:40 +03:00
embedding_learn_rate ,
2022-10-15 09:24:59 +03:00
batch_size ,
2022-11-20 06:35:26 +03:00
gradient_step ,
2022-10-02 15:03:39 +03:00
dataset_directory ,
log_directory ,
2022-10-10 16:35:35 +03:00
training_width ,
training_height ,
2023-01-07 20:34:52 +03:00
varsize ,
2022-10-02 15:03:39 +03:00
steps ,
2022-10-28 06:31:27 +03:00
clip_grad_mode ,
clip_grad_value ,
2022-11-20 06:35:26 +03:00
shuffle_tags ,
tag_drop_out ,
latent_sampling_method ,
2022-10-02 15:03:39 +03:00
create_image_every ,
save_embedding_every ,
template_file ,
2022-10-09 07:40:57 +03:00
save_image_with_stored_embedding ,
2022-10-14 20:31:49 +03:00
preview_from_txt2img ,
* txt2img_preview_params ,
2022-10-02 15:03:39 +03:00
] ,
outputs = [
ti_output ,
ti_outcome ,
]
)
2022-10-07 23:22:22 +03:00
train_hypernetwork . click (
2022-10-11 15:54:34 +03:00
fn = wrap_gradio_gpu_call ( modules . hypernetworks . ui . train_hypernetwork , extra_outputs = [ gr . update ( ) ] ) ,
2022-10-07 23:22:22 +03:00
_js = " start_training_textual_inversion " ,
inputs = [
train_hypernetwork_name ,
2022-10-20 02:19:40 +03:00
hypernetwork_learn_rate ,
2022-10-15 09:24:59 +03:00
batch_size ,
2022-11-20 06:35:26 +03:00
gradient_step ,
2022-10-07 23:22:22 +03:00
dataset_directory ,
log_directory ,
2022-10-19 08:44:33 +03:00
training_width ,
training_height ,
2023-01-07 20:34:52 +03:00
varsize ,
2022-10-02 15:03:39 +03:00
steps ,
2022-10-28 05:44:56 +03:00
clip_grad_mode ,
clip_grad_value ,
2022-11-20 06:35:26 +03:00
shuffle_tags ,
tag_drop_out ,
latent_sampling_method ,
2022-10-02 15:03:39 +03:00
create_image_every ,
save_embedding_every ,
template_file ,
2022-10-14 20:31:49 +03:00
preview_from_txt2img ,
* txt2img_preview_params ,
2022-10-02 15:03:39 +03:00
] ,
outputs = [
ti_output ,
ti_outcome ,
]
)
interrupt_training . click (
fn = lambda : shared . state . interrupt ( ) ,
inputs = [ ] ,
outputs = [ ] ,
)
2022-11-18 05:03:57 +03:00
interrupt_preprocessing . click (
fn = lambda : shared . state . interrupt ( ) ,
inputs = [ ] ,
outputs = [ ] ,
)
2022-10-13 19:22:41 +03:00
def create_setting_component ( key , is_quicksettings = False ) :
2022-09-03 12:08:45 +03:00
def fun ( ) :
return opts . data [ key ] if key in opts . data else opts . data_labels [ key ] . default
info = opts . data_labels [ key ]
t = type ( info . default )
2022-09-11 23:00:42 +03:00
args = info . component_args ( ) if callable ( info . component_args ) else info . component_args
2022-09-03 12:08:45 +03:00
if info . component is not None :
2022-09-11 23:00:42 +03:00
comp = info . component
2022-09-03 12:08:45 +03:00
elif t == str :
2022-09-11 23:00:42 +03:00
comp = gr . Textbox
2022-09-03 12:08:45 +03:00
elif t == int :
2022-09-11 23:00:42 +03:00
comp = gr . Number
2022-09-03 12:08:45 +03:00
elif t == bool :
2022-09-11 23:00:42 +03:00
comp = gr . Checkbox
2022-09-03 12:08:45 +03:00
else :
raise Exception ( f ' bad options item type: { str ( t ) } for key { key } ' )
2022-10-17 21:15:32 +03:00
elem_id = " setting_ " + key
2022-10-13 19:22:41 +03:00
if info . refresh is not None :
if is_quicksettings :
2022-11-06 14:39:41 +03:00
res = comp ( label = info . label , value = fun ( ) , elem_id = elem_id , * * ( args or { } ) )
2022-10-17 21:15:32 +03:00
create_refresh_button ( res , info . refresh , info . component_args , " refresh_ " + key )
2022-10-13 19:22:41 +03:00
else :
2023-01-03 09:04:29 +03:00
with FormRow ( ) :
2022-11-06 14:39:41 +03:00
res = comp ( label = info . label , value = fun ( ) , elem_id = elem_id , * * ( args or { } ) )
2022-10-17 21:15:32 +03:00
create_refresh_button ( res , info . refresh , info . component_args , " refresh_ " + key )
2022-10-13 19:22:41 +03:00
else :
2022-11-06 14:39:41 +03:00
res = comp ( label = info . label , value = fun ( ) , elem_id = elem_id , * * ( args or { } ) )
2022-10-13 19:22:41 +03:00
return res
2022-09-03 12:08:45 +03:00
2022-09-10 11:10:00 +03:00
components = [ ]
2022-09-29 00:59:44 +03:00
component_dict = { }
2022-09-10 11:10:00 +03:00
2022-10-22 19:18:56 +03:00
script_callbacks . ui_settings_callback ( )
opts . reorder ( )
2022-09-03 12:08:45 +03:00
def run_settings ( * args ) :
2022-11-06 10:12:53 +03:00
changed = [ ]
2022-09-23 17:27:30 +03:00
for key , value , comp in zip ( opts . data_labels . keys ( ) , args , components ) :
2022-11-04 10:35:30 +03:00
assert comp == dummy_component or opts . same_type ( value , opts . data_labels [ key ] . default ) , f " Bad value for setting { key } : { value } ; expecting { type ( opts . data_labels [ key ] . default ) . __name__ } "
2022-09-03 12:08:45 +03:00
2022-09-10 11:10:00 +03:00
for key , value , comp in zip ( opts . data_labels . keys ( ) , args , components ) :
2022-10-09 22:24:07 +03:00
if comp == dummy_component :
continue
2022-11-19 15:15:24 +03:00
if opts . set ( key , value ) :
2022-11-06 10:12:53 +03:00
changed . append ( key )
2022-09-11 23:00:42 +03:00
2022-11-04 22:24:42 +03:00
try :
opts . save ( shared . config_filename )
except RuntimeError :
2022-11-06 10:12:53 +03:00
return opts . dumpjson ( ) , f ' { len ( changed ) } settings changed without save: { " , " . join ( changed ) } . '
2022-12-09 09:47:45 +03:00
return opts . dumpjson ( ) , f ' { len ( changed ) } settings changed { " : " if len ( changed ) > 0 else " " } { " , " . join ( changed ) } . '
2022-09-03 12:08:45 +03:00
2022-10-09 22:24:07 +03:00
def run_settings_single ( value , key ) :
if not opts . same_type ( value , opts . data_labels [ key ] . default ) :
return gr . update ( visible = True ) , opts . dumpjson ( )
2022-11-19 15:15:24 +03:00
if not opts . set ( key , value ) :
return gr . update ( value = getattr ( opts , key ) ) , opts . dumpjson ( )
2022-10-09 22:24:07 +03:00
opts . save ( shared . config_filename )
return gr . update ( value = value ) , opts . dumpjson ( )
2022-09-10 11:10:00 +03:00
with gr . Blocks ( analytics_enabled = False ) as settings_interface :
2023-01-03 07:20:20 +03:00
with gr . Row ( ) :
2023-01-03 20:23:17 +03:00
with gr . Column ( scale = 6 ) :
settings_submit = gr . Button ( value = " Apply settings " , variant = ' primary ' , elem_id = " settings_submit " )
with gr . Column ( ) :
restart_gradio = gr . Button ( value = ' Reload UI ' , variant = ' primary ' , elem_id = " settings_restart_gradio " )
2022-09-10 11:10:00 +03:00
2023-01-03 07:20:20 +03:00
result = gr . HTML ( elem_id = " settings_result " )
2022-09-10 11:10:00 +03:00
2022-10-13 16:07:18 +03:00
quicksettings_names = [ x . strip ( ) for x in opts . quicksettings . split ( " , " ) ]
2023-01-03 09:13:35 +03:00
quicksettings_names = { x : i for i , x in enumerate ( quicksettings_names ) if x != ' quicksettings ' }
2022-10-13 16:07:18 +03:00
2022-10-09 22:24:07 +03:00
quicksettings_list = [ ]
2022-09-22 21:32:44 +03:00
previous_section = None
2023-01-03 07:20:20 +03:00
current_tab = None
with gr . Tabs ( elem_id = " settings " ) :
2022-09-22 21:32:44 +03:00
for i , ( k , item ) in enumerate ( opts . data_labels . items ( ) ) :
2022-10-31 17:36:45 +03:00
section_must_be_skipped = item . section [ 0 ] is None
2022-09-22 19:26:26 +03:00
2022-10-31 17:36:45 +03:00
if previous_section != item . section and not section_must_be_skipped :
2023-01-03 07:20:20 +03:00
elem_id , text = item . section
2022-09-22 19:26:26 +03:00
2023-01-03 07:20:20 +03:00
if current_tab is not None :
current_tab . __exit__ ( )
2022-09-10 11:10:00 +03:00
2023-01-03 07:20:20 +03:00
current_tab = gr . TabItem ( elem_id = " settings_ {} " . format ( elem_id ) , label = text )
current_tab . __enter__ ( )
2022-09-22 21:32:44 +03:00
previous_section = item . section
2022-10-22 22:05:22 +03:00
if k in quicksettings_names and not shared . cmd_opts . freeze_settings :
2022-10-09 22:24:07 +03:00
quicksettings_list . append ( ( i , k , item ) )
components . append ( dummy_component )
2022-10-31 17:36:45 +03:00
elif section_must_be_skipped :
components . append ( dummy_component )
2022-10-09 22:24:07 +03:00
else :
component = create_setting_component ( k )
component_dict [ k ] = component
components . append ( component )
2022-09-03 12:08:45 +03:00
2023-01-03 07:20:20 +03:00
if current_tab is not None :
current_tab . __exit__ ( )
2022-10-17 21:15:32 +03:00
2023-01-03 07:20:20 +03:00
with gr . TabItem ( " Actions " ) :
request_notifications = gr . Button ( value = ' Request browser notifications ' , elem_id = " request_notifications " )
download_localization = gr . Button ( value = ' Download localization template ' , elem_id = " download_localization " )
reload_script_bodies = gr . Button ( value = ' Reload custom script bodies (No ui updates, No restart) ' , variant = ' secondary ' , elem_id = " settings_reload_script_bodies " )
2022-10-13 16:07:18 +03:00
2023-01-03 20:23:17 +03:00
if os . path . exists ( " html/licenses.html " ) :
with open ( " html/licenses.html " , encoding = " utf8 " ) as file :
with gr . TabItem ( " Licenses " ) :
gr . HTML ( file . read ( ) , elem_id = " licenses " )
2023-01-03 10:01:06 +03:00
gr . Button ( value = " Show all pages " , elem_id = " settings_show_all_pages " )
2022-10-13 16:07:18 +03:00
2022-09-19 04:41:57 +03:00
request_notifications . click (
fn = lambda : None ,
inputs = [ ] ,
outputs = [ ] ,
2022-09-22 13:15:33 +03:00
_js = ' function() {} '
2022-09-19 04:41:57 +03:00
)
2022-10-17 21:15:32 +03:00
download_localization . click (
fn = lambda : None ,
inputs = [ ] ,
outputs = [ ] ,
_js = ' download_localization '
)
2022-10-02 03:19:55 +03:00
def reload_scripts ( ) :
2022-10-02 21:26:06 +03:00
modules . scripts . reload_script_body_only ( )
2022-10-22 22:05:22 +03:00
reload_javascript ( ) # need to refresh the html page
2022-10-02 03:19:55 +03:00
reload_script_bodies . click (
fn = reload_scripts ,
inputs = [ ] ,
2022-11-02 09:47:53 +03:00
outputs = [ ]
2022-10-02 03:19:55 +03:00
)
2022-10-02 03:36:30 +03:00
def request_restart ( ) :
2022-10-05 06:43:05 +03:00
shared . state . interrupt ( )
2022-10-31 17:36:45 +03:00
shared . state . need_restart = True
2022-10-02 03:36:30 +03:00
restart_gradio . click (
fn = request_restart ,
2022-11-06 09:02:25 +03:00
_js = ' restart_reload ' ,
2022-10-02 03:36:30 +03:00
inputs = [ ] ,
outputs = [ ] ,
)
2022-10-10 04:26:52 +03:00
2022-09-03 12:08:45 +03:00
interfaces = [
2022-09-10 11:10:00 +03:00
( txt2img_interface , " txt2img " , " txt2img " ) ,
( img2img_interface , " img2img " , " img2img " ) ,
( extras_interface , " Extras " , " extras " ) ,
( pnginfo_interface , " PNG Info " , " pnginfo " ) ,
2022-09-26 02:22:12 +03:00
( modelmerger_interface , " Checkpoint Merger " , " modelmerger " ) ,
2022-10-12 11:05:57 +03:00
( train_interface , " Train " , " ti " ) ,
2022-09-03 12:08:45 +03:00
]
2022-10-22 13:34:49 +03:00
css = " "
for cssfile in modules . scripts . list_files_with_name ( " style.css " ) :
2022-10-22 14:28:56 +03:00
if not os . path . isfile ( cssfile ) :
continue
2022-10-22 13:34:49 +03:00
with open ( cssfile , " r " , encoding = " utf8 " ) as file :
css + = file . read ( ) + " \n "
2022-09-03 12:08:45 +03:00
2022-09-17 16:35:58 +03:00
if os . path . exists ( os . path . join ( script_path , " user.css " ) ) :
2022-09-17 16:28:19 +03:00
with open ( os . path . join ( script_path , " user.css " ) , " r " , encoding = " utf8 " ) as file :
2022-10-22 13:34:49 +03:00
css + = file . read ( ) + " \n "
2022-09-17 16:28:19 +03:00
2022-09-03 12:08:45 +03:00
if not cmd_opts . no_progressbar_hiding :
css + = css_hide_progressbar
2022-10-29 10:56:19 +03:00
interfaces + = script_callbacks . ui_tabs_callback ( )
interfaces + = [ ( settings_interface , " Settings " , " settings " ) ]
2022-10-31 17:36:45 +03:00
extensions_interface = ui_extensions . create_ui ( )
interfaces + = [ ( extensions_interface , " Extensions " , " extensions " ) ]
2022-09-10 11:10:00 +03:00
with gr . Blocks ( css = css , analytics_enabled = False , title = " Stable Diffusion " ) as demo :
2022-10-09 22:24:07 +03:00
with gr . Row ( elem_id = " quicksettings " ) :
2023-01-03 09:13:35 +03:00
for i , k , item in sorted ( quicksettings_list , key = lambda x : quicksettings_names . get ( x [ 1 ] , x [ 0 ] ) ) :
2022-10-13 19:22:41 +03:00
component = create_setting_component ( k , is_quicksettings = True )
2022-10-09 22:24:07 +03:00
component_dict [ k ] = component
2022-10-29 10:56:19 +03:00
parameters_copypaste . integrate_settings_paste_fields ( component_dict )
parameters_copypaste . run_bind ( )
2022-10-13 20:42:27 +03:00
with gr . Tabs ( elem_id = " tabs " ) as tabs :
2022-09-10 11:10:00 +03:00
for interface , label , ifid in interfaces :
2022-10-11 08:22:46 +03:00
with gr . TabItem ( label , id = ifid , elem_id = ' tab_ ' + ifid ) :
2022-09-10 11:10:00 +03:00
interface . render ( )
2022-10-10 04:26:52 +03:00
2022-09-26 23:57:31 +03:00
if os . path . exists ( os . path . join ( script_path , " notification.mp3 " ) ) :
audio_notification = gr . Audio ( interactive = False , value = os . path . join ( script_path , " notification.mp3 " ) , elem_id = " audio_notification " , visible = False )
2022-09-10 11:10:00 +03:00
2023-01-03 20:23:17 +03:00
if os . path . exists ( " html/footer.html " ) :
with open ( " html/footer.html " , encoding = " utf8 " ) as file :
2023-01-05 11:57:01 +03:00
footer = file . read ( )
footer = footer . format ( versions = versions_html ( ) )
gr . HTML ( footer , elem_id = " footer " )
2023-01-03 20:23:17 +03:00
2022-09-19 17:16:04 +03:00
text_settings = gr . Textbox ( elem_id = " settings_json " , value = lambda : opts . dumpjson ( ) , visible = False )
2022-09-18 22:25:18 +03:00
settings_submit . click (
2022-11-04 10:35:30 +03:00
fn = wrap_gradio_call ( run_settings , extra_outputs = [ gr . update ( ) ] ) ,
2022-09-24 00:13:32 +03:00
inputs = components ,
2022-11-04 10:35:30 +03:00
outputs = [ text_settings , result ] ,
2022-09-18 22:25:18 +03:00
)
2022-10-09 22:24:07 +03:00
for i , k , item in quicksettings_list :
component = component_dict [ k ]
component . change (
fn = lambda value , k = k : run_settings_single ( value , key = k ) ,
inputs = [ component ] ,
outputs = [ component , text_settings ] ,
)
2022-11-06 14:39:41 +03:00
component_keys = [ k for k in opts . data_labels . keys ( ) if k in component_dict ]
def get_settings_values ( ) :
return [ getattr ( opts , key ) for key in component_keys ]
demo . load (
fn = get_settings_values ,
inputs = [ ] ,
outputs = [ component_dict [ k ] for k in component_keys ] ,
)
2022-09-29 02:50:34 +03:00
def modelmerger ( * args ) :
try :
2022-10-02 15:03:39 +03:00
results = modules . extras . run_modelmerger ( * args )
2022-09-29 02:50:34 +03:00
except Exception as e :
print ( " Error loading/saving model file: " , file = sys . stderr )
print ( traceback . format_exc ( ) , file = sys . stderr )
2022-10-02 15:03:39 +03:00
modules . sd_models . list_models ( ) # to remove the potentially missing models from the list
2023-01-02 20:30:02 +03:00
return [ f " Error merging checkpoints: { e } " ] + [ gr . Dropdown . update ( choices = modules . sd_models . checkpoint_tiles ( ) ) for _ in range ( 4 ) ]
2022-09-29 02:50:34 +03:00
return results
2022-09-18 22:25:18 +03:00
2022-09-29 00:59:44 +03:00
modelmerger_merge . click (
2022-09-29 02:50:34 +03:00
fn = modelmerger ,
2022-09-29 00:59:44 +03:00
inputs = [
primary_model_name ,
secondary_model_name ,
2022-10-14 09:05:06 +03:00
tertiary_model_name ,
2022-09-29 00:59:44 +03:00
interp_method ,
interp_amount ,
save_as_half ,
2022-09-29 02:50:34 +03:00
custom_name ,
2022-11-27 15:51:29 +03:00
checkpoint_format ,
2023-01-11 09:10:07 +03:00
config_source ,
2022-09-29 00:59:44 +03:00
] ,
outputs = [
submit_result ,
primary_model_name ,
secondary_model_name ,
2022-10-14 09:05:06 +03:00
tertiary_model_name ,
2022-09-29 00:59:44 +03:00
component_dict [ ' sd_model_checkpoint ' ] ,
]
)
2022-09-23 22:49:21 +03:00
2022-09-10 08:18:54 +03:00
ui_config_file = cmd_opts . ui_config_file
2022-09-04 13:52:01 +03:00
ui_settings = { }
settings_count = len ( ui_settings )
error_loading = False
try :
if os . path . exists ( ui_config_file ) :
with open ( ui_config_file , " r " , encoding = " utf8 " ) as file :
ui_settings = json . load ( file )
except Exception :
error_loading = True
print ( " Error loading settings: " , file = sys . stderr )
print ( traceback . format_exc ( ) , file = sys . stderr )
def loadsave ( path , x ) :
2022-10-18 12:51:57 +03:00
def apply_field ( obj , field , condition = None , init_field = None ) :
2022-09-04 13:52:01 +03:00
key = path + " / " + field
2022-09-25 08:56:50 +03:00
2022-10-29 10:56:19 +03:00
if getattr ( obj , ' custom_script_source ' , None ) is not None :
2022-09-25 08:56:50 +03:00
key = ' customscript/ ' + obj . custom_script_source + ' / ' + key
2022-10-10 04:26:52 +03:00
2022-09-25 19:43:42 +03:00
if getattr ( obj , ' do_not_save_to_config ' , False ) :
return
2022-10-10 04:26:52 +03:00
2022-09-04 13:52:01 +03:00
saved_value = ui_settings . get ( key , None )
if saved_value is None :
ui_settings [ key ] = getattr ( obj , field )
2022-10-15 23:09:11 +03:00
elif condition and not condition ( saved_value ) :
print ( f ' Warning: Bad ui setting value: { key } : { saved_value } ; Default value " { getattr ( obj , field ) } " will be used instead. ' )
else :
2022-09-04 13:52:01 +03:00
setattr ( obj , field , saved_value )
2022-10-18 12:51:57 +03:00
if init_field is not None :
init_field ( saved_value )
2022-09-04 13:52:01 +03:00
2023-01-06 16:03:43 +03:00
if type ( x ) in [ gr . Slider , gr . Radio , gr . Checkbox , gr . Textbox , gr . Number , gr . Dropdown ] and x . visible :
2022-09-25 19:43:42 +03:00
apply_field ( x , ' visible ' )
2022-09-04 13:52:01 +03:00
if type ( x ) == gr . Slider :
apply_field ( x , ' value ' )
apply_field ( x , ' minimum ' )
apply_field ( x , ' maximum ' )
apply_field ( x , ' step ' )
if type ( x ) == gr . Radio :
2022-09-05 19:11:29 +03:00
apply_field ( x , ' value ' , lambda val : val in x . choices )
2022-09-04 13:52:01 +03:00
2022-09-25 08:31:02 +03:00
if type ( x ) == gr . Checkbox :
2022-09-25 08:40:37 +03:00
apply_field ( x , ' value ' )
2022-09-25 08:31:02 +03:00
if type ( x ) == gr . Textbox :
2022-09-25 08:40:37 +03:00
apply_field ( x , ' value ' )
2022-10-10 04:26:52 +03:00
2022-09-25 08:39:22 +03:00
if type ( x ) == gr . Number :
2022-09-25 08:40:37 +03:00
apply_field ( x , ' value ' )
2022-10-10 04:26:52 +03:00
2023-01-06 16:03:43 +03:00
if type ( x ) == gr . Dropdown :
2022-10-18 12:51:57 +03:00
apply_field ( x , ' value ' , lambda val : val in x . choices , getattr ( x , ' init_field ' , None ) )
2022-10-15 22:47:03 +03:00
2022-09-04 13:52:01 +03:00
visit ( txt2img_interface , loadsave , " txt2img " )
visit ( img2img_interface , loadsave , " img2img " )
2022-09-11 11:31:16 +03:00
visit ( extras_interface , loadsave , " extras " )
2022-10-17 19:56:23 +03:00
visit ( modelmerger_interface , loadsave , " modelmerger " )
2023-01-04 20:10:40 +03:00
visit ( train_interface , loadsave , " train " )
2022-09-04 13:52:01 +03:00
if not error_loading and ( not os . path . exists ( ui_config_file ) or settings_count != len ( ui_settings ) ) :
with open ( ui_config_file , " w " , encoding = " utf8 " ) as file :
json . dump ( ui_settings , file , indent = 4 )
2022-09-03 12:08:45 +03:00
return demo
2022-11-08 08:35:01 +03:00
def reload_javascript ( ) :
2022-10-14 20:04:47 +03:00
with open ( os . path . join ( script_path , " script.js " ) , " r " , encoding = " utf8 " ) as jsfile :
javascript = f ' <script> { jsfile . read ( ) } </script> '
2022-09-18 08:37:03 +03:00
2022-10-22 13:34:49 +03:00
scripts_list = modules . scripts . list_scripts ( " javascript " , " .js " )
2022-10-29 08:28:48 +03:00
2022-10-22 13:34:49 +03:00
for basedir , filename , path in scripts_list :
with open ( path , " r " , encoding = " utf8 " ) as jsfile :
2022-10-14 20:04:47 +03:00
javascript + = f " \n <!-- { filename } --><script> { jsfile . read ( ) } </script> "
2022-09-03 12:08:45 +03:00
2022-10-19 09:43:49 +03:00
if cmd_opts . theme is not None :
javascript + = f " \n <script>set_theme( ' { cmd_opts . theme } ' );</script> \n "
2022-09-03 12:08:45 +03:00
2022-10-19 09:43:49 +03:00
javascript + = f " \n <script> { localization . localization_js ( shared . opts . localization ) } </script> "
2022-09-03 12:08:45 +03:00
2022-10-02 21:26:06 +03:00
def template_response ( * args , * * kwargs ) :
2022-11-08 08:35:01 +03:00
res = shared . GradioTemplateResponseOriginal ( * args , * * kwargs )
2022-10-14 20:04:47 +03:00
res . body = res . body . replace (
b ' </head> ' , f ' { javascript } </head> ' . encode ( " utf8 " ) )
2022-10-02 21:26:06 +03:00
res . init_headers ( )
return res
gradio . routes . templates . TemplateResponse = template_response
2022-10-12 19:19:34 +03:00
2022-10-14 20:04:47 +03:00
2022-11-08 08:35:01 +03:00
if not hasattr ( shared , ' GradioTemplateResponseOriginal ' ) :
shared . GradioTemplateResponseOriginal = gradio . routes . templates . TemplateResponse
2023-01-05 11:57:01 +03:00
def versions_html ( ) :
import torch
import launch
python_version = " . " . join ( [ str ( x ) for x in sys . version_info [ 0 : 3 ] ] )
commit = launch . commit_hash ( )
short_commit = commit [ 0 : 8 ]
if shared . xformers_available :
import xformers
xformers_version = xformers . __version__
else :
xformers_version = " N/A "
return f """
python : < span title = " {sys.version} " > { python_version } < / span >
•
torch : { torch . __version__ }
•
xformers : { xformers_version }
•
gradio : { gr . __version__ }
•
commit : < a href = " https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/ {commit} " > { short_commit } < / a >
"""