simplify component template

Author: adhami3310

reflex/compiler/compiler.py

@@ -333,7 +333,7 @@ def _compile_component(component: Component | StatefulComponent) -> str:
     Returns:
         The compiled component.
     """
-    return templates.COMPONENT.render(component=component)
+    return templates.component_template(component=component)
 
 
 def _compile_components(

reflex/compiler/templates.py

@@ -13,6 +13,7 @@
 
 if TYPE_CHECKING:
     from reflex.compiler.utils import _ImportDict
+    from reflex.components.component import Component, StatefulComponent
 
 
 def _sort_hooks(hooks: dict[str, VarData | None]):
@@ -513,28 +514,16 @@ def context_template(
 }}"""
 
 
-def _component_template(**kwargs):
+def component_template(component: Component | StatefulComponent):
     """Template to render a component tag.
 
     Args:
-        **kwargs: Template context variables including component.
+        component: The component to render.
 
     Returns:
         Rendered component as string.
     """
-    component = kwargs.get("component", {})
-    # If component has a render method, call it, otherwise use it as-is
-    rendered_data = component.render() if hasattr(component, "render") else component
-    result = _RenderUtils.render(rendered_data)
-
-    # Add trailing newline for HTML elements (those with quoted tag names)
-    # to match original Jinja behavior
-    if isinstance(rendered_data, dict) and rendered_data.get("name", "").startswith(
-        '"'
-    ):
-        result += "\n"
-
-    return result
+    return _RenderUtils.render(component.render())
 
 
 def page_template(
@@ -823,9 +812,6 @@ def render(self, **kwargs) -> str:
         return self.func(**kwargs)
 
 
-# Template to render a component tag.
-COMPONENT = TemplateFunction(_component_template)
-
 # Code to render Component instances as part of StatefulComponent
 STATEFUL_COMPONENT = TemplateFunction(_stateful_component_template)
 

tests/units/components/core/test_html.py

@@ -19,7 +19,7 @@ def test_html_create():
     assert str(html.dangerouslySetInnerHTML) == '({ ["__html"] : "<p>Hello !</p>" })'  # pyright: ignore [reportAttributeAccessIssue]
     assert (
         str(html)
-        == 'jsx("div",{className:"rx-Html",dangerouslySetInnerHTML:({ ["__html"] : "<p>Hello !</p>" })},)\n'
+        == 'jsx("div",{className:"rx-Html",dangerouslySetInnerHTML:({ ["__html"] : "<p>Hello !</p>" })},)'
     )
 
 
@@ -39,5 +39,5 @@ class TestState(State):
     )
     assert (
         str(html)
-        == f'jsx("div",{{className:"rx-Html",dangerouslySetInnerHTML:{html_dangerouslySetInnerHTML!s}}},)\n'
+        == f'jsx("div",{{className:"rx-Html",dangerouslySetInnerHTML:{html_dangerouslySetInnerHTML!s}}},)'
     )