Save options

app.py

@@ -27,6 +27,79 @@ else:
 pipe = StableDiffusionXLInpaintPipeline.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype = floatType, variant = variant)
 pipe = pipe.to(device)
 
+default_local_storage = {
+        "prompt":               "",
+        "negative_prompt":      "Ugly, malformed, noise, blur, watermark",
+        "num_inference_steps":  25,
+        "guidance_scale":       7,
+        "image_guidance_scale": 1.1,
+        "strength":             0.99,
+        "denoising_steps":      1000,
+        "randomize_seed":       True,
+        "seed":                 random.randint(0, max_64_bit_int),
+        "debug_mode":           False
+        }
+
+def save_preferences_prompt(preferences, value):
+    preferences["prompt"] = value
+    return preferences
+
+def save_preferences_negative_prompt(preferences, value):
+    preferences["negative_prompt"] = value
+    return preferences
+
+def save_preferences_num_inference_steps(preferences, value):
+    preferences["num_inference_steps"] = value
+    return preferences
+
+def save_preferences_guidance_scale(preferences, value):
+    preferences["guidance_scale"] = value
+    return preferences
+
+def save_preferences_image_guidance_scale(preferences, value):
+    preferences["image_guidance_scale"] = value
+    return preferences
+
+def save_preferences_strength(preferences, value):
+    preferences["strength"] = value
+    return preferences
+
+def save_preferences_denoising_steps(preferences, value):
+    preferences["denoising_steps"] = value
+    return preferences
+
+def save_preferences_randomize_seed(preferences, value):
+    preferences["randomize_seed"] = value
+    return preferences
+
+def save_preferences_seed(preferences, value):
+    preferences["seed"] = value
+    return preferences
+
+def save_preferences_debug_mode(preferences, value):
+    preferences["debug_mode"] = value
+    return preferences
+
+def load_preferences(saved_prefs):
+    saved_prefs = init_preferences(saved_prefs)
+    return [
+        saved_prefs["prompt"],
+        saved_prefs["negative_prompt"],
+        saved_prefs["num_inference_steps"],
+        saved_prefs["guidance_scale"],
+        saved_prefs["image_guidance_scale"],
+        saved_prefs["strength"],
+        saved_prefs["denoising_steps"],
+        saved_prefs["randomize_seed"],
+        saved_prefs["seed"],
+        saved_prefs["debug_mode"]
+    ]
+
+def init_preferences(saved_prefs):
+    if saved_prefs is None:
+        saved_prefs = default_local_storage
+    return saved_prefs
+
 def update_seed(is_randomize_seed, seed):
     if is_randomize_seed:
         return random.randint(0, max_64_bit_int)
@@ -241,6 +314,7 @@ def inpaint_on_gpu(
     ).images[0]
 
 with gr.Blocks() as interface:
+    local_storage = gr.BrowserState(default_local_storage)
     gr.HTML(
         """
         <h1 style="text-align: center;">Inpaint / Outpaint</h1>
@@ -333,6 +407,76 @@ with gr.Blocks() as interface:
         mask_image
     ], scroll_to_output = True)
 
+    prompt.change(fn = save_preferences_prompt, inputs = [
+        local_storage,
+        prompt,
+    ], outputs = [
+        local_storage
+    ])
+
+    negative_prompt.change(fn = save_preferences_negative_prompt, inputs = [
+        local_storage,
+        negative_prompt,
+    ], outputs = [
+        local_storage
+    ])
+
+    num_inference_steps.change(fn = save_preferences_num_inference_steps, inputs = [
+        local_storage,
+        num_inference_steps,
+    ], outputs = [
+        local_storage
+    ])
+
+    guidance_scale.change(fn = save_preferences_guidance_scale, inputs = [
+        local_storage,
+        guidance_scale,
+    ], outputs = [
+        local_storage
+    ])
+
+    image_guidance_scale.change(fn = save_preferences_image_guidance_scale, inputs = [
+        local_storage,
+        image_guidance_scale,
+    ], outputs = [
+        local_storage
+    ])
+
+    strength.change(fn = save_preferences_strength, inputs = [
+        local_storage,
+        strength,
+    ], outputs = [
+        local_storage
+    ])
+
+    denoising_steps.change(fn = save_preferences_denoising_steps, inputs = [
+        local_storage,
+        denoising_steps,
+    ], outputs = [
+        local_storage
+    ])
+
+    randomize_seed.change(fn = save_preferences_randomize_seed, inputs = [
+        local_storage,
+        randomize_seed,
+    ], outputs = [
+        local_storage
+    ])
+
+    seed.change(fn = save_preferences_seed, inputs = [
+        local_storage,
+        seed,
+    ], outputs = [
+        local_storage
+    ])
+
+    debug_mode.change(fn = save_preferences_debug_mode, inputs = [
+        local_storage,
+        debug_mode,
+    ], outputs = [
+        local_storage
+    ])
+
     gr.Examples(
         fn = inpaint,
 	    inputs = [
@@ -480,4 +624,22 @@ with gr.Blocks() as interface:
         """
     )
 
+    # Load saved preferences when the page loads
+    interface.load(
+        fn=load_preferences, inputs = [
+        local_storage
+    ], outputs = [
+       prompt,
+       negative_prompt,
+       num_inference_steps,
+       guidance_scale,
+       image_guidance_scale,
+       strength,
+       denoising_steps,
+       randomize_seed,
+       seed,
+       debug_mode
+    ]
+    )
+
     interface.queue().launch()