From 61dcdf18de7867a551ddf9d16d1aab1e63a0af0e Mon Sep 17 00:00:00 2001
From: XuanYu-github <XuanYu-github@users.noreply.github.com>
Date: Mon, 29 Dec 2025 11:13:44 +0800
Subject: [PATCH] Update node names and add requirements

---
 __init__.py                | 16 ++++++++--------
 requirements.txt           |  2 ++
 web/js/gaussian_preview.js |  4 ++--
 3 files changed, 12 insertions(+), 10 deletions(-)
 create mode 100644 requirements.txt

diff --git a/__init__.py b/__init__.py
index b360027..68a9b76 100644
--- a/__init__.py
+++ b/__init__.py
@@ -11,17 +11,17 @@ from .preview_gaussian import PreviewGaussianNode
 WEB_DIRECTORY = "web"
 
 NODE_CLASS_MAPPINGS = {
-    "PlyPreviewLoadGaussianPLY": LoadGaussianPLY,
-    "PlyPreviewLoadGaussianPLYPath": LoadGaussianPLYPath,
-    "PlyPreviewProcessGaussianPLY": ProcessGaussianPLY,
-    "PlyPreviewPreviewGaussian": PreviewGaussianNode,
+    "PlyPreviewLoadGaussianPLYEnhance": LoadGaussianPLY,
+    "PlyPreviewLoadGaussianPLYPathEnhance": LoadGaussianPLYPath,
+    "PlyPreviewProcessGaussianPLYEnhance": ProcessGaussianPLY,
+    "PlyPreviewPreviewGaussianEnhance": PreviewGaussianNode,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "PlyPreviewLoadGaussianPLY": "Load Gaussian PLY",
-    "PlyPreviewLoadGaussianPLYPath": "Load Gaussian PLY (Path)",
-    "PlyPreviewProcessGaussianPLY": "Process Gaussian PLY",
-    "PlyPreviewPreviewGaussian": "Preview Gaussian",
+    "PlyPreviewLoadGaussianPLYEnhance": "Load Gaussian PLY Enhance",
+    "PlyPreviewLoadGaussianPLYPathEnhance": "Load Gaussian PLY (Path) Enhance",
+    "PlyPreviewProcessGaussianPLYEnhance": "Process Gaussian PLY Enhance",
+    "PlyPreviewPreviewGaussianEnhance": "Preview Gaussian Enhance",
 }
 
 __all__ = [
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..690e2e8
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,2 @@
+numpy>=1.23
+plyfile>=1.0.3
diff --git a/web/js/gaussian_preview.js b/web/js/gaussian_preview.js
index 60cce29..3ddb4b5 100644
--- a/web/js/gaussian_preview.js
+++ b/web/js/gaussian_preview.js
@@ -18,8 +18,8 @@ app.registerExtension({
     name: "plypreview.gaussianpreview",
 
     async beforeRegisterNodeDef(nodeType, nodeData, app) {
-        if (nodeData.name === "PlyPreviewPreviewGaussian") {
-            console.log("[PlyPreview Gaussian] Registering Preview Gaussian node");
+        if (nodeData.name === "PlyPreviewPreviewGaussianEnhance") {
+            console.log("[PlyPreview Gaussian] Registering Preview Gaussian Enhance node");
 
             const onNodeCreated = nodeType.prototype.onNodeCreated;
             nodeType.prototype.onNodeCreated = function() {