From 3d534f36d5c63cdfbbd87179f174a6c9e7e4c8b2 Mon Sep 17 00:00:00 2001
From: LEv145 <40071054+LEv145@users.noreply.github.com>
Date: Tue, 20 Aug 2024 00:07:44 +0200
Subject: [PATCH] Update Gradio 4 support

Now the `value` parameter is passed as a positional argument (https://github.com/gradio-app/gradio/blob/04b7d327ec1227a693fc2dfea51b1e2729851bde/gradio/components/button.py#L27C9-L27C11)
Now the `__module__` patch is only used in Gradio 4
---
 scripts/sd-webui-ar.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/scripts/sd-webui-ar.py b/scripts/sd-webui-ar.py
index 9599d14..fc39eb0 100644
--- a/scripts/sd-webui-ar.py
+++ b/scripts/sd-webui-ar.py
@@ -25,7 +25,7 @@ is_reverse_logic_mode = False  # FIXME: Global value
 
 
 class ResButton(ToolButton):
-    def __init__(self, res=(512, 512), **kwargs):
+    def __init__(self, value, *, res=(512, 512), **kwargs):
         super().__init__(**kwargs)
 
         self.w, self.h = res
@@ -35,7 +35,7 @@ class ResButton(ToolButton):
 
 
 class ARButton(ToolButton):
-    def __init__(self, ar=1.0, **kwargs):
+    def __init__(self, value, *, ar=1.0, **kwargs):
         super().__init__(**kwargs)
 
         self.ar = ar
@@ -263,7 +263,7 @@ class AspectRatioScript(scripts.Script):
 
                 # Aspect Ratio buttons
                 ar_btns = [
-                    ARButton(ar=ar, value=label)
+                    ARButton(label, ar=ar)
                     for ar, label in zip(
                         self.aspect_ratios,
                         self.aspect_ratio_labels,
@@ -302,7 +302,7 @@ class AspectRatioScript(scripts.Script):
                 )
 
                 btns = [
-                    ResButton(res=res, value=label)
+                    ResButton(label, res=res)
                     for res, label in zip(self.res, self.res_labels)
                 ]
                 with contextlib.suppress(AttributeError):
@@ -591,6 +591,6 @@ def round_to_multiple(x, multiple):
 # TODO: Replace inheritance with association
 # https://github.com/LEv145/--sd-webui-ar-plus/issues/24
 # Hack for Forge with Gradio 4.0; see `get_component_class_id` in `venv/lib/site-packages/gradio/components/base.py`
-for class_ in (ResButton, ARButton):
-    class_.__module__ = "modules.ui_components"
-
+if IS_GRADIO_4:
+    for class_ in (ResButton, ARButton):
+        class_.__module__ = "modules.ui_components"