Surpress Startup Logs (#319)

* Add code

* code
This commit is contained in:
Freddy Boulton
2025-05-20 12:30:36 -04:00
committed by GitHub
parent ae95e973f6
commit c191f1ce90
2 changed files with 2618 additions and 2607 deletions

View File

@@ -109,6 +109,7 @@ class Stream(WebRTCConnectionMixin):
additional_inputs: list[Component] | None = None, additional_inputs: list[Component] | None = None,
additional_outputs: list[Component] | None = None, additional_outputs: list[Component] | None = None,
ui_args: UIArgs | None = None, ui_args: UIArgs | None = None,
verbose: bool = True,
): ):
""" """
Initialize the Stream instance. Initialize the Stream instance.
@@ -130,6 +131,7 @@ class Stream(WebRTCConnectionMixin):
additional_inputs: Optional list of extra Gradio input components. additional_inputs: Optional list of extra Gradio input components.
additional_outputs: Optional list of extra Gradio output components. Requires `additional_outputs_handler`. additional_outputs: Optional list of extra Gradio output components. Requires `additional_outputs_handler`.
ui_args: Optional dictionary to customize the default UI appearance (title, subtitle, icon, etc.). ui_args: Optional dictionary to customize the default UI appearance (title, subtitle, icon, etc.).
verbose: Whether to print verbose logging on startup.
Raises: Raises:
ValueError: If `additional_outputs` are provided without `additional_outputs_handler`. ValueError: If `additional_outputs` are provided without `additional_outputs_handler`.
@@ -157,6 +159,7 @@ class Stream(WebRTCConnectionMixin):
self.server_rtc_configuration = self.convert_to_aiortc_format( self.server_rtc_configuration = self.convert_to_aiortc_format(
server_rtc_configuration server_rtc_configuration
) )
self.verbose = verbose
self._ui = self._generate_default_ui(ui_args) self._ui = self._generate_default_ui(ui_args)
self._ui.launch = self._wrap_gradio_launch(self._ui.launch) self._ui.launch = self._wrap_gradio_launch(self._ui.launch)
@@ -280,12 +283,13 @@ class Stream(WebRTCConnectionMixin):
def print_startup_message(): def print_startup_message():
self._check_colab_or_spaces() self._check_colab_or_spaces()
print( if self.verbose:
click.style("INFO", fg="green") print(
+ ":\t Visit " click.style("INFO", fg="green")
+ click.style("https://fastrtc.org/userguide/api/", fg="cyan") + ":\t Visit "
+ " for WebRTC or Websocket API docs." + click.style("https://fastrtc.org/userguide/api/", fg="cyan")
) + " for WebRTC or Websocket API docs."
)
@contextlib.asynccontextmanager @contextlib.asynccontextmanager
async def new_lifespan(app: FastAPI): async def new_lifespan(app: FastAPI):

File diff suppressed because one or more lines are too long