add whitelist for environment in the report

add extra link to view the report instead of downloading it

1 files changed, 29 insertions, 1 deletions

diff --git a/modules/sysinfo.py b/modules/sysinfo.py
index f630c58b..5f15ac4f 100644
--- a/modules/sysinfo.py
+++ b/modules/sysinfo.py
@@ -13,6 +13,30 @@ import launch
 from modules import paths_internal, timer

 

 checksum_token = "DontStealMyGamePlz__WINNERS_DONT_USE_DRUGS__DONT_COPY_THAT_FLOPPY"

+environment_whitelist = {

+    "GIT",

+    "INDEX_URL",

+    "WEBUI_LAUNCH_LIVE_OUTPUT",

+    "GRADIO_ANALYTICS_ENABLED",

+    "PYTHONPATH",

+    "TORCH_INDEX_URL",

+    "TORCH_COMMAND",

+    "REQS_FILE",

+    "XFORMERS_PACKAGE",

+    "GFPGAN_PACKAGE",

+    "CLIP_PACKAGE",

+    "OPENCLIP_PACKAGE",

+    "STABLE_DIFFUSION_REPO",

+    "K_DIFFUSION_REPO",

+    "CODEFORMER_REPO",

+    "BLIP_REPO",

+    "STABLE_DIFFUSION_COMMIT_HASH",

+    "K_DIFFUSION_COMMIT_HASH",

+    "CODEFORMER_COMMIT_HASH",

+    "BLIP_COMMIT_HASH",

+    "COMMANDLINE_ARGS",

+    "IGNORE_CMD_ARGS_ERRORS",

+}

 

 

 def pretty_bytes(num, suffix="B"):

@@ -72,7 +96,7 @@ def get_dict():
         },

         "Extensions": get_extensions(enabled=True),

         "Inactive extensions": get_extensions(enabled=False),

-        "Environment": {k: os.environ[k] for k in sorted(os.environ)},

+        "Environment": get_environment(),

         "Config": get_config(),

         "Startup": timer.startup_record,

         "Packages": sorted([f"{pkg.key}=={pkg.version}" for pkg in pkg_resources.working_set]),

@@ -94,6 +118,10 @@ def get_exceptions():
         return str(e)

 

 

+def get_environment():

+    return {k: os.environ[k] for k in sorted(os.environ) if k in environment_whitelist}

+

+

 re_newline = re.compile(r"\r*\n")