Update app.py

app.py

@@ -39,6 +39,8 @@ inference_timeout = 400
 default_models = models[:num_models]
 MAX_SEED = 2**32-1
 
+def imgageHasUpdated(theImage):
+    print("Check")
 
 def extend_choices(choices):
     return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
@@ -87,7 +89,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     if task.done() and result is not None and not isinstance(result, tuple):
         with lock:
             png_path =  model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
-            lnk_output.update(value=png_path)
+            lnk_output(value=png_path)
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, theSeed)
         return image
     return None
@@ -195,7 +197,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(lines=1,label='The Link:',visible=True,interactive=False,elem_id="LnkBox")
+            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():
@@ -214,20 +216,15 @@ with gr.Blocks(head=CSS + JS) as demo:
     with gr.Column(scale=1):
         with gr.Group():
             with gr.Row():
-                output = [gr.Image(label=m, show_download_button=True, elem_classes=["image-monitor"],
-                          interactive=False, width=112, height=112, show_share_button=False, format="png",
-                          visible=True) for m in default_models]
+                output = [gr.Image(label=m, show_download_button=True, elem_classes=["image-monitor"], interactive=False, width=112, height=112, show_share_button=False, format="png", visible=True) for m in default_models]
+                output.change(imgageHasUpdated)
                 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) 
+        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)
+        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])
 
     with gr.Column(scale=4):