feat: add pygwalker web api tips in streamlit

Author: longxiaofei

pygwalker/__init__.py

@@ -10,7 +10,7 @@
 from pygwalker.services.global_var import GlobalVarManager
 from pygwalker.services.kaggle import show_tips_user_kaggle as __show_tips_user_kaggle
 
-__version__ = "0.4.9"
+__version__ = "0.4.9.1"
 __hash__ = __rand_str()
 
 from pygwalker.api.jupyter import walk, render, table

pygwalker/communications/streamlit_comm.py

@@ -8,6 +8,7 @@
 import streamlit as st
 
 from pygwalker.utils.encode import DataFrameEncoder
+from pygwalker.errors import StreamlitPygwalkerApiError
 from .base import BaseCommunication
 
 streamlit_comm_map = {}
@@ -46,11 +47,16 @@ def post(self, gid: str):
 def hack_streamlit_server():
     tornado_obj = None
     for obj in gc.get_objects():
-        if isinstance(obj, Application):
-            tornado_obj = obj
-            break
+        try:
+            if isinstance(obj, Application) and any(("streamlit" in str(rule.target) for rule in obj.wildcard_router.rules)):
+                tornado_obj = obj
+        except Exception:
+            pass
 
-    tornado_obj.add_handlers(".*", [(PYGWALKER_API_PATH, PygwalkerHandler)])
+    if tornado_obj:
+        tornado_obj.add_handlers(".*", [(PYGWALKER_API_PATH, PygwalkerHandler)])
+    else:
+        raise StreamlitPygwalkerApiError()
 
 
 class StreamlitCommunication(BaseCommunication):

pygwalker/errors.py

@@ -50,3 +50,12 @@ def __init__(self) -> None:
             "The query returned too many data entries, making it difficult for the frontend to render. Please adjust your chart configuration and try again.",
             code=ErrorCode.UNKNOWN_ERROR
         )
+
+
+class StreamlitPygwalkerApiError(BaseError):
+    """Raised when the config is invalid."""
+    def __init__(self) -> None:
+        super().__init__(
+            "Adding pygwalker web api to streamlit failed. If possible, please report this case to the pygwalker team. Thanks!",
+            code=ErrorCode.UNKNOWN_ERROR
+        )