From 9ecf3471339d011983f2e3c878e920e49718ff90 Mon Sep 17 00:00:00 2001
From: AlUlkesh <99896447+AlUlkesh@users.noreply.github.com>
Date: Mon, 27 Mar 2023 20:01:19 +0200
Subject: [PATCH 1/2] fix: lightboxModal, selectedTab

---
 javascript/generationParams.js | 2 +-
 javascript/imageviewer.js      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/javascript/generationParams.js b/javascript/generationParams.js
index 95f05093..06a771bc 100644
--- a/javascript/generationParams.js
+++ b/javascript/generationParams.js
@@ -16,7 +16,7 @@ onUiUpdate(function(){
 
 let modalObserver = new MutationObserver(function(mutations) {
 	mutations.forEach(function(mutationRecord) {
-		let selectedTab = gradioApp().querySelector('#tabs div button.bg-white')?.innerText
+		let selectedTab = gradioApp().querySelector('#tabs div button')?.innerText
 		if (mutationRecord.target.style.display === 'none' && selectedTab === 'txt2img' || selectedTab === 'img2img')
 			gradioApp().getElementById(selectedTab+"_generation_info_button").click()
 	});
diff --git a/javascript/imageviewer.js b/javascript/imageviewer.js
index d6483562..bb61ee24 100644
--- a/javascript/imageviewer.js
+++ b/javascript/imageviewer.js
@@ -251,7 +251,7 @@ document.addEventListener("DOMContentLoaded", function() {
 
     modal.appendChild(modalNext)
 
-    gradioApp().appendChild(modal)
+    gradioApp().body.appendChild(modal)
 
 
     document.body.appendChild(modal);

From 5a25826d841a13574ab6afbeb9c50c81a491fa21 Mon Sep 17 00:00:00 2001
From: AlUlkesh <99896447+AlUlkesh@users.noreply.github.com>
Date: Tue, 28 Mar 2023 23:28:46 +0200
Subject: [PATCH 2/2] try both versions of appendChild

---
 javascript/imageviewer.js | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/javascript/imageviewer.js b/javascript/imageviewer.js
index bb61ee24..3deffa9b 100644
--- a/javascript/imageviewer.js
+++ b/javascript/imageviewer.js
@@ -251,8 +251,11 @@ document.addEventListener("DOMContentLoaded", function() {
 
     modal.appendChild(modalNext)
 
-    gradioApp().body.appendChild(modal)
-
+    try {
+		gradioApp().appendChild(modal);
+	} catch (e) {
+		gradioApp().body.appendChild(modal);
+	}
 
     document.body.appendChild(modal);