From a628c45717cf733c9e27bc7bb0703b6c885bad68 Mon Sep 17 00:00:00 2001
From: Victor Seiji Hariki <victorseijih@gmail.com>
Date: Thu, 29 Dec 2022 00:37:18 -0300
Subject: [PATCH] Add restore faces option

Add restore faces option; Resolves #129

Signed-off-by: Victor Seiji Hariki <victorseijih@gmail.com>
---
 index.html  |  8 +++++++-
 js/index.js | 17 +++++++++++++++++
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/index.html b/index.html
index cf30edf..a351d64 100644
--- a/index.html
+++ b/index.html
@@ -100,6 +100,12 @@
 					<input type="checkbox" id="cbxHRFix" onchange="changeHiResFix()" />
 					<label for="cbxHRFix">Auto txt2img HRfix</label>
 					<br />
+					<input
+						type="checkbox"
+						id="cbxRestoreFaces"
+						onchange="changeRestoreFaces()" />
+					<label for="cbxRestoreFaces">Restore Faces</label>
+					<br />
 					<input
 						type="checkbox"
 						id="cbxSyncCursorSize"
@@ -328,7 +334,7 @@
 
 		<!-- Content -->
 		<script src="js/prompt.js?v=7a1c68c" type="text/javascript"></script>
-		<script src="js/index.js?v=36e55fc" type="text/javascript"></script>
+		<script src="js/index.js?v=3feb99d" type="text/javascript"></script>
 
 		<script
 			src="js/ui/floating/history.js?v=fc92d14"
diff --git a/js/index.js b/js/index.js
index 8c9505d..8ad8d81 100644
--- a/js/index.js
+++ b/js/index.js
@@ -105,6 +105,7 @@ var stableDiffusionData = {
 	inpaint_full_res: false,
 	inpainting_fill: 2,
 	enable_hr: false,
+	restore_faces: false,
 	firstphase_width: 0,
 	firstphase_height: 0,
 	styles: [],
@@ -158,6 +159,7 @@ function startup() {
 	changeSmoothRendering();
 	changeSeed();
 	changeHiResFix();
+	changeRestoreFaces();
 	changeSyncCursorSize();
 }
 
@@ -616,6 +618,16 @@ function changeHiResFix() {
 	localStorage.setItem("openoutpaint/enable_hr", stableDiffusionData.enable_hr);
 }
 
+function changeRestoreFaces() {
+	stableDiffusionData.restore_faces = Boolean(
+		document.getElementById("cbxRestoreFaces").checked
+	);
+	localStorage.setItem(
+		"openoutpaint/restore_faces",
+		stableDiffusionData.restore_faces
+	);
+}
+
 function changeSyncCursorSize() {
 	stableDiffusionData.sync_cursor_size = Boolean(
 		document.getElementById("cbxSyncCursorSize").checked
@@ -1007,6 +1019,10 @@ function loadSettings() {
 		localStorage.getItem("openoutpaint/enable_hr") === null
 			? false
 			: localStorage.getItem("openoutpaint/enable_hr") === "true";
+	let _restore_faces =
+		localStorage.getItem("openoutpaint/restore_faces") === null
+			? false
+			: localStorage.getItem("openoutpaint/restore_faces") === "true";
 
 	let _sync_cursor_size =
 		localStorage.getItem("openoutpaint/sync_cursor_size") === null
@@ -1017,6 +1033,7 @@ function loadSettings() {
 	document.getElementById("maskBlur").value = Number(_mask_blur);
 	document.getElementById("seed").value = Number(_seed);
 	document.getElementById("cbxHRFix").checked = Boolean(_enable_hr);
+	document.getElementById("cbxRestoreFaces").checked = Boolean(_restore_faces);
 	document.getElementById("cbxSyncCursorSize").checked =
 		Boolean(_sync_cursor_size);
 }