From c2bfa6f6a3fc59ccb3b32082336a1c08e73c2557 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Fri, 26 Jul 2024 13:08:15 -0700
Subject: [PATCH] fix gradio

---
 modules_forge/forge_canvas/canvas.py | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/modules_forge/forge_canvas/canvas.py b/modules_forge/forge_canvas/canvas.py
index d7161874..39efd5c5 100644
--- a/modules_forge/forge_canvas/canvas.py
+++ b/modules_forge/forge_canvas/canvas.py
@@ -3,6 +3,21 @@
 # by lllyasviel
 # Commercial Use is not allowed. (Contact us for commercial use.)
 
+import gradio.component_meta
+
+
+create_or_modify_pyi_org = gradio.component_meta.create_or_modify_pyi
+
+
+def create_or_modify_pyi_org_patched(component_class, class_name, events):
+    if component_class.__name__ == 'LogicalImage':
+        return
+
+    return create_or_modify_pyi_org(component_class, class_name, events)
+
+
+gradio.component_meta.create_or_modify_pyi = create_or_modify_pyi_org_patched
+
 
 import os
 import uuid
@@ -16,13 +31,6 @@ from gradio.context import Context
 from functools import wraps
 
 
-class FormComponent:
-    webui_do_not_create_gradio_pyi_thank_you = True
-
-    def get_expected_parent(self):
-        return gr.components.Form
-
-
 canvas_js_root_path = os.path.dirname(__file__)
 
 
@@ -63,7 +71,7 @@ def base64_to_image(base64_str, numpy=True):
     return image_array
 
 
-class LogicalImage(gr.Textbox, FormComponent):
+class LogicalImage(gr.Textbox):
     @wraps(gr.Textbox.__init__)
     def __init__(self, *args, numpy=True, **kwargs):
         self.numpy = numpy