Update app.py

app.py

@@ -39,6 +39,10 @@ inference_timeout = 400
 default_models = models[:num_models]
 MAX_SEED = 2**32-1
 
+def imgageHasUpdated(theImage):
+    print(theImage)
+    # outputs=lnk_output
+
 def extend_choices(choices):
     return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
 
@@ -230,6 +234,7 @@ with gr.Blocks(head=CSS + JS) as demo:
     with gr.Column(scale=2):
         with gr.Group():
             txt_input = gr.Textbox(label='Your prompt:', value=preSetPrompt, lines=3, autofocus=1)
+            lnk_output = gr.Textbox(label='The Link:',visible=True,interactive=False,elem_id="LnkBox")
             neg_input = gr.Textbox(label='Negative prompt:', value=negPreSetPrompt, lines=1)
             with gr.Accordion("Advanced", open=False, visible=True):
                 with gr.Row():
@@ -242,7 +247,7 @@ with gr.Blocks(head=CSS + JS) as demo:
                     seed_rand = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
                     seed_rand.click(randomize_seed, None, [seed], queue=False)
         with gr.Row():
-            gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3, elem_id=["theButt"])
+            gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3, elem_classes=["butt"])
             random_button = gr.Button(f'Randomize Models', variant='secondary', scale=1)
 
     with gr.Column(scale=1):
@@ -251,6 +256,14 @@ with gr.Blocks(head=CSS + JS) as demo:
                 output = [gr.Image(label=m, show_download_button=True, interactive=False, width=112, height=112, show_share_button=False, format="png", visible=True) for m in default_models]
                 current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
+    with gr.Column(scale=2):
+        gallery = gr.Gallery(label="Output", show_download_button=True,interactive=False, show_share_button=False, container=True, format="png", preview=True, object_fit="cover", columns=2, rows=2) 
+
+    for m, o in zip(current_models, output):
+        gen_event = gr.on(triggers=[gen_button.click, txt_input.submit], fn=gen_fn,inputs=[m, txt_input, neg_input, height, width, steps, cfg, seed], outputs=[o], concurrency_limit=None, queue=False)
+        # o.change(add_gallery, [o, m, gallery], [gallery])
+        # o.change(imgageHasUpdated,[o])
+
     with gr.Column(scale=4):
         with gr.Accordion('Model selection'):
             model_choice = gr.CheckboxGroup(models, label = f'Choose up to {int(num_models)} different models from the {len(models)} available!', value=default_models, interactive=True)