mirror of
https://github.com/Ladebeze66/projetcbaollm.git
synced 2025-12-16 20:47:52 +01:00
56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
# 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.renderable import Renderable
|
|
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
|
|
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
|
|
)
|
|
renderable: ContextVar[Renderable | None] = ContextVar("renderable", default=None)
|
|
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.renderable.get():
|
|
return LocalContext.render_block.get()
|
|
else:
|
|
return Context.block
|
|
|
|
|
|
def set_render_context(block: BlockContext | None):
|
|
if LocalContext.renderable.get():
|
|
LocalContext.render_block.set(block)
|
|
else:
|
|
Context.block = block
|
|
|
|
|
|
def get_blocks_context() -> BlocksConfig | None:
|
|
if LocalContext.renderable.get():
|
|
return LocalContext.blocks_config.get()
|
|
elif Context.root_block:
|
|
return Context.root_block.default_config
|