From f51e9568c55790bca40db21fa405ad5c82d23361 Mon Sep 17 00:00:00 2001
From: john <john@server31.io>
Date: Wed, 3 Jul 2024 15:30:36 +0800
Subject: [PATCH] =?UTF-8?q?fix=F0=9F=90=9B:=20=20save=20Token=20to=20local?=
 =?UTF-8?q?=20and=20show=20api=20Token=20page?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 js/comfyui-share-copus.js | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/js/comfyui-share-copus.js b/js/comfyui-share-copus.js
index 8b1943e6..36ff977f 100644
--- a/js/comfyui-share-copus.js
+++ b/js/comfyui-share-copus.js
@@ -1,6 +1,6 @@
 import { app } from "../../scripts/app.js";
 import { $el, ComfyDialog } from "../../scripts/ui.js";
-const env = "prod";
+const env = "dev";
 
 let DEFAULT_HOMEPAGE_URL = "https://copus.io";
 
@@ -222,7 +222,7 @@ export class CopusShareDialog extends ComfyDialog {
           ...hyperLinkStyle,
           color: "#59E8C6",
         },
-        href: DEFAULT_HOMEPAGE_URL,
+        href: `${DEFAULT_HOMEPAGE_URL}?fromPage=comfyUI`,
         target: "_blank",
       },
       ["👉 Get your API key here"]
@@ -603,6 +603,7 @@ export class CopusShareDialog extends ComfyDialog {
       );
 
      if (res.status && res.data.status && res.data) {
+      localStorage.setItem("copus_token",this.keyInput.value);
        const { data } = res.data;
        if (data) {
          const url = `${DEFAULT_HOMEPAGE_URL}/work/${data}`;
@@ -657,13 +658,13 @@ export class CopusShareDialog extends ComfyDialog {
     }
     potential_output_nodes = sorted_potential_output_nodes;
     potential_outputs = sorted_potential_outputs;
-
+    const apiToken = localStorage.getItem("copus_token");
     this.message.innerHTML = "";
     this.message.textContent = "";
     this.element.style.display = "block";
     this.previewImage.src = "";
     this.previewImage.style.display = "none";
-    this.keyInput.value = "";
+    this.keyInput.value = apiToken!=null?apiToken:"";
     this.uploadedImages = [];
     this.allFilesImages = [];
     this.allFiles = [];