File size: 2,116 Bytes
8fdc036 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
# Defines the Context class, which is used to store the state of all Blocks that are being rendered.
from __future__ import annotations
from contextvars import ContextVar
from typing import TYPE_CHECKING
if TYPE_CHECKING: # Only import for type checking (is False at runtime).
from gradio.blocks import BlockContext, Blocks, BlocksConfig
from gradio.helpers import Progress
from gradio.routes import Request
class Context:
root_block: Blocks | None = None # The current root block that holds all blocks.
block: BlockContext | None = None # The current block that children are added to.
id: int = 0 # Running id to uniquely refer to any block that gets defined
ip_address: str | None = None # The IP address of the user.
hf_token: str | None = None # The token provided when loading private HF repos
class LocalContext:
blocks: ContextVar[Blocks | None] = ContextVar("blocks", default=None)
blocks_config: ContextVar[BlocksConfig | None] = ContextVar(
"blocks_config", default=None
)
is_render: ContextVar[bool] = ContextVar("is_render", default=False)
render_block: ContextVar[BlockContext | None] = ContextVar(
"render_block", default=None
)
in_event_listener: ContextVar[bool] = ContextVar("in_event_listener", default=False)
event_id: ContextVar[str | None] = ContextVar("event_id", default=None)
request: ContextVar[Request | None] = ContextVar("request", default=None)
progress: ContextVar[Progress | None] = ContextVar("progress", default=None)
def get_render_context() -> BlockContext | None:
if LocalContext.is_render.get():
return LocalContext.render_block.get()
else:
return Context.block
def set_render_context(block: BlockContext | None):
if LocalContext.is_render.get():
LocalContext.render_block.set(block)
else:
Context.block = block
def get_blocks_context() -> BlocksConfig | None:
if LocalContext.is_render.get():
return LocalContext.blocks_config.get()
elif Context.root_block:
return Context.root_block.default_config
|