add get_style_lists interface

app.py

@@ -20,6 +20,7 @@ MAX_SEED = np.iinfo(np.int32).max
 device = "cuda" if torch.cuda.is_available() else "cpu"
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "Watercolor"
+open_template_lists = False
 
 # download checkpoints
 from huggingface_hub import hf_hub_download
@@ -63,6 +64,9 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
+def get_style_lists() -> list:
+    return STYLE_NAMES
+
 
 # def remove_tips():
 #     return gr.update(visible=False)
@@ -341,7 +345,7 @@ with gr.Blocks(css=css) as demo:
             )
 
             submit = gr.Button("Submit", variant="primary")
-
+            style_list = gr.Checkbox(label="display the style templates", value=open_template_lists)
             style = gr.Dropdown(label="Style template", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
 
             # strength
@@ -427,6 +431,12 @@ with gr.Blocks(css=css) as demo:
             ],
             outputs=output_image,
         )
+
+        style_list.input(
+            fn=get_style_lists,
+            outputs=style_lists,
+            queue=False,
+        )
     
     # gr.Examples(
     #     examples=get_example(),