Add Image Gallery of generated images(of current session)

this will allow to compare images you created while using space :)


Co-authored-by: Zenafey Baenro <[email protected]>

app.py

@@ -159,7 +159,7 @@ for model_name in model_list:
     name_without_ext = remove_id_and_ext(model_name)
     model_names[name_without_ext] = model_name
 
-def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, gallery):
     result = prodia_client.generate({
         "prompt": prompt,
         "negative_prompt": negative_prompt,
@@ -174,9 +174,12 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
 
     job = prodia_client.wait(result)
 
-    return job["imageUrl"]
+    new_images_list = [img['name'] for img in gallery]
+    new_images_list.insert(0, job["imageUrl"])
 
-def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+    return {image_output: job["imageUrl"], gallery_obj: new_images_list}
+
+def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, gallery):
     result = prodia_client.transform({
         "imageData": image_to_base64(input_image),
         "denoising_strength": denoising,
@@ -193,7 +196,10 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
 
     job = prodia_client.wait(result)
 
-    return job["imageUrl"]
+    new_images_list = [img['name'] for img in gallery]
+    new_images_list.insert(0, job["imageUrl"])
+
+    return {i2i_image_output: job["imageUrl"], gallery_obj: new_images_list}
 
 
 css = """
@@ -288,8 +294,7 @@ with gr.Blocks(css=css) as demo:
                 with gr.Column(scale=2):
                     image_output = gr.Image(value="https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png")
     
-            text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
-        
+       
         with gr.Tab("img2img", id='i2i'):
             with gr.Row():
                 with gr.Column(scale=6, min_width=600):
@@ -327,8 +332,7 @@ with gr.Blocks(css=css) as demo:
                 with gr.Column(scale=2):
                     i2i_image_output = gr.Image(value="https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png")
     
-            i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt, model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height, i2i_seed], outputs=i2i_image_output)
-        
+
         with gr.Tab("PNG Info"):
             def plaintext_to_html(text, classname=None):
                 content = "<br>\n".join(html.escape(x) for x in text.split('\n'))
@@ -361,10 +365,16 @@ with gr.Blocks(css=css) as demo:
                 with gr.Column():
                     exif_output = gr.HTML(label="EXIF Data")
                     send_to_txt2img_btn = gr.Button("Send to txt2img")
-    
-            image_input.upload(get_exif_data, inputs=[image_input], outputs=exif_output)
-            send_to_txt2img_btn.click(send_to_txt2img, inputs=[image_input], outputs=[tabs, prompt, negative_prompt, steps, seed,
+
+        with gr.Tab("Gallery"):
+            gallery_obj = gr.Gallery(height=500, columns=4)
+
+        text_button.click(txt2img, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, gallery_obj], outputs=[image_output, gallery_obj])
+        image_input.upload(get_exif_data, inputs=[image_input], outputs=exif_output)
+        send_to_txt2img_btn.click(send_to_txt2img, inputs=[image_input], outputs=[tabs, prompt, negative_prompt, steps, seed,
                                                                                           model, sampler, width, height, cfg_scale])
-            
+        i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt, model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height, i2i_seed, gallery_obj], outputs=[i2i_image_output, gallery_obj])
+        
+
 demo.queue(concurrency_count=32)
 demo.launch()