From 3694379f26500f54a7c6ece3d171ffd6635e7a93 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Thu, 18 May 2023 00:03:16 +0300
Subject: rework #8863 to work with all img2img tabs

---
 modules/ui.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'modules/ui.py')

diff --git a/modules/ui.py b/modules/ui.py
index b915482f..9ae0e2a5 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -954,6 +954,14 @@ def create_ui():
 
             res_switch_btn.click(fn=None, _js="function(){switchWidthHeight('img2img')}", inputs=None, outputs=None, show_progress=False)
 
+            detect_image_size_btn.click(
+                fn=lambda w, h, _: (w or gr.update(), h or gr.update()),
+                _js="currentImg2imgSourceResolution",
+                inputs=[dummy_component, dummy_component, dummy_component],
+                outputs=[width, height],
+                show_progress=False,
+            )
+
             restore_progress_button.click(
                 fn=progress.restore_progress,
                 _js="restoreProgressImg2img",
@@ -967,8 +975,6 @@ def create_ui():
                 show_progress=False,
             )
             
-            detect_image_size_btn.click(lambda i, w, h : i.size if i is not None else (w, h), inputs=[init_img, width, height], outputs=[width, height])
-
             img2img_interrogate.click(
                 fn=lambda *args: process_interrogate(interrogate, *args),
                 **interrogate_args,
-- 
cgit v1.2.1