Add Image Gallery of generated images(of current session)

app.py

@@ -162,7 +162,7 @@ for model_name in model_list:
     name_without_ext = remove_id_and_ext(model_name)
     model_names[name_without_ext] = model_name
 
-def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed, gallery):
     result = prodia_client.generate({
         "prompt": prompt,
         "negative_prompt": negative_prompt,
@@ -177,7 +177,10 @@ def flip_text(prompt, negative_prompt, model, steps, sampler, cfg_scale, width,
 
     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 {image_output: job["imageUrl"], gallery_obj: new_images_list}
 
 
 css = """
@@ -250,8 +253,6 @@ 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(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
         
         with gr.Tab("PNG Info"):
             def plaintext_to_html(text, classname=None):
@@ -285,9 +286,13 @@ 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(flip_text, 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])
 
 demo.queue(concurrency_count=32)