2022-09-03 12:08:45 +03:00
import base64
import html
import io
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-09-05 23:08:06 +03:00
import random
2022-09-03 12:08:45 +03:00
import sys
import time
import traceback
2022-09-06 19:33:51 +03:00
import numpy as np
import torch
2022-09-03 12:08:45 +03:00
from PIL import Image
import gradio as gr
import gradio . utils
2022-09-03 17:21:15 +03:00
import gradio . routes
2022-09-03 12:08:45 +03:00
from modules . paths import script_path
from modules . shared import opts , cmd_opts
import modules . shared as shared
from modules . sd_samplers import samplers , samplers_for_img2img
2022-09-21 16:06:37 +03:00
import modules . ldsr_model
2022-09-03 17:21:15 +03:00
import modules . scripts
2022-09-07 13:35:02 +03:00
import modules . gfpgan_model
import modules . codeformer_model
2022-09-09 23:16:02 +03:00
import modules . styles
2022-09-03 12:08:45 +03:00
# this is a fix for Windows users. Without it, javascript files will be served with text/html content-type and the bowser will not show any UI
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 '
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 ; }
. wrap . m - 12 : : before { content : " Loading... " }
. progress - bar { display : none ! important ; }
. meta - text { display : none ! important ; }
"""
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-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 image_from_url_text ( filedata ) :
if type ( filedata ) == list :
if len ( filedata ) == 0 :
return None
filedata = filedata [ 0 ]
if filedata . startswith ( " data:image/png;base64, " ) :
filedata = filedata [ len ( " data:image/png;base64, " ) : ]
filedata = base64 . decodebytes ( filedata . encode ( ' utf-8 ' ) )
image = Image . open ( io . BytesIO ( filedata ) )
return image
def send_gradio_gallery_to_image ( x ) :
if len ( x ) == 0 :
return None
return image_from_url_text ( x [ 0 ] )
2022-09-14 05:15:37 +03:00
2022-09-14 05:11:46 +03:00
def save_files ( js_data , images , index ) :
2022-09-03 12:08:45 +03:00
import csv
os . makedirs ( opts . outdir_save , exist_ok = True )
filenames = [ ]
data = json . loads ( js_data )
2022-09-14 05:11:46 +03:00
2022-09-14 06:01:30 +03:00
if index > - 1 and opts . save_selected_only and ( index > 0 or not opts . return_grid ) : # ensures we are looking at a specific non-grid picture, and we have save_selected_only
2022-09-14 05:11:46 +03:00
images = [ images [ index ] ]
data [ " seed " ] + = ( index - 1 if opts . return_grid else index )
2022-09-03 12:08:45 +03:00
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
filename_base = str ( int ( time . time ( ) * 1000 ) )
for i , filedata in enumerate ( images ) :
filename = filename_base + ( " " if len ( images ) == 1 else " - " + str ( i + 1 ) ) + " .png "
filepath = os . path . join ( opts . outdir_save , filename )
if filedata . startswith ( " data:image/png;base64, " ) :
filedata = filedata [ len ( " data:image/png;base64, " ) : ]
with open ( filepath , " wb " ) as imgfile :
imgfile . write ( base64 . decodebytes ( filedata . encode ( ' utf-8 ' ) ) )
filenames . append ( filename )
2022-09-12 19:57:31 +03:00
writer . writerow ( [ data [ " prompt " ] , data [ " seed " ] , data [ " width " ] , data [ " height " ] , data [ " sampler " ] , data [ " cfg_scale " ] , data [ " steps " ] , filenames [ 0 ] , data [ " negative_prompt " ] ] )
2022-09-03 12:08:45 +03:00
return ' ' , ' ' , plaintext_to_html ( f " Saved: { filenames [ 0 ] } " )
def wrap_gradio_call ( func ) :
def f ( * args , * * kwargs ) :
2022-09-18 14:03:17 +03:00
run_memmon = opts . memmon_poll_rate > 0 and not shared . mem_mon . disabled
if run_memmon :
2022-09-18 12:20:33 +03:00
shared . mem_mon . monitor ( )
2022-09-03 12:08:45 +03:00
t = time . perf_counter ( )
try :
res = list ( func ( * args , * * kwargs ) )
except Exception as e :
print ( " Error completing request " , file = sys . stderr )
print ( " Arguments: " , args , kwargs , file = sys . stderr )
print ( traceback . format_exc ( ) , file = sys . stderr )
2022-09-06 19:33:51 +03:00
shared . state . job = " "
shared . state . job_count = 0
2022-09-03 12:08:45 +03:00
res = [ None , ' ' , f " <div class= ' error ' > { plaintext_to_html ( type ( e ) . __name__ + ' : ' + str ( e ) ) } </div> " ]
elapsed = time . perf_counter ( ) - t
2022-09-18 14:03:17 +03:00
if run_memmon :
2022-09-18 12:20:33 +03:00
mem_stats = { k : - ( v / / - ( 1024 * 1024 ) ) for k , v in shared . mem_mon . stop ( ) . items ( ) }
active_peak = mem_stats [ ' active_peak ' ]
reserved_peak = mem_stats [ ' reserved_peak ' ]
sys_peak = mem_stats [ ' system_peak ' ]
sys_total = mem_stats [ ' total ' ]
sys_pct = round ( sys_peak / max ( sys_total , 1 ) * 100 , 2 )
2022-09-19 00:10:32 +03:00
vram_html = f " <p class= ' vram ' >Torch active/reserved: { active_peak } / { reserved_peak } MiB, <wbr>Sys VRAM: { sys_peak } / { sys_total } MiB ( { sys_pct } %)</p> "
2022-09-18 12:20:33 +03:00
else :
vram_html = ' '
2022-09-17 09:23:31 +03:00
2022-09-03 12:08:45 +03:00
# last item is always HTML
2022-09-17 09:23:31 +03:00
res [ - 1 ] + = f " <div class= ' performance ' ><p class= ' time ' >Time taken: <wbr> { elapsed : .2f } s</p> { vram_html } </div> "
2022-09-03 12:08:45 +03:00
shared . state . interrupted = False
return tuple ( res )
return f
2022-09-23 20:46:02 +03:00
def check_progress_call ( id_part ) :
2022-09-06 02:09:01 +03:00
if shared . state . job_count == 0 :
2022-09-06 19:33:51 +03:00
return " " , gr_show ( False ) , gr_show ( False )
2022-09-06 02:09:01 +03:00
2022-09-06 10:11:25 +03:00
progress = 0
if shared . state . job_count > 0 :
progress + = shared . state . job_no / shared . state . job_count
2022-09-06 02:09:01 +03:00
if shared . state . sampling_steps > 0 :
progress + = 1 / shared . state . job_count * shared . state . sampling_step / shared . state . sampling_steps
progress = min ( progress , 1 )
2022-09-06 19:33:51 +03:00
progressbar = " "
if opts . show_progressbar :
progressbar = f """ <div class= ' progressDiv ' ><div class= ' progress ' style= " width: { progress * 100 } % " > { str ( int ( progress * 100 ) ) + " % " if progress > 0.01 else " " } </div></div> """
image = gr_show ( False )
preview_visibility = gr_show ( False )
if opts . show_progress_every_n_steps > 0 :
2022-09-06 23:10:12 +03:00
if shared . parallel_processing_allowed :
2022-09-06 19:33:51 +03:00
2022-09-06 23:10:12 +03:00
if shared . state . sampling_step - shared . state . current_image_sampling_step > = opts . show_progress_every_n_steps and shared . state . current_latent is not None :
shared . state . current_image = modules . sd_samplers . sample_to_image ( shared . state . current_latent )
shared . state . current_image_sampling_step = shared . state . sampling_step
2022-09-06 20:26:09 +03:00
2022-09-06 19:33:51 +03:00
image = shared . state . current_image
if image is None or progress > = 1 :
image = gr . update ( value = None )
else :
preview_visibility = gr_show ( True )
2022-09-06 02:09:01 +03:00
2022-09-23 20:46:02 +03:00
return f " <span id= ' { id_part } _progress_span ' style= ' display: none ' > { time . time ( ) } </span><p> { progressbar } </p> " , preview_visibility , image
2022-09-06 02:09:01 +03:00
2022-09-23 20:46:02 +03:00
def check_progress_call_initial ( id_part ) :
2022-09-18 11:14:42 +03:00
shared . state . job_count = - 1
2022-09-18 16:36:34 +03:00
shared . state . current_latent = None
shared . state . current_image = None
2022-09-18 11:14:42 +03:00
2022-09-23 20:46:02 +03:00
return check_progress_call ( id_part )
2022-09-18 11:14:42 +03:00
2022-09-05 23:08:06 +03:00
def roll_artist ( prompt ) :
allowed_cats = set ( [ x for x in shared . artist_db . categories ( ) if len ( opts . random_artist_categories ) == 0 or x in opts . random_artist_categories ] )
artist = random . choice ( [ x for x in shared . artist_db . artists if x . category in allowed_cats ] )
return prompt + " , " + artist . name if prompt != ' ' else artist . name
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-09-09 23:16:02 +03:00
return [ gr_show ( ) , gr_show ( ) ]
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
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 ) :
prompt = shared . interrogator . interrogate ( image )
return gr_show ( True ) if prompt is None else prompt
2022-09-14 17:56:21 +03:00
2022-09-16 22:20:56 +03:00
def create_seed_inputs ( ) :
with gr . Row ( ) :
with gr . Box ( ) :
with gr . Row ( elem_id = ' seed_row ' ) :
2022-09-20 20:10:01 +03:00
seed = ( gr . Textbox if cmd_opts . use_textbox_seed else gr . Number ) ( label = ' Seed ' , value = - 1 )
2022-09-16 22:20:56 +03:00
seed . style ( container = False )
2022-09-18 18:10:22 +03:00
random_seed = gr . Button ( random_symbol , elem_id = ' random_seed ' )
reuse_seed = gr . Button ( reuse_symbol , elem_id = ' reuse_seed ' )
2022-09-16 22:20:56 +03:00
with gr . Box ( elem_id = ' subseed_show_box ' ) :
seed_checkbox = gr . Checkbox ( label = ' Extra ' , elem_id = ' subseed_show ' , value = False )
# Components to show/hide based on the 'Extra' checkbox
seed_extras = [ ]
with gr . Row ( visible = False ) as seed_extra_row_1 :
seed_extras . append ( seed_extra_row_1 )
with gr . Box ( ) :
with gr . Row ( elem_id = ' subseed_row ' ) :
subseed = gr . Number ( label = ' Variation seed ' , value = - 1 )
subseed . style ( container = False )
random_subseed = gr . Button ( random_symbol , elem_id = ' random_subseed ' )
reuse_subseed = gr . Button ( reuse_symbol , elem_id = ' reuse_subseed ' )
subseed_strength = gr . Slider ( label = ' Variation strength ' , value = 0.0 , minimum = 0 , maximum = 1 , step = 0.01 )
with gr . Row ( visible = False ) as seed_extra_row_2 :
seed_extras . append ( seed_extra_row_2 )
seed_resize_from_w = gr . Slider ( minimum = 0 , maximum = 2048 , step = 64 , label = " Resize seed from width " , value = 0 )
seed_resize_from_h = gr . Slider ( minimum = 0 , maximum = 2048 , step = 64 , label = " Resize seed from height " , value = 0 )
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-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-09-19 16:42:56 +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 " ) :
with gr . Column ( scale = 4 ) :
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 ( ) :
prompt = gr . Textbox ( label = " Prompt " , elem_id = " prompt " , show_label = False , placeholder = " Prompt " , lines = 2 )
roll = gr . Button ( ' Roll ' , elem_id = " roll " , visible = len ( shared . artist_db . artists ) > 0 )
2022-09-23 20:46:02 +03:00
with gr . Column ( scale = 10 , 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 ( ) ) ) , visible = len ( shared . prompt_styles . styles ) > 1 )
2022-09-14 17:56:21 +03:00
with gr . Row ( ) :
with gr . Column ( scale = 8 ) :
negative_prompt = gr . Textbox ( label = " Negative prompt " , elem_id = " negative_prompt " , show_label = False , placeholder = " Negative prompt " , lines = 2 )
with gr . Column ( scale = 1 , elem_id = " style_neg_col " ) :
2022-09-23 20:46:02 +03:00
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 ( ) ) ) , visible = len ( shared . prompt_styles . styles ) > 1 )
2022-09-14 17:56:21 +03:00
with gr . Column ( scale = 1 ) :
with gr . Row ( ) :
2022-09-23 20:46:02 +03:00
interrupt = gr . Button ( ' Interrupt ' , elem_id = f " { id_part } _interrupt " )
2022-09-14 17:56:21 +03:00
submit = gr . Button ( ' Generate ' , elem_id = " generate " , variant = ' primary ' )
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
with gr . Row ( ) :
if is_img2img :
interrogate = gr . Button ( ' Interrogate ' , elem_id = " interrogate " )
else :
interrogate = None
prompt_style_apply = gr . Button ( ' Apply style ' , elem_id = " style_apply " )
save_style = gr . Button ( ' Create style ' , elem_id = " style_create " )
2022-09-18 11:14:42 +03:00
return prompt , roll , prompt_style , negative_prompt , prompt_style2 , submit , interrogate , prompt_style_apply , save_style
2022-09-14 17:56:21 +03:00
2022-09-18 11:14:42 +03:00
2022-09-23 20:46:02 +03:00
def setup_progressbar ( progressbar , preview , id_part ) :
check_progress = gr . Button ( ' Check progress ' , elem_id = f " { id_part } _check_progress " , visible = False )
2022-09-18 11:14:42 +03:00
check_progress . click (
2022-09-23 20:46:02 +03:00
fn = lambda : check_progress_call ( id_part ) ,
2022-09-18 11:14:42 +03:00
show_progress = False ,
inputs = [ ] ,
outputs = [ progressbar , preview , preview ] ,
)
2022-09-23 20:46:02 +03:00
check_progress_initial = gr . Button ( ' Check progress (first) ' , elem_id = f " { id_part } _check_progress_initial " , visible = False )
2022-09-18 11:14:42 +03:00
check_progress_initial . click (
2022-09-23 20:46:02 +03:00
fn = lambda : check_progress_call_initial ( id_part ) ,
2022-09-18 11:14:42 +03:00
show_progress = False ,
inputs = [ ] ,
outputs = [ progressbar , preview , preview ] ,
)
2022-09-14 17:56:21 +03:00
2022-09-04 13:52:01 +03:00
def create_ui ( txt2img , img2img , run_extras , run_pnginfo ) :
2022-09-03 12:08:45 +03:00
with gr . Blocks ( analytics_enabled = False ) as txt2img_interface :
2022-09-18 11:14:42 +03:00
txt2img_prompt , roll , txt2img_prompt_style , txt2img_negative_prompt , txt2img_prompt_style2 , submit , _ , txt2img_prompt_style_apply , txt2img_save_style = create_toprow ( is_img2img = False )
2022-09-19 09:02:10 +03:00
dummy_component = gr . Label ( 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 ) :
with gr . Column ( variant = ' panel ' ) :
steps = gr . Slider ( minimum = 1 , maximum = 150 , step = 1 , label = " Sampling Steps " , value = 20 )
sampler_index = gr . Radio ( label = ' Sampling method ' , elem_id = " txt2img_sampling " , choices = [ x . name for x in samplers ] , value = samplers [ 0 ] . name , type = " index " )
with gr . Row ( ) :
2022-09-07 12:32:28 +03:00
restore_faces = gr . Checkbox ( label = ' Restore faces ' , value = False , visible = len ( shared . face_restorers ) > 1 )
2022-09-05 03:25:37 +03:00
tiling = gr . Checkbox ( label = ' Tiling ' , value = False )
2022-09-19 16:42:56 +03:00
enable_hr = gr . Checkbox ( label = ' Highres. fix ' , value = False )
with gr . Row ( visible = False ) as hr_options :
2022-09-20 19:32:26 +03:00
scale_latent = gr . Checkbox ( label = ' Scale latent ' , value = False )
2022-09-19 16:42:56 +03:00
denoising_strength = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.01 , label = ' Denoising strength ' , value = 0.7 )
2022-09-03 12:08:45 +03:00
with gr . Row ( ) :
batch_count = gr . Slider ( minimum = 1 , maximum = cmd_opts . max_batch_count , step = 1 , label = ' Batch count ' , value = 1 )
batch_size = gr . Slider ( minimum = 1 , maximum = 8 , step = 1 , label = ' Batch size ' , value = 1 )
2022-09-12 11:55:27 +03:00
cfg_scale = gr . Slider ( minimum = 1.0 , maximum = 30.0 , step = 0.5 , label = ' CFG Scale ' , value = 7.0 )
2022-09-03 12:08:45 +03:00
with gr . Group ( ) :
width = gr . Slider ( minimum = 64 , maximum = 2048 , step = 64 , label = " Width " , value = 512 )
2022-09-12 22:41:59 +03:00
height = gr . Slider ( minimum = 64 , maximum = 2048 , step = 64 , label = " Height " , value = 512 )
2022-09-03 12:08:45 +03:00
2022-09-21 13:34:10 +03:00
seed , reuse_seed , subseed , reuse_subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox = create_seed_inputs ( )
2022-09-03 12:08:45 +03:00
2022-09-03 17:21:15 +03:00
with gr . Group ( ) :
2022-09-04 01:29:43 +03:00
custom_inputs = modules . scripts . scripts_txt2img . setup_ui ( is_img2img = False )
2022-09-03 12:08:45 +03:00
with gr . Column ( variant = ' panel ' ) :
2022-09-17 12:38:15 +03:00
2022-09-03 12:08:45 +03:00
with gr . Group ( ) :
2022-09-06 19:33:51 +03:00
txt2img_preview = gr . Image ( elem_id = ' txt2img_preview ' , visible = False )
2022-09-22 12:30:11 +03:00
txt2img_gallery = gr . Gallery ( label = ' Output ' , show_label = False , elem_id = ' txt2img_gallery ' ) . style ( grid = 4 )
2022-09-03 12:08:45 +03:00
with gr . Group ( ) :
with gr . Row ( ) :
save = gr . Button ( ' Save ' )
send_to_img2img = gr . Button ( ' Send to img2img ' )
send_to_inpaint = gr . Button ( ' Send to inpaint ' )
send_to_extras = gr . Button ( ' Send to extras ' )
with gr . Group ( ) :
html_info = gr . HTML ( )
generation_info = gr . Textbox ( visible = False )
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 (
fn = txt2img ,
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 ,
scale_latent ,
denoising_strength ,
2022-09-03 17:21:15 +03:00
] + custom_inputs ,
2022-09-03 12:08:45 +03:00
outputs = [
txt2img_gallery ,
generation_info ,
html_info
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-09-19 16:42:56 +03:00
enable_hr . change (
fn = lambda x : gr_show ( x ) ,
inputs = [ enable_hr ] ,
outputs = [ hr_options ] ,
)
2022-09-03 12:08:45 +03:00
save . click (
fn = wrap_gradio_call ( save_files ) ,
2022-09-19 09:02:10 +03:00
_js = " (x, y, z) => [x, y, selected_gallery_index()] " ,
2022-09-03 12:08:45 +03:00
inputs = [
generation_info ,
txt2img_gallery ,
2022-09-19 09:02:10 +03:00
html_info ,
2022-09-03 12:08:45 +03:00
] ,
outputs = [
html_info ,
html_info ,
html_info ,
]
)
2022-09-05 23:08:06 +03:00
roll . click (
fn = roll_artist ,
inputs = [
2022-09-09 23:16:02 +03:00
txt2img_prompt ,
2022-09-05 23:08:06 +03:00
] ,
outputs = [
2022-09-09 23:16:02 +03:00
txt2img_prompt ,
2022-09-05 23:08:06 +03:00
]
)
2022-09-03 12:08:45 +03:00
with gr . Blocks ( analytics_enabled = False ) as img2img_interface :
2022-09-18 11:14:42 +03:00
img2img_prompt , roll , img2img_prompt_style , img2img_negative_prompt , img2img_prompt_style2 , submit , img2img_interrogate , img2img_prompt_style_apply , img2img_save_style = 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 ' ) :
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
2022-09-03 12:08:45 +03:00
with gr . Row ( ) . style ( equal_height = False ) :
with gr . Column ( variant = ' panel ' ) :
2022-09-09 19:43:16 +03:00
2022-09-22 12:11:48 +03:00
with gr . Tabs ( elem_id = " mode_img2img " ) as tabs_img2img_mode :
with gr . TabItem ( ' img2img ' ) :
init_img = gr . Image ( label = " Image for img2img " , show_label = False , source = " upload " , interactive = True , type = " pil " )
2022-09-03 12:08:45 +03:00
2022-09-22 12:11:48 +03:00
with gr . TabItem ( ' 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 " )
init_img_inpaint = gr . Image ( label = " Image for img2img " , show_label = False , source = " upload " , interactive = True , type = " pil " , visible = False )
init_mask_inpaint = gr . Image ( label = " Mask " , source = " upload " , interactive = True , type = " pil " , visible = False )
mask_blur = gr . Slider ( label = ' Mask blur ' , minimum = 0 , maximum = 64 , step = 1 , value = 4 )
with gr . Row ( ) :
mask_mode = gr . Radio ( label = " Mask mode " , show_label = False , choices = [ " Draw mask " , " Upload mask " ] , type = " index " , value = " Draw mask " )
inpainting_mask_invert = gr . Radio ( label = ' Masking mode ' , show_label = False , choices = [ ' Inpaint masked ' , ' Inpaint not masked ' ] , value = ' Inpaint masked ' , type = " index " )
inpainting_fill = gr . Radio ( label = ' Masked content ' , choices = [ ' fill ' , ' original ' , ' latent noise ' , ' latent nothing ' ] , value = ' fill ' , type = " index " )
with gr . Row ( ) :
inpaint_full_res = gr . Checkbox ( label = ' Inpaint at full resolution ' , value = False )
inpaint_full_res_padding = gr . Slider ( label = ' Inpaint at full resolution padding, pixels ' , minimum = 0 , maximum = 256 , step = 4 , value = 32 )
with gr . TabItem ( ' Batch img2img ' ) :
gr . HTML ( " <p class= \" text-gray-500 \" >Process images in a directory on the same machine where the server is running.</p> " )
img2img_batch_input_dir = gr . Textbox ( label = " Input directory " )
img2img_batch_output_dir = gr . Textbox ( label = " Output directory " )
2022-09-03 12:08:45 +03:00
with gr . Row ( ) :
2022-09-22 12:11:48 +03:00
resize_mode = gr . Radio ( label = " Resize mode " , elem_id = " resize_mode " , show_label = False , choices = [ " Just resize " , " Crop and resize " , " Resize and fill " ] , type = " index " , value = " Just resize " )
steps = gr . Slider ( minimum = 1 , maximum = 150 , step = 1 , label = " Sampling Steps " , value = 20 )
sampler_index = gr . Radio ( label = ' Sampling method ' , choices = [ x . name for x in samplers_for_img2img ] , value = samplers_for_img2img [ 0 ] . name , type = " index " )
2022-09-03 21:02:38 +03:00
with gr . Row ( ) :
2022-09-07 12:32:28 +03:00
restore_faces = gr . Checkbox ( label = ' Restore faces ' , value = False , visible = len ( shared . face_restorers ) > 1 )
2022-09-05 03:25:37 +03:00
tiling = gr . Checkbox ( label = ' Tiling ' , value = False )
2022-09-03 12:08:45 +03:00
with gr . Row ( ) :
batch_count = gr . Slider ( minimum = 1 , maximum = cmd_opts . max_batch_count , step = 1 , label = ' Batch count ' , value = 1 )
batch_size = gr . Slider ( minimum = 1 , maximum = 8 , step = 1 , label = ' Batch size ' , value = 1 )
with gr . Group ( ) :
2022-09-12 11:55:27 +03:00
cfg_scale = gr . Slider ( minimum = 1.0 , maximum = 30.0 , step = 0.5 , label = ' CFG Scale ' , value = 7.0 )
2022-09-08 15:02:06 +03:00
denoising_strength = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.01 , label = ' Denoising strength ' , value = 0.75 )
2022-09-03 12:08:45 +03:00
with gr . Group ( ) :
width = gr . Slider ( minimum = 64 , maximum = 2048 , step = 64 , label = " Width " , value = 512 )
2022-09-12 22:41:59 +03:00
height = gr . Slider ( minimum = 64 , maximum = 2048 , step = 64 , label = " Height " , value = 512 )
2022-09-03 12:08:45 +03:00
2022-09-21 13:34:10 +03:00
seed , reuse_seed , subseed , reuse_subseed , subseed_strength , seed_resize_from_h , seed_resize_from_w , seed_checkbox = create_seed_inputs ( )
2022-09-03 12:08:45 +03:00
2022-09-03 17:21:15 +03:00
with gr . Group ( ) :
2022-09-04 01:29:43 +03:00
custom_inputs = modules . scripts . scripts_img2img . setup_ui ( is_img2img = True )
2022-09-03 17:21:15 +03:00
2022-09-03 12:08:45 +03:00
with gr . Column ( variant = ' panel ' ) :
2022-09-17 12:38:15 +03:00
2022-09-03 12:08:45 +03:00
with gr . Group ( ) :
2022-09-06 19:33:51 +03:00
img2img_preview = gr . Image ( elem_id = ' img2img_preview ' , visible = False )
2022-09-22 12:30:11 +03:00
img2img_gallery = gr . Gallery ( label = ' Output ' , show_label = False , elem_id = ' img2img_gallery ' ) . style ( grid = 4 )
2022-09-03 12:08:45 +03:00
with gr . Group ( ) :
with gr . Row ( ) :
save = gr . Button ( ' Save ' )
2022-09-06 19:33:51 +03:00
img2img_send_to_img2img = gr . Button ( ' Send to img2img ' )
img2img_send_to_inpaint = gr . Button ( ' Send to inpaint ' )
2022-09-03 12:08:45 +03:00
img2img_send_to_extras = gr . Button ( ' Send to extras ' )
with gr . Group ( ) :
html_info = gr . HTML ( )
generation_info = gr . Textbox ( visible = False )
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-09 19:43:16 +03:00
mask_mode . change (
2022-09-22 12:11:48 +03:00
lambda mode , img : {
2022-09-09 19:43:16 +03:00
init_img_with_mask : gr_show ( mode == 0 ) ,
2022-09-22 12:11:48 +03:00
init_img_inpaint : gr_show ( mode == 1 ) ,
init_mask_inpaint : gr_show ( mode == 1 ) ,
2022-09-09 19:43:16 +03:00
} ,
2022-09-22 12:11:48 +03:00
inputs = [ mask_mode , init_img_with_mask ] ,
2022-09-09 19:43:16 +03:00
outputs = [
init_img_with_mask ,
2022-09-22 12:11:48 +03:00
init_img_inpaint ,
init_mask_inpaint ,
2022-09-09 19:43:16 +03:00
] ,
)
2022-09-03 12:08:45 +03:00
img2img_args = dict (
fn = img2img ,
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 ,
init_img_with_mask ,
2022-09-22 12:11:48 +03:00
init_img_inpaint ,
init_mask_inpaint ,
2022-09-09 19:43:16 +03:00
mask_mode ,
2022-09-03 12:08:45 +03:00
steps ,
sampler_index ,
mask_blur ,
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 ,
html_info
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-09-03 12:08:45 +03:00
save . click (
fn = wrap_gradio_call ( save_files ) ,
2022-09-18 11:14:42 +03:00
_js = " (x, y, z) => [x, y, selected_gallery_index()] " ,
2022-09-03 12:08:45 +03:00
inputs = [
generation_info ,
img2img_gallery ,
2022-09-14 05:11:46 +03:00
html_info
2022-09-03 12:08:45 +03:00
] ,
outputs = [
html_info ,
html_info ,
html_info ,
]
)
2022-09-14 17:56:21 +03:00
roll . click (
fn = roll_artist ,
inputs = [
img2img_prompt ,
] ,
outputs = [
img2img_prompt ,
]
)
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 ) ]
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 ] ,
)
for button , ( prompt , negative_prompt ) , ( style1 , style2 ) in zip ( [ txt2img_prompt_style_apply , img2img_prompt_style_apply ] , prompts , style_dropdowns ) :
button . click (
fn = apply_styles ,
inputs = [ prompt , negative_prompt , style1 , style2 ] ,
outputs = [ prompt , negative_prompt , style1 , style2 ] ,
2022-09-09 23:16:02 +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 ) :
with gr . Column ( variant = ' panel ' ) :
2022-09-22 12:11:48 +03:00
with gr . Tabs ( elem_id = " mode_extras " ) :
2022-09-16 06:23:37 +03:00
with gr . TabItem ( ' Single Image ' ) :
image = gr . Image ( label = " Source " , source = " upload " , interactive = True , type = " pil " )
with gr . TabItem ( ' Batch Process ' ) :
2022-09-16 10:21:59 +03:00
image_batch = gr . File ( label = " Batch Process " , file_count = " multiple " , interactive = True , type = " file " )
2022-09-04 18:54:12 +03:00
upscaling_resize = gr . Slider ( minimum = 1.0 , maximum = 4.0 , step = 0.05 , label = " Resize " , value = 2 )
with gr . Group ( ) :
extras_upscaler_1 = gr . Radio ( label = ' Upscaler 1 ' , choices = [ x . name for x in shared . sd_upscalers ] , value = shared . sd_upscalers [ 0 ] . name , type = " index " )
with gr . Group ( ) :
extras_upscaler_2 = gr . Radio ( label = ' Upscaler 2 ' , choices = [ x . name for x in shared . sd_upscalers ] , value = shared . sd_upscalers [ 0 ] . name , type = " index " )
extras_upscaler_2_visibility = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.001 , label = " Upscaler 2 visibility " , value = 1 )
with gr . Group ( ) :
2022-09-07 13:35:02 +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 )
with gr . Group ( ) :
codeformer_visibility = gr . Slider ( minimum = 0.0 , maximum = 1.0 , step = 0.001 , label = " CodeFormer visibility " , value = 0 , interactive = modules . codeformer_model . have_codeformer )
2022-09-07 13:46:08 +03:00
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 )
2022-09-03 12:08:45 +03:00
submit = gr . Button ( ' Generate ' , elem_id = " extras_generate " , variant = ' primary ' )
with gr . Column ( variant = ' panel ' ) :
2022-09-22 12:30:11 +03:00
result_images = gr . Gallery ( label = " Result " , show_label = False )
2022-09-03 12:08:45 +03:00
html_info_x = gr . HTML ( )
html_info = gr . HTML ( )
2022-09-23 06:40:39 +03:00
extras_send_to_img2img = gr . Button ( ' Send to img2img ' )
2022-09-23 14:17:23 +03:00
extras_send_to_inpaint = gr . Button ( ' Send to inpaint ' )
2022-09-03 12:08:45 +03:00
2022-09-22 12:11:48 +03:00
submit . click (
2022-09-03 12:08:45 +03:00
fn = run_extras ,
2022-09-22 12:11:48 +03:00
_js = " get_extras_tab_index " ,
2022-09-03 12:08:45 +03:00
inputs = [
2022-09-22 12:11:48 +03:00
dummy_component ,
2022-09-03 12:08:45 +03:00
image ,
2022-09-16 06:23:37 +03:00
image_batch ,
2022-09-07 13:35:02 +03:00
gfpgan_visibility ,
codeformer_visibility ,
codeformer_weight ,
2022-09-04 18:54:12 +03:00
upscaling_resize ,
extras_upscaler_1 ,
extras_upscaler_2 ,
extras_upscaler_2_visibility ,
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-09-23 06:40:39 +03:00
extras_send_to_img2img . click (
fn = lambda x : image_from_url_text ( x ) ,
_js = " extract_image_from_gallery_img2img " ,
inputs = [ result_images ] ,
outputs = [ init_img ] ,
)
2022-09-23 14:17:23 +03:00
extras_send_to_inpaint . click (
fn = lambda x : image_from_url_text ( x ) ,
_js = " extract_image_from_gallery_img2img " ,
inputs = [ result_images ] ,
outputs = [ init_img_with_mask ] ,
)
2022-09-03 12:08:45 +03:00
pnginfo_interface = gr . Interface (
wrap_gradio_call ( run_pnginfo ) ,
inputs = [
2022-09-21 01:28:03 +03:00
gr . Image ( elem_id = " pnginfo_image " , label = " Source " , source = " upload " , interactive = True , type = " pil " ) ,
2022-09-03 12:08:45 +03:00
] ,
outputs = [
gr . HTML ( ) ,
gr . HTML ( ) ,
gr . HTML ( ) ,
] ,
allow_flagging = " never " ,
analytics_enabled = False ,
2022-09-19 20:18:16 +03:00
live = True ,
2022-09-03 12:08:45 +03:00
)
def create_setting_component ( key ) :
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-09-11 23:00:42 +03:00
return comp ( label = info . label , value = fun , * * ( args or { } ) )
2022-09-03 12:08:45 +03:00
2022-09-10 11:10:00 +03:00
components = [ ]
2022-09-03 12:08:45 +03:00
def run_settings ( * args ) :
2022-09-23 17:27:30 +03:00
changed = 0
for key , value , comp in zip ( opts . data_labels . keys ( ) , args , components ) :
if not opts . same_type ( value , opts . data_labels [ key ] . default ) :
return 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-09-11 23:00:42 +03:00
comp_args = opts . data_labels [ key ] . component_args
if comp_args and isinstance ( comp_args , dict ) and comp_args . get ( ' visible ' ) is False :
continue
2022-09-17 12:05:04 +03:00
oldval = opts . data . get ( key , None )
2022-09-03 12:08:45 +03:00
opts . data [ key ] = value
2022-09-17 12:05:04 +03:00
2022-09-23 17:27:30 +03:00
if oldval != value :
if opts . data_labels [ key ] . onchange is not None :
opts . data_labels [ key ] . onchange ( )
2022-09-17 12:05:04 +03:00
2022-09-23 17:27:30 +03:00
changed + = 1
2022-09-03 12:08:45 +03:00
opts . save ( shared . config_filename )
2022-09-23 17:27:30 +03:00
return f ' { changed } settings changed. '
2022-09-03 12:08:45 +03:00
2022-09-10 11:10:00 +03:00
with gr . Blocks ( analytics_enabled = False ) as settings_interface :
2022-09-18 22:25:18 +03:00
settings_submit = gr . Button ( value = " Apply settings " , variant = ' primary ' )
2022-09-10 11:10:00 +03:00
result = gr . HTML ( )
2022-09-22 21:32:44 +03:00
settings_cols = 3
items_per_col = int ( len ( opts . data_labels ) * 0.9 / settings_cols )
2022-09-10 11:10:00 +03:00
2022-09-22 21:32:44 +03:00
cols_displayed = 0
items_displayed = 0
previous_section = None
column = None
with gr . Row ( elem_id = " settings " ) . style ( equal_height = False ) :
for i , ( k , item ) in enumerate ( opts . data_labels . items ( ) ) :
2022-09-22 19:26:26 +03:00
2022-09-22 21:32:44 +03:00
if previous_section != item . section :
if cols_displayed < settings_cols and ( items_displayed > = items_per_col or previous_section is None ) :
if column is not None :
column . __exit__ ( )
2022-09-22 19:26:26 +03:00
2022-09-22 21:32:44 +03:00
column = gr . Column ( variant = ' panel ' )
column . __enter__ ( )
2022-09-10 11:10:00 +03:00
2022-09-22 21:32:44 +03:00
items_displayed = 0
cols_displayed + = 1
previous_section = item . section
gr . HTML ( elem_id = " settings_header_text_ {} " . format ( item . section [ 0 ] ) , value = ' <h1 class= " gr-button-lg " > {} </h1> ' . format ( item . section [ 1 ] ) )
components . append ( create_setting_component ( k ) )
items_displayed + = 1
2022-09-03 12:08:45 +03:00
2022-09-22 13:15:33 +03:00
request_notifications = gr . Button ( value = ' Request browser notifications ' , elem_id = " request_notifications " )
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-09-22 21:32:44 +03:00
if column is not None :
column . __exit__ ( )
settings_submit . click (
fn = run_settings ,
inputs = components ,
outputs = [ result ]
)
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 " ) ,
( settings_interface , " Settings " , " settings " ) ,
2022-09-03 12:08:45 +03:00
]
with open ( os . path . join ( script_path , " style.css " ) , " r " , encoding = " utf8 " ) as file :
css = file . read ( )
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 :
usercss = file . read ( )
css + = usercss
2022-09-03 12:08:45 +03:00
if not cmd_opts . no_progressbar_hiding :
css + = css_hide_progressbar
2022-09-10 11:10:00 +03:00
with gr . Blocks ( css = css , analytics_enabled = False , title = " Stable Diffusion " ) as demo :
2022-09-18 22:25:18 +03:00
2022-09-10 11:10:00 +03:00
with gr . Tabs ( ) as tabs :
for interface , label , ifid in interfaces :
with gr . TabItem ( label , id = ifid ) :
interface . render ( )
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 (
fn = lambda : opts . dumpjson ( ) ,
inputs = [ ] ,
outputs = [ text_settings ] ,
)
2022-09-10 11:10:00 +03:00
send_to_img2img . click (
fn = lambda x : image_from_url_text ( x ) ,
2022-09-11 00:21:47 +03:00
_js = " extract_image_from_gallery_img2img " ,
2022-09-10 11:10:00 +03:00
inputs = [ txt2img_gallery ] ,
outputs = [ init_img ] ,
)
send_to_inpaint . click (
fn = lambda x : image_from_url_text ( x ) ,
2022-09-11 00:21:47 +03:00
_js = " extract_image_from_gallery_img2img " ,
2022-09-10 11:10:00 +03:00
inputs = [ txt2img_gallery ] ,
outputs = [ init_img_with_mask ] ,
)
img2img_send_to_img2img . click (
fn = lambda x : image_from_url_text ( x ) ,
_js = " extract_image_from_gallery " ,
inputs = [ img2img_gallery ] ,
outputs = [ init_img ] ,
)
img2img_send_to_inpaint . click (
fn = lambda x : image_from_url_text ( x ) ,
_js = " extract_image_from_gallery " ,
inputs = [ img2img_gallery ] ,
outputs = [ init_img_with_mask ] ,
)
send_to_extras . click (
fn = lambda x : image_from_url_text ( x ) ,
2022-09-11 00:21:47 +03:00
_js = " extract_image_from_gallery_extras " ,
2022-09-10 11:10:00 +03:00
inputs = [ txt2img_gallery ] ,
outputs = [ image ] ,
)
img2img_send_to_extras . click (
fn = lambda x : image_from_url_text ( x ) ,
2022-09-11 00:21:47 +03:00
_js = " extract_image_from_gallery_extras " ,
2022-09-10 11:10:00 +03:00
inputs = [ img2img_gallery ] ,
outputs = [ image ] ,
)
2022-09-03 12:08:45 +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-09-05 19:11:29 +03:00
def apply_field ( obj , field , condition = None ) :
2022-09-04 13:52:01 +03:00
key = path + " / " + field
saved_value = ui_settings . get ( key , None )
if saved_value is None :
ui_settings [ key ] = getattr ( obj , field )
2022-09-05 19:11:29 +03:00
elif condition is None or condition ( saved_value ) :
2022-09-04 13:52:01 +03:00
setattr ( obj , field , saved_value )
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
visit ( txt2img_interface , loadsave , " txt2img " )
visit ( img2img_interface , loadsave , " img2img " )
2022-09-11 11:31:16 +03:00
visit ( extras_interface , loadsave , " extras " )
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-09-03 17:21:15 +03:00
with open ( os . path . join ( script_path , " script.js " ) , " r " , encoding = " utf8 " ) as jsfile :
2022-09-18 08:37:03 +03:00
javascript = f ' <script> { jsfile . read ( ) } </script> '
jsdir = os . path . join ( script_path , " javascript " )
2022-09-23 05:05:42 +03:00
for filename in sorted ( os . listdir ( jsdir ) ) :
2022-09-18 08:37:03 +03:00
with open ( os . path . join ( jsdir , filename ) , " r " , encoding = " utf8 " ) as jsfile :
javascript + = f " \n <script> { jsfile . read ( ) } </script> "
2022-09-03 12:08:45 +03:00
2022-09-03 17:21:15 +03:00
def template_response ( * args , * * kwargs ) :
res = gradio_routes_templates_response ( * args , * * kwargs )
2022-09-18 08:37:03 +03:00
res . body = res . body . replace ( b ' </head> ' , f ' { javascript } </head> ' . encode ( " utf8 " ) )
2022-09-03 17:21:15 +03:00
res . init_headers ( )
return res
2022-09-03 12:08:45 +03:00
2022-09-03 17:21:15 +03:00
gradio_routes_templates_response = gradio . routes . templates . TemplateResponse
gradio . routes . templates . TemplateResponse = template_response