Upload adetailer_patch.py

RuinedFooocus/modules/adetailer_patch.py

@@ -0,0 +1,69 @@
+"""
+Monkey patch for integrating ADetailer into RuinedFooocus. This module patches the process function from modules.sdxl_pipeline so that after the original image generation, ADetailer can optionally refine the results if the `use_adetailer` flag is enabled in settings or in the gen_data dictionary. This patch is loaded by importing modules.adetailer_patch in settings.py.
+"""
+from modules import settings
+from modules import sdxl_pipeline
+import numpy as np
+
+try:
+    import adetailer
+except Exception:
+    adetailer = None
+
+# Save reference to original process
+_original_process = sdxl_pipeline.process
+
+
+def patched_process(*args, **kwargs):
+    """Wrapper for sdxl_pipeline.process that applies ADetailer if enabled."""
+    # Call original sdxl pipeline process to generate images (list of numpy arrays)
+    images = _original_process(*args, **kwargs)
+    # Determine whether to use ADetailer
+    use_flag = False
+    gen_data = None
+    if args:
+        gen_data = args[0]
+    # Check if gen_data is a dict or object with use_adetailer flag
+    try:
+        if isinstance(gen_data, dict) and gen_data.get("use_adetailer") is not None:
+            use_flag = bool(gen_data.get("use_adetailer"))
+        elif hasattr(gen_data, "use_adetailer"):
+            use_flag = bool(getattr(gen_data, "use_adetailer"))
+    except Exception:
+        pass
+    # Fall back to global settings
+    if not use_flag:
+        try:
+            use_flag = settings.get_setting("use_adetailer")
+        except Exception:
+            use_flag = False
+    # If enabled and adetailer available, process each image
+    if use_flag and adetailer is not None:
+        processed = []
+        for img in images:
+            try:
+                from PIL import Image
+                pil_img = Image.fromarray(img)
+                result = pil_img
+                # Attempt various functions to run adetailer
+                try:
+                    if hasattr(adetailer, "run"):
+                        result = adetailer.run(pil_img)
+                    elif hasattr(adetailer, "process"):
+                        result = adetailer.process(pil_img)
+                    else:
+                        from adetailer import scripts  # type: ignore
+                        if hasattr(scripts, "ladetailer"):
+                            result = scripts.ladetailer.process(pil_img)  # type: ignore
+                except Exception:
+                    result = pil_img
+                processed.append(np.array(result))
+            except Exception as e:
+                print("ADetailer processing error:", e)
+                processed.append(img)
+        images = processed
+    return images
+
+# Apply patch
+sdxl_pipeline.process = patched_process
+print("ADetailer patch loaded: sdxl_pipeline.process has been patched.")