From 2846bbf5ccba97c91221a9ca3260a7d26ad26c39 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 4 Feb 2025 22:29:44 +0200
Subject: [PATCH] workaround to not always save the mesh for preview

---
 nodes.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index 0dc4027..4854417 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1385,6 +1385,9 @@ class Hy3DExportMesh:
                 "filename_prefix": ("STRING", {"default": "3D/Hy3D"}),
                 "file_format": (["glb", "obj", "ply", "stl", "3mf", "dae"],),
             },
+            "optional": {
+                "save_file": ("BOOLEAN", {"default": True}),
+            },
         }
 
     RETURN_TYPES = ("STRING",)
@@ -1392,13 +1395,17 @@ class Hy3DExportMesh:
     FUNCTION = "process"
     CATEGORY = "Hunyuan3DWrapper"
 
-    def process(self, mesh, filename_prefix, file_format):
+    def process(self, mesh, filename_prefix, file_format, save_file=True):
         full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, folder_paths.get_output_directory())
         output_glb_path = Path(full_output_folder, f'{filename}_{counter:05}_.{file_format}')
         output_glb_path.parent.mkdir(exist_ok=True)
-        mesh.export(output_glb_path, file_type=file_format)
-
-        relative_path = Path(subfolder) / f'{filename}_{counter:05}_.{file_format}'
+        if save_file:
+            mesh.export(output_glb_path, file_type=file_format)
+            relative_path = Path(subfolder) / f'{filename}_{counter:05}_.{file_format}'
+        else:
+            temp_file = Path(full_output_folder, f'hy3dtemp_.{file_format}')
+            mesh.export(temp_file, file_type=file_format)
+            relative_path = Path(subfolder) / f'hy3dtemp_.{file_format}'
         
         return (str(relative_path), )