From 7d3c868d0a67fe7cba6df412bd5f34ec8a5d289d Mon Sep 17 00:00:00 2001 From: "Yuichiro Tachibana (Tsuchiya)" Date: Mon, 18 Mar 2024 23:35:24 +0900 Subject: [PATCH] Detailed error message for wasm_utils.get_registered_app() (#7572) * Detailed error message for wasm_utils.get_registered_app() * add changeset --------- Co-authored-by: gradio-pr-bot Co-authored-by: Ali Abdalla --- .changeset/major-results-stick.md | 5 +++++ gradio/wasm_utils.py | 11 ++++++++++- 2 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 .changeset/major-results-stick.md diff --git a/.changeset/major-results-stick.md b/.changeset/major-results-stick.md new file mode 100644 index 000000000000..1b7bef898119 --- /dev/null +++ b/.changeset/major-results-stick.md @@ -0,0 +1,5 @@ +--- +"gradio": minor +--- + +feat:Detailed error message for wasm_utils.get_registered_app() diff --git a/gradio/wasm_utils.py b/gradio/wasm_utils.py index c4cba87d9a25..bd78667629a1 100644 --- a/gradio/wasm_utils.py +++ b/gradio/wasm_utils.py @@ -46,5 +46,14 @@ def register_app(_app): app_map[app_id] = _app +class GradioAppNotFoundError(Exception): + pass + + def get_registered_app(app_id: str): - return app_map[app_id] + try: + return app_map[app_id] + except KeyError as e: + raise GradioAppNotFoundError( + f"Gradio app not found (ID: {app_id}). Forgot to call demo.launch()?" + ) from e