Browse Source

add Block component creation callback

w-e-w 1 year ago
parent
commit
b30cc87b78
1 changed files with 10 additions and 0 deletions
  1. 10 0
      modules/gradio_extensons.py

+ 10 - 0
modules/gradio_extensons.py

@@ -47,10 +47,20 @@ def Block_get_config(self):
 
 
 
 
 def BlockContext_init(self, *args, **kwargs):
 def BlockContext_init(self, *args, **kwargs):
+    if scripts.scripts_current is not None:
+        scripts.scripts_current.before_component(self, **kwargs)
+
+    scripts.script_callbacks.before_component_callback(self, **kwargs)
+
     res = original_BlockContext_init(self, *args, **kwargs)
     res = original_BlockContext_init(self, *args, **kwargs)
 
 
     add_classes_to_gradio_component(self)
     add_classes_to_gradio_component(self)
 
 
+    scripts.script_callbacks.after_component_callback(self, **kwargs)
+
+    if scripts.scripts_current is not None:
+        scripts.scripts_current.after_component(self, **kwargs)
+
     return res
     return res