refactor: remove unnecessary code

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-deploy"
 description = "Open source comfyui deployment platform, a vercel for generative workflow infra."
-version = "2.3.6"
+version = "2.3.4"
 license = { file = "LICENSE" }
 dependencies = ["aiofiles", "pydantic", "opencv-python", "imageio-ffmpeg", "tabulate", "brotli"]
 

web-plugin/index.js

@@ -10,6 +10,7 @@ import {
 } from "./workflow-list.js";
 import { initializeMachineManager } from "./machine-manager.js";
 import { initializeModelManager } from "./model-manager.js";
+import { fetchSnapshotSimple } from "./snapshot-utils.js";
 
 const styles = `
 .comfydeploy-menu-item {
@@ -1354,6 +1355,19 @@ async function deployWorkflow() {
   if (endpoint.endsWith("/")) {
     endpoint = endpoint.slice(0, -1);
   }
+  loadingDialog.showLoading("Generating snapshot");
+
+  const snapshot = await fetchSnapshotSimple();
+  console.log(snapshot);
+  loadingDialog.close();
+
+  if (!snapshot) {
+    showError(
+      "Error when deploying",
+      "Unable to generate snapshot, please install ComfyUI Manager"
+    );
+    return;
+  }
 
   const title = deploy.querySelector("#button-title");
 

web-plugin/machine-manager.js

@@ -1,7 +1,7 @@
 // Machine Manager Script
 // Handles local storage machine ID functionality
 
-import { fetchSnapshot } from "./snapshot-utils.js";
+import { fetchSnapshot, fetchSnapshotSimple } from "./snapshot-utils.js";
 
 const MACHINE_STORAGE_KEY = "comfy_deploy_machine_id";
 

web-plugin/snapshot-utils.js

@@ -66,3 +66,17 @@ export async function fetchSnapshot(getDataFn = null) {
     throw error;
   }
 }
+
+/**
+ * Simple snapshot fetch without ComfyUI version fallback
+ * Use this when you don't need the ComfyUI version fallback logic
+ *
+ * @returns {Promise<Object>} - The snapshot data as-is
+ */
+export async function fetchSnapshotSimple() {
+  const response = await fetch("/snapshot/get_current");
+  if (!response.ok) {
+    throw new Error(`Snapshot fetch failed: ${response.status}`);
+  }
+  return response.json();
+}