fix: element render for each level

Author: ObservedObserver

Home.py

@@ -53,5 +53,5 @@
     with ui.element("card", key="base_ele2") as card2:
         card2.add_child(ui.element("input", key="nst2_input", label="Value"))
         card2.add_child(ui.element("button", key="nst2_btn", text="Nest Submmit", variant="outline"))
-    card.add_child(card2)
+        card.add_child(card2)
     card.add_child(ui.element("button", key="nst_btn", text="Hello World"))

streamlit_shadcn_ui/py_components/base/context.py

@@ -0,0 +1,9 @@
+import streamlit as st
+
+def get_context():
+    if (st.session_state.get("_shadcn_ui_context", None) is None):
+        st.session_state["_shadcn_ui_context"] = {
+            "in_render": False,
+        }
+    ctx = st.session_state.get("_shadcn_ui_context")
+    return ctx

streamlit_shadcn_ui/py_components/base/element.py

@@ -1,5 +1,5 @@
 from streamlit_shadcn_ui.py_components.utils.declare import declare_component
-
+from .context import get_context
 component_func = declare_component("element_renderer", release=False)
 
 class UIElement:
@@ -8,6 +8,7 @@ def __init__(self, name: str, props=None, key=None):
         self.props = props
         self.name = name
         self.children = []
+        self.is_root = False
 
     def renderTree(self, tree):
         c = component_func(comp="element_renderer", props={
@@ -27,11 +28,19 @@ def render(self):
         return tree
 
     def __enter__(self):
+        ctx = get_context()
+        if (ctx["in_render"] is False):
+            ctx["in_render"] = True
+            self.is_root = True
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         tree = self.render()
-        return self.renderTree(tree)
+        ctx = get_context()
+        if (self.is_root):
+            ctx["in_render"] = False
+            return self.renderTree(tree)
+        return
 
     def add_child(self, child):
         self.children.append(child)