Add caching and allow scale boxes

app.py

@@ -61,7 +61,7 @@ layout_placeholder = """Caption: A realistic photo of a gray cat and an orange d
 Objects: [('a gray cat', [67, 243, 120, 126]), ('an orange dog', [265, 193, 190, 210])]
 Background prompt: A realistic photo of a grassy area."""
 
-def get_lmd_prompt(prompt, template=""):
+def get_lmd_prompt(prompt, template=default_template):
     if prompt == "":
         prompt = prompt_placeholder
     if template == "":
@@ -88,7 +88,7 @@ def get_layout_image(response):
 def get_layout_image_gallery(response):
     return [get_layout_image(response)]
 
-def get_ours_image(response, seed, num_inference_steps, dpm_scheduler, fg_seed_start, fg_blending_ratio=0.1, frozen_step_ratio=0.4, gligen_scheduled_sampling_beta=0.3, so_negative_prompt="", overall_negative_prompt="", show_so_imgs=False, scale_boxes=False):
+def get_ours_image(response, seed, num_inference_steps=20, dpm_scheduler=True, fg_seed_start=20, fg_blending_ratio=0.1, frozen_step_ratio=0.4, gligen_scheduled_sampling_beta=0.3, so_negative_prompt=DEFAULT_SO_NEGATIVE_PROMPT, overall_negative_prompt=DEFAULT_OVERALL_NEGATIVE_PROMPT, show_so_imgs=False, scale_boxes=False):
     if response == "":
         response = layout_placeholder
     gen_boxes, bg_prompt = parse_input(response)
@@ -116,7 +116,7 @@ def get_ours_image(response, seed, num_inference_steps, dpm_scheduler, fg_seed_s
         images.extend([np.asarray(so_img) for so_img in so_img_list])
     return images
 
-def get_baseline_image(prompt, seed):
+def get_baseline_image(prompt, seed=0):
     if prompt == "":
         prompt = prompt_placeholder
     
@@ -222,8 +222,11 @@ with gr.Blocks(
         generate_btn.click(fn=get_lmd_prompt, inputs=[prompt, template], outputs=output, api_name="get_lmd_prompt")
     
         gr.Examples(
-            stage1_examples,
-            [prompt]
+            examples=stage1_examples,
+            inputs=[prompt],
+            outputs=[output],
+            fn=get_lmd_prompt,
+            cache_examples=True
         )
     
     # with gr.Tab("(Optional) Visualize ChatGPT-generated Layout"):
@@ -251,17 +254,21 @@ with gr.Blocks(
                     gligen_scheduled_sampling_beta = gr.Slider(0, 1, value=0.3, step=0.1, label="GLIGEN guidance steps ratio (the beta value)")
                     so_negative_prompt = gr.Textbox(lines=1, label="Negative prompt for single object generation", value=DEFAULT_SO_NEGATIVE_PROMPT)
                     overall_negative_prompt = gr.Textbox(lines=1, label="Negative prompt for overall generation", value=DEFAULT_OVERALL_NEGATIVE_PROMPT)
-                    show_so_imgs = gr.Checkbox(label="Show annotated single object generations", show_label=False)
+                    show_so_imgs = gr.Checkbox(label="Show annotated single object generations", show_label=False, value=False)
+                    scale_boxes = gr.Checkbox(label="Scale bounding boxes to just fit the scene", show_label=False, value=False)
             with gr.Column(scale=1):
                 gallery = gr.Gallery(
                     label="Generated image", show_label=False, elem_id="gallery"
                 ).style(columns=[1], rows=[1], object_fit="contain", preview=True)
         visualize_btn.click(fn=get_layout_image_gallery, inputs=response, outputs=gallery, api_name="visualize-layout")
-        generate_btn.click(fn=get_ours_image, inputs=[response, seed, num_inference_steps, dpm_scheduler, fg_seed_start, fg_blending_ratio, frozen_step_ratio, gligen_scheduled_sampling_beta, so_negative_prompt, overall_negative_prompt, show_so_imgs], outputs=gallery, api_name="layout-to-image")
+        generate_btn.click(fn=get_ours_image, inputs=[response, seed, num_inference_steps, dpm_scheduler, fg_seed_start, fg_blending_ratio, frozen_step_ratio, gligen_scheduled_sampling_beta, so_negative_prompt, overall_negative_prompt, show_so_imgs, scale_boxes], outputs=gallery, api_name="layout-to-image")
 
         gr.Examples(
-            stage2_examples,
-            [response, seed]
+            examples=stage2_examples,
+            inputs=[response, seed],
+            outputs=[gallery],
+            fn=get_ours_image,
+            cache_examples=True
         )
 
     with gr.Tab("Baseline: Stable Diffusion"):
@@ -280,8 +287,11 @@ with gr.Blocks(
         generate_btn.click(fn=get_baseline_image, inputs=[sd_prompt, seed], outputs=gallery, api_name="baseline")
 
         gr.Examples(
-            stage1_examples,
-            [sd_prompt]
+            examples=stage1_examples,
+            inputs=[sd_prompt],
+            outputs=[gallery],
+            fn=get_baseline_image,
+            cache_examples=True
         )
 
 g.launch()