""" Predefined buttons with bound events that can be included in a gr.Blocks for convenience. """ from __future__ import annotations import json from typing import Literal from gradio_client.documentation import document, set_documentation_group from gradio.components import Button, Component set_documentation_group("component") @document("add") class ClearButton(Button): """ Button that clears the value of a component or a list of components when clicked. It is instantiated with the list of components to clear. Preprocessing: passes the button value as a {str} into the function Postprocessing: expects a {str} to be returned from a function, which is set as the label of the button """ is_template = True def __init__( self, components: None | list[Component] | Component = None, *, value: str = "Clear", variant: Literal["primary", "secondary", "stop"] = "secondary", size: Literal["sm", "lg"] | None = None, icon: str | None = None, link: str | None = None, visible: bool = True, interactive: bool = True, elem_id: str | None = None, elem_classes: list[str] | str | None = None, scale: int | None = None, min_width: int | None = None, **kwargs, ): super().__init__( value, variant=variant, size=size, icon=icon, link=link, visible=visible, interactive=interactive, elem_id=elem_id, elem_classes=elem_classes, scale=scale, min_width=min_width, **kwargs, ) self.add(components) def add(self, components: None | Component | list[Component]) -> ClearButton: """ Adds a component or list of components to the list of components that will be cleared when the button is clicked. """ if not components: # This needs to be here because when the ClearButton is created in an gr.Interface, we don't # want to create dependencies for it before we have created the dependencies for the submit function. # We generally assume that the submit function dependency is the first thing created in an gr.Interface. return self if isinstance(components, Component): components = [components] clear_values = json.dumps( [component.postprocess(None) for component in components] ) self.click(None, [], components, _js=f"() => {clear_values}") return self