return a Dependency instance from Blocks.load event listener

CHANGELOG.md

@@ -2,7 +2,7 @@
 
 ## New Features:
 
-No changes to highlight.
+* Make `Blocks.load` behave like other event listeners (allows chaining `then` off of it) [@anentropic](https://github.com/anentropic/) in [PR 4304](https://github.com/gradio-app/gradio/pull/4304)
 
 ## Bug Fixes:
 

gradio/blocks.py

@@ -1482,7 +1482,9 @@ def get_time():
                 name=name, src=src, hf_token=api_key, alias=alias, **kwargs
             )
         else:
-            return self_or_cls.set_event_trigger(
+            from gradio.events import Dependency
+
+            dep, dep_index = self_or_cls.set_event_trigger(
                 event_name="load",
                 fn=fn,
                 inputs=inputs,
@@ -1498,7 +1500,8 @@ def get_time():
                 max_batch_size=max_batch_size,
                 every=every,
                 no_target=True,
-            )[0]
+            )
+            return Dependency(self_or_cls, dep, dep_index)
 
     def clear(self):
         """Resets the layout of the Blocks object."""

test/test_events.py

@@ -1,8 +1,12 @@
+import os
+
 import pytest
 from fastapi.testclient import TestClient
 
 import gradio as gr
 
+os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
+
 
 class TestEvent:
     def test_clear_event(self):
@@ -69,6 +73,25 @@ def clear():
         assert not parent.config["dependencies"][2]["trigger_only_on_success"]
         assert parent.config["dependencies"][3]["trigger_only_on_success"]
 
+    def test_load_chaining(self):
+        calls = 0
+
+        def increment():
+            calls += 1
+            return str(calls)
+
+        with gr.Blocks() as demo:
+            out = gr.Textbox(label="Call counter")
+            demo.load(increment, inputs=None, outputs=out).then(
+                increment, inputs=None, outputs=out
+            )
+
+        assert demo.config["dependencies"][0]["trigger"] == "load"
+        assert demo.config["dependencies"][0]["trigger_after"] is None
+        assert demo.config["dependencies"][1]["trigger"] == "then"
+        assert demo.config["dependencies"][1]["trigger_after"] == 0
+        demo.launch(prevent_thread_lock=True)
+
 
 class TestEventErrors:
     def test_event_defined_invalid_scope(self):