From 63e5fd17907fa2100725d6d46053084c14065232 Mon Sep 17 00:00:00 2001
From: Jairo Correa <jn.j41r0@gmail.com>
Date: Wed, 4 Oct 2023 19:45:15 -0300
Subject: [PATCH] Option to input directory
---
comfy/cli_args.py | 1 +
folder_paths.py | 4 ++++
main.py | 5 +++++
3 files changed, 10 insertions(+)
diff --git a/comfy/cli_args.py b/comfy/cli_args.py
index ffae81c4..35d44164 100644
--- a/comfy/cli_args.py
+++ b/comfy/cli_args.py
@@ -39,6 +39,7 @@ parser.add_argument("--enable-cors-header", type=str, default=None, metavar="ORI
parser.add_argument("--extra-model-paths-config", type=str, default=None, metavar="PATH", nargs='+', action='append', help="Load one or more extra_model_paths.yaml files.")
parser.add_argument("--output-directory", type=str, default=None, help="Set the ComfyUI output directory.")
parser.add_argument("--temp-directory", type=str, default=None, help="Set the ComfyUI temp directory (default is in the ComfyUI directory).")
+parser.add_argument("--input-directory", type=str, default=None, help="Set the ComfyUI input directory.")
parser.add_argument("--auto-launch", action="store_true", help="Automatically launch ComfyUI in the default browser.")
parser.add_argument("--disable-auto-launch", action="store_true", help="Disable auto launching the browser.")
parser.add_argument("--cuda-device", type=int, default=None, metavar="DEVICE_ID", help="Set the id of the cuda device this instance will use.")
diff --git a/folder_paths.py b/folder_paths.py
index 4a10c68e..898513b0 100644
--- a/folder_paths.py
+++ b/folder_paths.py
@@ -46,6 +46,10 @@ def set_temp_directory(temp_dir):
global temp_directory
temp_directory = temp_dir
+def set_input_directory(input_dir):
+ global input_directory
+ input_directory = input_dir
+
def get_output_directory():
global output_directory
return output_directory
diff --git a/main.py b/main.py
index 7c5eaee0..875ea1aa 100644
--- a/main.py
+++ b/main.py
@@ -175,6 +175,11 @@ if __name__ == "__main__":
print(f"Setting output directory to: {output_dir}")
folder_paths.set_output_directory(output_dir)
+ if args.input_directory:
+ input_dir = os.path.abspath(args.input_directory)
+ print(f"Setting input directory to: {input_dir}")
+ folder_paths.set_input_directory(input_dir)
+
if args.quick_test_for_ci:
exit(0)