mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-07-10 17:37:17 +08:00
support upload 3d model to custom subfolder (#8597)
This commit is contained in:
parent
772de7c006
commit
79ed752748
@ -5,6 +5,8 @@ import os
|
|||||||
from comfy.comfy_types import IO
|
from comfy.comfy_types import IO
|
||||||
from comfy_api.input_impl import VideoFromFile
|
from comfy_api.input_impl import VideoFromFile
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
|
||||||
def normalize_path(path):
|
def normalize_path(path):
|
||||||
return path.replace('\\', '/')
|
return path.replace('\\', '/')
|
||||||
@ -16,7 +18,14 @@ class Load3D():
|
|||||||
|
|
||||||
os.makedirs(input_dir, exist_ok=True)
|
os.makedirs(input_dir, exist_ok=True)
|
||||||
|
|
||||||
files = [normalize_path(os.path.join("3d", f)) for f in os.listdir(input_dir) if f.endswith(('.gltf', '.glb', '.obj', '.fbx', '.stl'))]
|
input_path = Path(input_dir)
|
||||||
|
base_path = Path(folder_paths.get_input_directory())
|
||||||
|
|
||||||
|
files = [
|
||||||
|
normalize_path(str(file_path.relative_to(base_path)))
|
||||||
|
for file_path in input_path.rglob("*")
|
||||||
|
if file_path.suffix.lower() in {'.gltf', '.glb', '.obj', '.fbx', '.stl'}
|
||||||
|
]
|
||||||
|
|
||||||
return {"required": {
|
return {"required": {
|
||||||
"model_file": (sorted(files), {"file_upload": True}),
|
"model_file": (sorted(files), {"file_upload": True}),
|
||||||
@ -61,7 +70,14 @@ class Load3DAnimation():
|
|||||||
|
|
||||||
os.makedirs(input_dir, exist_ok=True)
|
os.makedirs(input_dir, exist_ok=True)
|
||||||
|
|
||||||
files = [normalize_path(os.path.join("3d", f)) for f in os.listdir(input_dir) if f.endswith(('.gltf', '.glb', '.fbx'))]
|
input_path = Path(input_dir)
|
||||||
|
base_path = Path(folder_paths.get_input_directory())
|
||||||
|
|
||||||
|
files = [
|
||||||
|
normalize_path(str(file_path.relative_to(base_path)))
|
||||||
|
for file_path in input_path.rglob("*")
|
||||||
|
if file_path.suffix.lower() in {'.gltf', '.glb', '.fbx'}
|
||||||
|
]
|
||||||
|
|
||||||
return {"required": {
|
return {"required": {
|
||||||
"model_file": (sorted(files), {"file_upload": True}),
|
"model_file": (sorted(files), {"file_upload": True}),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user