Update app.py

app.py

@@ -4,6 +4,9 @@ from random import randint
 from all_models import models
 from datetime import datetime
 
+from threading import RLock
+lock = RLock()
+
 now2 = 0
 nb_models=24
 
@@ -145,6 +148,12 @@ def gen_fn(model_str, prompt):
     
     return m
 
+def add_gallery(image, model_str, gallery):
+    if gallery is None: gallery = []
+    with lock:
+        if image is not None: gallery.insert(0, (image, model_str))
+    return gallery
+
 def make_me():
    # with gr.Tab('The Dream'): 
         with gr.Row():
@@ -173,11 +182,16 @@ def make_me():
             for m, o in zip(current_models, output):
                 gen_event = gen_button.click(gen_fn, [m, txt_input], o)
                 stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
+                o.change(add_gallery, [o, m, gallery], [gallery])
         """with gr.Accordion('Model selection'):
             model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, multiselect=True, max_choices=num_models, interactive=True, filterable=False)
             model_choice.change(update_imgbox, (gen_button,stop_button,group_model_choice), output)
             model_choice.change(extend_choices, model_choice, current_models)
         """     
+        with gr.Row():
+            gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+                                interactive=False, show_share_button=True, container=True, format="png",
+                                preview=True, object_fit="cover", columns=2, rows=2) 
         
         with gr.Accordion("test", open=True):
             """group_model_choice = gr.Dropdown(label="test Model", show_label=False, choices=list(models_test) , allow_custom_value=True)"""
@@ -196,6 +210,7 @@ def make_me():
                 <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77 and Omnibus's Maximum Multiplier!
                 </p>
             """)
+            
 
 
 js_code = """