Revert "Use ContextVar instead of threading.local()"

This reverts commit 2a76eb4.

gradio/blocks.py

@@ -91,9 +91,9 @@
 
 
 def in_event_listener():
-    from gradio.context import LocalContext
+    from gradio import context
 
-    return LocalContext.in_event_listener.get()
+    return getattr(context.thread_data, "in_event_listener", False)
 
 
 def updateable(fn):

gradio/context.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from contextvars import ContextVar
+import threading
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:  # Only import for type checking (is False at runtime).
@@ -17,7 +17,4 @@ class Context:
     hf_token: str | None = None  # The token provided when loading private HF repos
 
 
-class LocalContext:
-    blocks: ContextVar[Blocks | None] = ContextVar("blocks", default=None)
-    in_event_listener: ContextVar[bool] = ContextVar("in_event_listener", default=False)
-    event_id: ContextVar[str | None] = ContextVar("event_id", default=None)
+thread_data = threading.local()

gradio/helpers.py

@@ -1098,23 +1098,22 @@ def __init__(self, target: Block | None, _data: Any):
 
 
 def log_message(message: str, level: Literal["info", "warning"] = "info"):
-    from gradio.context import LocalContext
+    from gradio import context
 
-    blocks = LocalContext.blocks.get()
-    if blocks is None:  # Function called outside of Gradio
+    if not hasattr(context.thread_data, "blocks"):  # Function called outside of Gradio
         if level == "info":
             print(message)
         elif level == "warning":
             warnings.warn(message)
         return
-    if not blocks.enable_queue:
+    if not context.thread_data.blocks.enable_queue:
         warnings.warn(
             f"Queueing must be enabled to issue {level.capitalize()}: '{message}'."
         )
         return
-    event_id = LocalContext.event_id.get()
-    assert event_id
-    blocks._queue.log_message(event_id=event_id, log=message, level=level)
+    context.thread_data.blocks._queue.log_message(
+        event_id=context.thread_data.event_id, log=message, level=level
+    )
 
 
 @document()

gradio/utils.py

@@ -663,16 +663,16 @@ def get_function_with_locals(
     fn: Callable, blocks: Blocks, event_id: str | None, in_event_listener: bool
 ):
     def before_fn(blocks, event_id):
-        from gradio.context import LocalContext
+        from gradio.context import thread_data
 
-        LocalContext.blocks.set(blocks)
-        LocalContext.in_event_listener.set(in_event_listener)
-        LocalContext.event_id.set(event_id)
+        thread_data.blocks = blocks
+        thread_data.in_event_listener = in_event_listener
+        thread_data.event_id = event_id
 
     def after_fn():
-        from gradio.context import LocalContext
+        from gradio.context import thread_data
 
-        LocalContext.in_event_listener.set(False)
+        thread_data.in_event_listener = False
 
     return function_wrapper(
         fn, before_fn=before_fn, before_args=(blocks, event_id), after_fn=after_fn