from typing import List import gradio as gr from modules.shared import opts from modules import sd_models, sd_samplers def face_inpainting_ui( name: str, id_prefix: str = "faceswaplab", description: str = "" ) -> List[gr.components.Component]: with gr.Accordion(name, open=False): gr.Markdown(description) inpainting_denoising_strength = gr.Slider( 0, 1, 0, step=0.01, elem_id=f"{id_prefix}_pp_inpainting_denoising_strength", label="Denoising strenght", ) inpainting_denoising_prompt = gr.Textbox( opts.data.get( "faceswaplab_pp_default_inpainting_prompt", "Portrait of a [gender]" ), elem_id=f"{id_prefix}_pp_inpainting_denoising_prompt", label="Inpainting prompt use [gender] instead of men or woman", ) inpainting_denoising_negative_prompt = gr.Textbox( opts.data.get( "faceswaplab_pp_default_inpainting_negative_prompt", "blurry" ), elem_id=f"{id_prefix}_pp_inpainting_denoising_neg_prompt", label="Inpainting negative prompt use [gender] instead of men or woman", ) with gr.Row(): samplers_names = [s.name for s in sd_samplers.all_samplers] inpainting_sampler = gr.Dropdown( choices=samplers_names, value=[samplers_names[0]], label="Inpainting Sampler", elem_id=f"{id_prefix}_pp_inpainting_sampler", ) inpainting_denoising_steps = gr.Slider( 1, 150, 20, step=1, label="Inpainting steps", elem_id=f"{id_prefix}_pp_inpainting_steps", ) inpaiting_model = gr.Dropdown( choices=["Current"] + sd_models.checkpoint_tiles(), default="Current", label="sd model (experimental)", elem_id=f"{id_prefix}_pp_inpainting_sd_model", ) gradio_components: List[gr.components.Component] = [ inpainting_denoising_strength, inpainting_denoising_prompt, inpainting_denoising_negative_prompt, inpainting_denoising_steps, inpainting_sampler, inpaiting_model, ] return gradio_components