From 7b72d894c5a4646c19deb2f03ba56dbd9ceacd21 Mon Sep 17 00:00:00 2001
From: Manuel Schmid <9307310+mashb1t@users.noreply.github.com>
Date: Tue, 12 Dec 2023 22:26:01 +0100
Subject: [PATCH] analytics via argument (#1315)

---
 args_manager.py | 2 ++
 webui.py        | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/args_manager.py b/args_manager.py
index 12033b85..7d2ba547 100644
--- a/args_manager.py
+++ b/args_manager.py
@@ -16,6 +16,8 @@ args_parser.parser.add_argument("--disable-offload-from-vram", action="store_tru
 args_parser.parser.add_argument("--theme", type=str, help="launches the UI with light or dark theme", default=None)
 args_parser.parser.add_argument("--disable-image-log", action='store_true',
                                 help="Prevent writing images and logs to hard drive.")
+args_parser.parser.add_argument("--disable-analytics", action='store_true', 
+                                help="Disables analytics for Gradio", default=False)
 
 args_parser.parser.set_defaults(
     disable_cuda_malloc=True,
diff --git a/webui.py b/webui.py
index ff314940..8fca171b 100644
--- a/webui.py
+++ b/webui.py
@@ -526,5 +526,6 @@ shared.gradio_root.launch(
     server_port=args_manager.args.port,
     share=args_manager.args.share,
     auth=check_auth if args_manager.args.share and auth_enabled else None,
-    blocked_paths=[constants.AUTH_FILENAME]
+    blocked_paths=[constants.AUTH_FILENAME],
+    analytics_enabled=not args_manager.args.disable_analytics
 )