From c9b89f8af0e3fddc74572f0bf0fcfd2556cea2db Mon Sep 17 00:00:00 2001
From: anapnoe <124302297+anapnoe@users.noreply.github.com>
Date: Wed, 19 Apr 2023 05:22:02 +0300
Subject: [PATCH] fix Generate button state gets stuck sometimes when switching
 tabs #60

---
 .../sd_theme_editor/javascript/ui_theme.js       |  2 +-
 javascript/progressbar.js                        | 16 +++++++++++++---
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/extensions-builtin/sd_theme_editor/javascript/ui_theme.js b/extensions-builtin/sd_theme_editor/javascript/ui_theme.js
index 3d2f43a1..9a42527d 100644
--- a/extensions-builtin/sd_theme_editor/javascript/ui_theme.js
+++ b/extensions-builtin/sd_theme_editor/javascript/ui_theme.js
@@ -254,7 +254,7 @@ function initTheme() {
 
 
 	const current_style = gradioApp().querySelector('.gradio-container > style');
-	console.log(current_style);
+	//console.log(current_style);
 	//const head = document.head;		
 	//head.appendChild(current_style);
 
diff --git a/javascript/progressbar.js b/javascript/progressbar.js
index d96f2be5..19463cde 100644
--- a/javascript/progressbar.js
+++ b/javascript/progressbar.js
@@ -93,9 +93,19 @@ function requestProgress(id_task, progressbarContainer, gallery, atEnd, onProgre
 
     var removeProgressBar = function(){
         setTitle("")
-        parentProgressbar.removeChild(divProgress)
-        if(parentGallery) parentGallery.removeChild(livePreview)
-        atEnd()
+		if(divProgress){
+			parentProgressbar.removeChild(divProgress)
+		}
+		
+		if(progressbarContainer.id == "txt2img_gallery_container"){			
+			showSubmitButtons('txt2img', true);
+		}else if(progressbarContainer.id == "img2img_gallery_container"){
+			showSubmitButtons('img2img', true);
+		}
+		if(livePreview){
+			if(parentGallery) parentGallery.removeChild(livePreview)
+		}
+		atEnd()
     }
 
     var fun = function(id_task, id_live_preview){