From 5f1ca4a27759a37d76b8f391d1ca5cd5669098fe Mon Sep 17 00:00:00 2001
From: Bingsu <ks2515@naver.com>
Date: Sat, 1 Jul 2023 15:09:27 +0900
Subject: [PATCH] feat: update traceback info

---
 adetailer/traceback.py | 33 ++++++++++++++++++++++++++++++---
 1 file changed, 30 insertions(+), 3 deletions(-)

diff --git a/adetailer/traceback.py b/adetailer/traceback.py
index 2348152..3ebc444 100644
--- a/adetailer/traceback.py
+++ b/adetailer/traceback.py
@@ -8,6 +8,8 @@ from rich.panel import Panel
 from rich.table import Table
 from rich.traceback import Traceback
 
+from adetailer.__version__ import __version__
+
 
 def processing(*args):
     try:
@@ -29,14 +31,34 @@ def processing(*args):
     if not p:
         return {}
 
-    return {
+    info = {
         "prompt": p.prompt,
         "negative_prompt": p.negative_prompt,
         "n_iter": p.n_iter,
         "batch_size": p.batch_size,
         "width": p.width,
         "height": p.height,
+        "sampler_name": p.sampler_name,
         "enable_hr": getattr(p, "enable_hr", False),
+        "hr_upscaler": getattr(p, "hr_upscaler", ""),
+    }
+
+    info.update(sd_models())
+    return info
+
+
+def sd_models():
+    try:
+        from modules import shared
+
+        opts = shared.opts
+    except Exception:
+        return {}
+
+    return {
+        "checkpoint": getattr(opts, "sd_model_checkpoint", "------"),
+        "vae": getattr(opts, "sd_vae", "------"),
+        "unet": getattr(opts, "sd_unet", "------"),
     }
 
 
@@ -51,6 +73,7 @@ def ad_args(*args):
 
     arg0 = args[0]
     return {
+        "version": __version__,
         "ad_model": arg0["ad_model"],
         "ad_prompt": arg0.get("ad_prompt", ""),
         "ad_negative_prompt": arg0.get("ad_negative_prompt", ""),
@@ -60,6 +83,7 @@ def ad_args(*args):
 
 def sys_info():
     import platform
+    import sys
 
     try:
         import launch
@@ -67,18 +91,21 @@ def sys_info():
         version = launch.git_tag()
         commit = launch.commit_hash()
     except Exception:
-        return {}
+        version = commit = "------"
 
     return {
         "Platform": platform.platform(),
-        "Python": platform.python_version(),
+        "Python": sys.version,
         "Version": version,
         "Commit": commit,
+        "Commandline": sys.argv,
     }
 
 
 def get_table(title: str, data: dict[str, Any]) -> Table:
     table = Table(title=title, highlight=True)
+    table.add_column(" ", style="dim")
+    table.add_column("Value")
     for key, value in data.items():
         if not isinstance(value, str):
             value = repr(value)