From 3b97e49dd814a43fc75a26b2495603e38b8b2f62 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Wed, 15 Nov 2023 13:23:27 -0800
Subject: [PATCH] --disable-image-log

---
 args_manager.py           | 2 ++
 fooocus_version.py        | 2 +-
 modules/private_logger.py | 4 ++++
 update_log.md             | 4 ++++
 webui.py                  | 3 ++-
 5 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/args_manager.py b/args_manager.py
index b5768ac..38b182b 100644
--- a/args_manager.py
+++ b/args_manager.py
@@ -16,6 +16,8 @@ fcbh_cli.parser.add_argument("--enable-smart-memory", action="store_true",
                                   "Some Mac users may need this.")
 
 fcbh_cli.parser.add_argument("--theme", type=str, help="launches the UI with light or dark theme", default=None)
+fcbh_cli.parser.add_argument("--disable-image-log", action='store_true',
+                             help="Prevent writing images and logs to hard drive.")
 
 fcbh_cli.parser.set_defaults(
     disable_cuda_malloc=True,
diff --git a/fooocus_version.py b/fooocus_version.py
index 6acc275..75462ff 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1 +1 @@
-version = '2.1.819'
+version = '2.1.820'
diff --git a/modules/private_logger.py b/modules/private_logger.py
index de40917..ee09f98 100644
--- a/modules/private_logger.py
+++ b/modules/private_logger.py
@@ -1,4 +1,5 @@
 import os
+import args_manager
 import modules.config
 
 from PIL import Image
@@ -16,6 +17,9 @@ def get_current_html_path():
 
 
 def log(img, dic, single_line_number=3):
+    if args_manager.args.disable_image_log:
+        return
+
     date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, extension='png')
     os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
     Image.fromarray(img).save(local_temp_filename)
diff --git a/update_log.md b/update_log.md
index 57b1865..b0e4b68 100644
--- a/update_log.md
+++ b/update_log.md
@@ -1,3 +1,7 @@
+# 2.1.820
+
+* support "--disable-image-log" to prevent writing images and logs to hard drive.
+
 # 2.1.819
 
 * Allow disabling preview in dev tools.
diff --git a/webui.py b/webui.py
index 8e106e1..59a5fa7 100644
--- a/webui.py
+++ b/webui.py
@@ -222,7 +222,8 @@ with shared.gradio_root:
 
                 seed_random.change(random_checked, inputs=[seed_random], outputs=[image_seed], queue=False)
 
-                gr.HTML(f'<a href="/file={get_current_html_path()}" target="_blank">\U0001F4DA History Log</a>')
+                if not args_manager.args.disable_image_log:
+                    gr.HTML(f'<a href="/file={get_current_html_path()}" target="_blank">\U0001F4DA History Log</a>')
 
             with gr.Tab(label='Style'):
                 style_sorter.try_load_sorted_styles(