Upload app.py

app.py

@@ -91,7 +91,7 @@ def gen_fn_single_gallery(model_str, prompt):
 def add_gallery(image, model_str, gallery):
     if gallery is None: gallery = []
     with lock:
-        gallery.append((image, model_str))
+        if image is not None: gallery.insert(0, (image, model_str))
     return gallery
 
 
@@ -103,7 +103,7 @@ def gen_fn_gallery(model_str, prompt, gallery):
         loop = asyncio.new_event_loop()
         result = loop.run_until_complete(infer(model_str, prompt, timeout))
         with lock:
-            if result: gallery.append(result)
+            if result: gallery.insert(0, result)
     except (Exception, asyncio.CancelledError) as e:
         print(e)
         print(f"Task aborted: {model_str}")
@@ -113,7 +113,8 @@ def gen_fn_gallery(model_str, prompt, gallery):
 
 
 CSS="""
-.output { width=120px; height=120px !important; }
+.output { width=120px; height=120px; !important; }
+.gallery { width=100%; height=512px; max_height=512px !important; }
 """
 
 with gr.Blocks(css=CSS) as demo:
@@ -135,12 +136,15 @@ with gr.Blocks(css=CSS) as demo:
             </div>
         """
                )
-        with gr.Row():
-            output = [gr.Image(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True, visible=True) for m in default_models]
-            #output = [gr.Image(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True) for m in default_models]
-            #output = [gr.Gallery(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True, container=True, format="png", object_fit="cover") for m in default_models]
-        current_models = [gr.Textbox(m, visible = False) for m in default_models]
-        gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="output", interactive=False, show_share_button=True, container=True, format="png", object_fit="cover")
+        with gr.Group():
+            with gr.Row():
+                output = [gr.Image(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True, min_width=80, visible=True) for m in default_models]
+                #output = [gr.Image(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True) for m in default_models]
+                #output = [gr.Gallery(label = m, show_download_button=True, elem_classes="output", interactive=False, show_share_button=True, container=True, format="png", object_fit="cover") for m in default_models]
+                current_models = [gr.Textbox(m, visible = False) for m in default_models]
+        gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+                            interactive=False, show_share_button=True, container=True, format="png",
+                            object_fit="cover", columns=2) # , height=768
                         
         for m, o in zip(current_models, output):
             #gen_event = gen_button.click(gen_fn, [m, txt_input], o)
@@ -151,7 +155,7 @@ with gr.Blocks(css=CSS) as demo:
 
         with gr.Accordion('Model selection'):
             model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 866 available!', value = default_models, interactive = True)
-            #model_choice.change(update_imgbox, model_choice, output)
+            model_choice.change(update_imgbox, model_choice, output)
             #model_choice.change(update_imgbox_gallery, model_choice, output)
             model_choice.change(extend_choices, model_choice, current_models)
         with gr.Row():