From 7b7becc34c065168be0ffab4b2b70caae13266f2 Mon Sep 17 00:00:00 2001
From: huyua9 <huyua9@outlook.com>
Date: Mon, 4 May 2026 18:26:04 +0800
Subject: [PATCH] Create custom input directory when configured

---
 folder_paths.py            |  5 +++++
 tests/test_folder_paths.py | 13 +++++++++++++
 2 files changed, 18 insertions(+)
 create mode 100644 tests/test_folder_paths.py

diff --git a/folder_paths.py b/folder_paths.py
index 80f4b291a..463d803b3 100644
--- a/folder_paths.py
+++ b/folder_paths.py
@@ -118,6 +118,11 @@ def set_temp_directory(temp_dir: str) -> None:
 def set_input_directory(input_dir: str) -> None:
     global input_directory
     input_directory = input_dir
+    if not os.path.exists(input_directory):
+        try:
+            os.makedirs(input_directory)
+        except:
+            logging.error("Failed to create input directory")
 
 def get_output_directory() -> str:
     global output_directory
diff --git a/tests/test_folder_paths.py b/tests/test_folder_paths.py
new file mode 100644
index 000000000..9f89cb41b
--- /dev/null
+++ b/tests/test_folder_paths.py
@@ -0,0 +1,13 @@
+import folder_paths
+
+
+def test_set_input_directory_creates_missing_directory(tmp_path):
+    original_input_directory = folder_paths.get_input_directory()
+    custom_input_directory = tmp_path / "custom-input"
+
+    try:
+        folder_paths.set_input_directory(str(custom_input_directory))
+
+        assert custom_input_directory.is_dir()
+    finally:
+        folder_paths.set_input_directory(original_input_directory)