From e481e2b3dad5dcc2833904e9d5000ef7d5caa30a Mon Sep 17 00:00:00 2001
From: Victor Seiji Hariki <victorseijih@gmail.com>
Date: Mon, 2 Jan 2023 21:43:33 -0300
Subject: [PATCH] adds first run inpainting model prompts

Should finish implementing (Fixes #125)

Signed-off-by: Victor Seiji Hariki <victorseijih@gmail.com>
---
 index.html   |  4 ++--
 js/global.js |  7 +++++++
 js/index.js  | 32 +++++++++++++++++++++++++++-----
 3 files changed, 36 insertions(+), 7 deletions(-)

diff --git a/index.html b/index.html
index 30ced11..d41284c 100644
--- a/index.html
+++ b/index.html
@@ -316,7 +316,7 @@
 		</div>
 
 		<!-- Basics -->
-		<script src="js/global.js?v=3da0b2e" type="text/javascript"></script>
+		<script src="js/global.js?v=1807d6e" type="text/javascript"></script>
 
 		<!-- Base Libs -->
 		<script src="js/lib/util.js?v=7f6847c" type="text/javascript"></script>
@@ -337,7 +337,7 @@
 
 		<!-- Content -->
 		<script src="js/prompt.js?v=7a1c68c" type="text/javascript"></script>
-		<script src="js/index.js?v=133b74b" type="text/javascript"></script>
+		<script src="js/index.js?v=3071414" type="text/javascript"></script>
 
 		<script
 			src="js/ui/floating/history.js?v=fc92d14"
diff --git a/js/global.js b/js/global.js
index 19a3542..72cb565 100644
--- a/js/global.js
+++ b/js/global.js
@@ -3,6 +3,11 @@
  */
 
 const global = {
+	// If this is the first run of openOutpaint
+	get firstRun() {
+		return this._firstRun;
+	},
+
 	// Connection
 	_connection: "offline",
 	set connection(v) {
@@ -46,3 +51,5 @@ const global = {
 		this.debug = !this.debug;
 	},
 };
+
+global._firstRun = !localStorage.getItem("openoutpaint/host");
diff --git a/js/index.js b/js/index.js
index fb68673..06b7285 100644
--- a/js/index.js
+++ b/js/index.js
@@ -137,7 +137,6 @@ var host = "";
 var url = "/sdapi/v1/";
 const basePixelCount = 64; //64 px - ALWAYS 64 PX
 
-//
 function startup() {
 	testHostConfiguration();
 	loadSettings();
@@ -804,12 +803,14 @@ async function getUpscalers() {
 }
 
 async function getModels() {
-	var url = document.getElementById("host").value + "/sdapi/v1/sd-models";
+	const url = document.getElementById("host").value + "/sdapi/v1/sd-models";
+	let opt = null;
+
 	try {
 		const response = await fetch(url);
 		const data = await response.json();
 
-		modelAutoComplete.options = data.map((option) => ({
+		opt = data.map((option) => ({
 			name: option.title,
 			value: option.title,
 			optionelcb: (el) => {
@@ -818,6 +819,8 @@ async function getModels() {
 			},
 		}));
 
+		modelAutoComplete.options = opt;
+
 		try {
 			const optResponse = await fetch(
 				document.getElementById("host").value + "/sdapi/v1/options"
@@ -838,10 +841,10 @@ async function getModels() {
 
 	modelAutoComplete.onchange.on(async ({value}) => {
 		console.log(`[index] Changing model to [${value}]`);
-		var payload = {
+		const payload = {
 			sd_model_checkpoint: value,
 		};
-		var url = document.getElementById("host").value + "/sdapi/v1/options/";
+		const url = document.getElementById("host").value + "/sdapi/v1/options/";
 		try {
 			await fetch(url, {
 				method: "POST",
@@ -861,6 +864,25 @@ async function getModels() {
 			);
 		}
 	});
+
+	// If first time running, ask if user wants to switch to an inpainting model
+	if (global.firstRun && !modelAutoComplete.value.includes("inpainting")) {
+		const inpainting = opt.find(({name}) => name.includes("inpainting"));
+
+		let message =
+			"It seems this is your first time using openOutpaint. It is highly recommended that you switch to an inpainting model. \
+			These are highlighted as green in the model selector.";
+
+		if (inpainting) {
+			message += `\n\nWe have found the inpainting model\n\n - ${inpainting.name}\n\navailable in the webui. Do you want to switch to it?`;
+			if (confirm(message)) {
+				modelAutoComplete.value = inpainting.value;
+			}
+		} else {
+			message += `\n\nNo inpainting model seems to be available in the webui. It is recommended that you download an inpainting model, or outpainting results may not be optimal.`;
+			alert(message);
+		}
+	}
 }
 
 async function getConfig() {