Update app_ctrlx.py

app_ctrlx.py

@@ -85,7 +85,8 @@ title = """
         SDXL v1.0
     </h3>
 </div>
-<div style="display: flex; align-items: center; justify-content: center;margin-bottom: 25px">
+"""
+description = """<div style="display: flex; align-items: center; justify-content: center;margin-bottom: 25px">
     <h3 style="text-align: center">
         [<a href="https://genforce.github.io/ctrl-x/">Page</a>]
         &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;
@@ -167,6 +168,7 @@ def inference(
     width, height,
     structure_schedule, appearance_schedule, use_advanced_config,
     control_config,
+    progress=gr.Progress(track_tqdm=True)
 ):
     torch.manual_seed(seed)
     
@@ -235,54 +237,56 @@ def inference(
     
 with gr.Blocks(theme=gr.themes.Default(), css=css, title="Ctrl-X (SDXL v1.0)") as app:
     gr.HTML(title)
-    
+    with gr.Accordion("Instructions", open=False):
+        gr.HTML(description)
     with gr.Row():
         
         with gr.Column(scale=55):
             with gr.Group():
                 kwargs = {}  # {"width": 400, "height": 400}
+                with gr.Row():
+                    structure_image = gr.Image(label="Upload structure image (optional)", type="pil", **kwargs)
+                    appearance_image = gr.Image(label="Upload appearance image (optional)", type="pil", **kwargs)
                 with gr.Row():
                     result = gr.Image(label="Output image", format="jpg", **kwargs)
                     result_refiner = gr.Image(label="Output image w/ refiner", format="jpg", **kwargs)
                 with gr.Row():
                     structure_recon = gr.Image(label="Structure image", format="jpg", **kwargs)
                     appearance_recon = gr.Image(label="Style image", format="jpg", **kwargs)
-                with gr.Row():
-                    structure_image = gr.Image(label="Upload structure image (optional)", type="pil", **kwargs)
-                    appearance_image = gr.Image(label="Upload appearance image (optional)", type="pil", **kwargs)
-        
+                
         with gr.Column(scale=45):
             with gr.Group():
                 with gr.Row():
-                    structure_prompt = gr.Textbox(label="Structure prompt (optional)", placeholder="Prompt which describes the structure image")
-                    appearance_prompt = gr.Textbox(label="Appearance prompt (optional)", placeholder="Prompt which describes the style image")
+                    structure_prompt = gr.Textbox(label="Structure prompt (optional)", placeholder="Describes the structure image")
+                    appearance_prompt = gr.Textbox(label="Appearance prompt (optional)", placeholder="Describes the style image")
                 with gr.Row():
                     prompt = gr.Textbox(label="Output prompt", placeholder="Prompt which describes the output image")
                 with gr.Row():
                     positive_prompt = gr.Textbox(label="Positive prompt", value="high quality", placeholder="")
                     negative_prompt = gr.Textbox(label="Negative prompt", value="ugly, blurry, dark, low res, unrealistic", placeholder="")
-                with gr.Row():
-                    guidance_scale = gr.Slider(label="Target guidance scale", value=5.0, minimum=1, maximum=10)
-                    structure_guidance_scale = gr.Slider(label="Structure guidance scale", value=5.0, minimum=1, maximum=10)
-                    appearance_guidance_scale = gr.Slider(label="Appearance guidance scale", value=5.0, minimum=1, maximum=10)
-                with gr.Row():
-                    num_inference_steps = gr.Slider(label="# inference steps", value=50, minimum=1, maximum=200, step=1)
-                    eta = gr.Slider(label="Eta (noise)", value=1.0, minimum=0, maximum=1.0, step=0.01)
-                    seed = gr.Slider(0, 2147483647, label="Seed", value=90095, step=1)
-                with gr.Row():
-                    width = gr.Slider(label="Width", value=1024, minimum=256, maximum=2048, step=pipe.vae_scale_factor)
-                    height = gr.Slider(label="Height", value=1024, minimum=256, maximum=2048, step=pipe.vae_scale_factor)
-                with gr.Row():
-                    structure_schedule = gr.Slider(label="Structure schedule", value=0.6, minimum=0.0, maximum=1.0, step=0.01, scale=2)
-                    appearance_schedule = gr.Slider(label="Appearance schedule", value=0.6, minimum=0.0, maximum=1.0, step=0.01, scale=2)
-                    use_advanced_config = gr.Checkbox(label="Use advanced config", value=False, scale=1)
-                with gr.Row():
-                    control_config = gr.Textbox(
-                        label="Advanced control config", lines=20, value=get_control_config(0.6, 0.6), elem_classes=["config"], visible=False,
-                    )
-                    use_advanced_config.change(
-                        fn=lambda value: gr.update(visible=value), inputs=use_advanced_config, outputs=control_config,
-                    )
+                with gr.Accordion("Advanced Options", open=False):
+                    with gr.Row():
+                        guidance_scale = gr.Slider(label="Target guidance scale", value=5.0, minimum=1, maximum=10)
+                        structure_guidance_scale = gr.Slider(label="Structure guidance scale", value=5.0, minimum=1, maximum=10)
+                        appearance_guidance_scale = gr.Slider(label="Appearance guidance scale", value=5.0, minimum=1, maximum=10)
+                    with gr.Row():
+                        num_inference_steps = gr.Slider(label="# inference steps", value=50, minimum=1, maximum=200, step=1)
+                        eta = gr.Slider(label="Eta (noise)", value=1.0, minimum=0, maximum=1.0, step=0.01)
+                        seed = gr.Slider(0, 2147483647, label="Seed", value=90095, step=1)
+                    with gr.Row():
+                        width = gr.Slider(label="Width", value=1024, minimum=256, maximum=2048, step=pipe.vae_scale_factor)
+                        height = gr.Slider(label="Height", value=1024, minimum=256, maximum=2048, step=pipe.vae_scale_factor)
+                    with gr.Row():
+                        structure_schedule = gr.Slider(label="Structure schedule", value=0.6, minimum=0.0, maximum=1.0, step=0.01, scale=2)
+                        appearance_schedule = gr.Slider(label="Appearance schedule", value=0.6, minimum=0.0, maximum=1.0, step=0.01, scale=2)
+                        use_advanced_config = gr.Checkbox(label="Use advanced config", value=False, scale=1)
+                    with gr.Row():
+                        control_config = gr.Textbox(
+                            label="Advanced control config", lines=20, value=get_control_config(0.6, 0.6), elem_classes=["config"], visible=False,
+                        )
+                        use_advanced_config.change(
+                            fn=lambda value: gr.update(visible=value), inputs=use_advanced_config, outputs=control_config,
+                        )
                 with gr.Row():
                     generate = gr.Button(value="Run")