From 829c4ea27b2e288bc32712e998ec9eadef724f02 Mon Sep 17 00:00:00 2001
From: Michael Chung <6252325+ArEnSc@users.noreply.github.com>
Date: Wed, 18 Sep 2024 01:06:03 -0400
Subject: [PATCH] Fixed hack.

---
 src/KonvaApp/Engine.ts | 25 ++++++++-----------------
 1 file changed, 8 insertions(+), 17 deletions(-)

diff --git a/src/KonvaApp/Engine.ts b/src/KonvaApp/Engine.ts
index 5181dcd586..ceb0091267 100644
--- a/src/KonvaApp/Engine.ts
+++ b/src/KonvaApp/Engine.ts
@@ -36,9 +36,6 @@ export class Engine {
 
   private selectionManager: SelectionManager;
 
-  // WIL need to fix this ui issue.
-  public ranOnce: Boolean = false;
-
   // signal reference
   constructor(canvasReference: HTMLDivElement) {
     console.log("Engine Created!");
@@ -193,22 +190,16 @@ export class Engine {
 
     uiEvents.aiStylize.onRequest(async (data) => {
       console.log("Engine heard AI Stylize request: ", data);
-      //console.log(data);
-      // Hack to ensure this doesn't break because UI BUG
 
-      if (this.ranOnce === true) {
-        try {
-          await this.renderEngine.startProcessing(data);
-        } catch (error) {
-          // throw error to retry
-          uiAccess.dialogueError.show({
-            title: "Generation Error",
-            message: error.toString(),
-          });
-        }
+      try {
+        await this.renderEngine.startProcessing(data);
+      } catch (error) {
+        // throw error to retry
+        uiAccess.dialogueError.show({
+          title: "Generation Error",
+          message: error.toString(),
+        });
       }
-
-      this.ranOnce = true;
     });
 
     // TODO: You may listen to all the image toolbar events here