Update app.py

app.py

@@ -23,8 +23,7 @@ sa_args = sa_handler.StyleAlignedArgs(share_group_norm=False,
 
 handler.register(sa_args, )
 
-
-# run StyleAligned
+# example of set of prompts
 sets_of_prompts = [
   "a toy train. macro photo. 3d game asset",
   "a toy airplane. macro photo. 3d game asset",
@@ -33,20 +32,44 @@ sets_of_prompts = [
   "a toy boat. macro photo. 3d game asset",
 ]
 
-def style_aligned_sdxl(prompt): 
-    images = pipeline([prompt],).images
+# run StyleAligned
+def style_aligned_sdxl(initial_prompt1, initial_prompt2, initial_prompt3, initial_prompt4, initial_prompt5, style_prompt): 
+    sets_of_prompts = [ prompt + ". " + style_prompt for prompt in [initial_prompt1, initial_prompt2, initial_prompt3, initial_prompt4, initial_prompt5,]]
+    images = pipeline(sets_of_prompts,).images
     #mediapy.show_images(images)
     print(images)
-    return images[0]
+    return images
 
 with gr.Blocks() as demo:
     with gr.Group():
-      with gr.Row():
-        prompt = gr.Textbox(label="Prompt", scale=8)
-        btn = gr.Button("Greet", scale=2)
-    output = gr.Image(label="Style-Aligned SDXL", type='pil')
+      with gr.Column():
+        with gr.Accordion(label='Enter upto 5 different initial prompts', open=True):
+          with gr.Row():
+            initial_prompt1 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy train')
+            initial_prompt2 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy airplane')
+            initial_prompt3 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy bicycle')
+            initial_prompt4 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy car')
+            initial_prompt5 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy boat')
+        with gr.Row():
+          style_prompt = gr.Textbox(label="Enter a style prompt", placeholder='macro photo, 3d game asset')
+        btn = gr.Button("Generate a set of Style-aligned SDXL images",)
+    output = gr.Gallery(label="Style-Aligned SDXL", )
     
-    btn.click(fn=style_aligned_sdxl, inputs=prompt, outputs=output, api_name="style_aligned_sdxl")
+    btn.click(fn=style_aligned_sdxl, 
+              inputs=[initial_prompt1, initial_prompt2, initial_prompt3, initial_prompt4, initial_prompt5, style_prompt], 
+              outputs=output, 
+              api_name="style_aligned_sdxl")
+
+    gr.Examples(examples=[
+                    ["a toy train", "a toy airplane", "a toy bicycle", "a toy car", "a toy boat", "macro photo. 3d game asset."],
+                    ["a toy train", "a toy airplane", "a toy bicycle", "a toy car", "a toy boat", "BW logo. high contrast."],
+                    ["a cat", "a dog", "a bear", "a man on a bicycle", "a girl working on laptop", "minimal origami."],
+                    ["Firewoman", "Gradner", "Scientist", "Policewoman", "Saxophone player", "made of claymation, stop motion animation."],
+                    ["Firewoman", "Gradner", "Scientist", "Policewoman", "Saxophone player", "sketch, character sheet."],
+                    ],
+            inputs=[initial_prompt1, initial_prompt2, initial_prompt3, initial_prompt4, initial_prompt5, style_prompt],
+            outputs=[output],
+            fn=style_aligned_sdxl)
 
 demo.launch()