cfg+preview

app.py

@@ -346,8 +346,8 @@ def process_box():
     return [gr.update(visible=True), *visibilities]
 
 @torch.inference_mode()
-@spaces.GPU(enable_queue=True)
-def generate_image(bg_prompt, bg_class, bg_tags, seed, *conditions):
+@spaces.GPU(enable_queue=True, duration=30)
+def generate_image(bg_prompt, bg_class, bg_tags, seed, cfg, *conditions):
 
     stack_cp = deepcopy(stack)
     print(f"conditions: {conditions}")
@@ -408,6 +408,7 @@ def generate_image(bg_prompt, bg_class, bg_tags, seed, *conditions):
             texts=prompts,
             bboxes=bboxes,
             num_inference_steps=50,
+            guidance_scale=cfg,
             generator=generator,
             text_attn_mask=None,
         ).images[0]
@@ -416,9 +417,10 @@ def generate_image(bg_prompt, bg_class, bg_tags, seed, *conditions):
 
     return image
 
-def process_example(bg_prompt, bg_class, bg_tags, color_str, style_str, text_str, box_str, seed):
+def process_example(prev_img, bg_prompt, bg_class, bg_tags, color_str, style_str, text_str, box_str, seed, cfg):
     global stack
     global state
+    print("CHANGE EXAMPLE!!!")
     
     colors = color_str.split(",")
     styles = style_str.split(",")
@@ -538,6 +540,7 @@ def main():
                                 ))
 
                     seed_ = gr.Slider(label="Seed", minimum=0, maximum=2147483647, value=42, step=1)
+                    cfg_ = gr.Slider(label="CFG Scale", minimum=1, maximum=10, value=5)
                     button_generate = gr.Button("(2) I've finished my texts, colors and styles, generate!", elem_id="main_button", interactive=True, variant='primary')
 
                 button_layout.click(process_box, inputs=[], outputs=[post_box, *color_row])
@@ -545,17 +548,19 @@ def main():
             with gr.Column():
                 output_image = gr.Image(label="Output Image", interactive=False)
 
-            button_generate.click(generate_image, inputs=[bg_prompt, bg_class, bg_tags, seed_, *(prompts + colors + styles)], outputs=[output_image], queue=True)
+            button_generate.click(generate_image, inputs=[bg_prompt, bg_class, bg_tags, seed_, cfg_, *(prompts + colors + styles)], outputs=[output_image], queue=True)
 
         # examples
         color_str = gr.Textbox(label="Color list", value="", visible=False)
         style_str = gr.Textbox(label="Font type list", value="", visible=False)
         box_str = gr.Textbox(label="Bbox list", value="", visible=False)
         text_str = gr.Textbox(label="Text list", value="", visible=False)
+        prev_img = gr.Image(label="Preview", visible = False)
 
         gr.Examples(
             examples=[
                 [
+                    'assets/previews/image1.webp',
                     'The image features a small bunny rabbit sitting in a basket filled with various flowers. The basket is placed on a yellow background, creating a vibrant and cheerful scene. The flowers surrounding the rabbit come in different sizes and colors, adding to the overall visual appeal of the image. The rabbit appears to be the main focus of the scene, and its presence among the flowers creates a sense of harmony and balance.',
                     'Facebook Post',
                     'green, yellow, minimalist, easter day, happy easter day, easter, happy easter, decoration, happy, egg, spring, selebration, poster, illustration, greeting, season, design, colorful, cute, template',
@@ -564,8 +569,10 @@ def main():
                     'MAY ALLYOUR PRAYERS BE ANSWERED**********HAVE A HAPPY**********Easter Day',
                     '[0.08267477203647416, 0.5355623100303951, 0.42857142857142855, 0.07477203647416414]; [0.08389057750759879, 0.1951367781155015, 0.38054711246200607, 0.03768996960486322]; [0.07537993920972644, 0.2601823708206687, 0.49544072948328266, 0.14650455927051673]',
                     1,
+                    5
                 ],
                 [
+                    'assets/previews/image2.webp',
                     'The image features a large gray elephant sitting in a field of flowers, holding a smaller elephant in its arms. The scene is quite serene and picturesque, with the two elephants being the main focus of the image. The field is filled with various flowers, creating a beautiful and vibrant backdrop for the elephants.',
                     'Cards and invitations',
                     'Light green, orange, Illustration, watercolor, playful, Baby shower invitation, baby boy shower invitation, baby boy, welcoming baby boy, koala baby shower invitation, baby shower invitation for baby shower, baby boy invitation, background, playful baby shower card, baby shower, card, newborn, born, Baby Shirt Baby Shower Invitation',
@@ -574,8 +581,10 @@ def main():
                     "RSVP to +123-456-7890**********Olivia Wilson**********Baby Shower**********Please Join Us For a**********In Honoring**********23 November, 2021 | 03:00 PM Fauget Hotels",
                     '[0.07112462006079028, 0.6462006079027356, 0.3373860182370821, 0.026747720364741642]; [0.07051671732522796, 0.38662613981762917, 0.37264437689969604, 0.059574468085106386]; [0.07234042553191489, 0.15623100303951368, 0.6547112462006079, 0.12401215805471125]; [0.0662613981762918, 0.06747720364741641, 0.3981762917933131, 0.035866261398176294]; [0.07051671732522796, 0.31550151975683893, 0.22006079027355624, 0.03951367781155015]; [0.06990881458966565, 0.48328267477203646, 0.39878419452887537, 0.1094224924012158]',
                     1,
+                    5
                 ],
                 [
+                    'assets/previews/image3.webp',
                     'The image features a white background with a variety of colorful flowers and decorations. There are several pink flowers scattered throughout the scene, with some positioned closer to the top and others near the bottom. A blue flower can also be seen in the middle of the image. The overall composition creates a visually appealing and vibrant display.',
                     'Instagram Posts',
                     'grey, navy, purple, pink, teal, colorful, illustration, happy, celebration, post, party, year, new, event, celebrate, happy new year, new year, countdown, sparkle, firework',
@@ -583,9 +592,11 @@ def main():
                     'Caveat-Regular, Gagalin-Regular, Quicksand-Light, Quicksand-Light',
                     'Happy New Year**********2024**********All THE BEST**********A fresh start to start a change for the better.',
                     '[0.2936170212765957, 0.2887537993920973, 0.40303951367781155, 0.07173252279635259]; [0.24984802431610942, 0.3951367781155015, 0.46200607902735563, 0.17203647416413373]; [0.3951367781155015, 0.1094224924012158, 0.2109422492401216, 0.02796352583586626]; [0.20911854103343466, 0.6127659574468085, 0.5586626139817629, 0.08085106382978724]',
-                    1,
+                    0,
+                    5
                 ],
                 [
+                    'assets/previews/image4.webp',
                     'The image features a stack of pancakes with syrup and strawberries on top. The pancakes are arranged in a visually appealing manner, with some pancakes placed on top of each other. The syrup is drizzled generously over the pancakes, and the strawberries are scattered around, adding a touch of color and freshness to the scene. The overall presentation of the pancakes is appetizing and inviting.',
                     'Instagram Posts',
                     'brown, peach, grey, modern, minimalist, simple, colorful, illustration, Instagram post, instagram, post, national pancake day, international pancake day, happy pancake day, pancake day, pancake, sweet, cake, discount, sale',
@@ -594,9 +605,11 @@ def main():
                     'Get 75% Discount for your first order**********Order Now**********National Pancake Day',
                     '[0.043161094224924014, 0.5963525835866261, 0.2936170212765957, 0.08389057750759879]; [0.12279635258358662, 0.79209726443769, 0.26382978723404255, 0.05167173252279635]; [0.044984802431610946, 0.09787234042553192, 0.4413373860182371, 0.4158054711246201]',
                     1,
+                    5
                 ]
             ],
             inputs=[
+                prev_img,
                 bg_prompt,
                 bg_class,
                 bg_tags,
@@ -605,6 +618,7 @@ def main():
                 text_str,
                 box_str,
                 seed_,
+                cfg_
             ],
             outputs=[post_box, box_sketch_template, seed_, *color_row, *colors, *styles, *prompts],
             fn=process_example,