_apply_common_imports

Author: adhami3310

reflex/compiler/compiler.py

@@ -30,6 +30,13 @@
 from reflex.vars.base import LiteralVar, Var
 
 
+def _apply_common_imports(
+    imports: dict[str, list[ImportVar]],
+):
+    imports.setdefault("@emotion/react", []).append(ImportVar("jsx"))
+    imports.setdefault("react", []).append(ImportVar("Fragment"))
+
+
 def _compile_document_root(root: Component) -> str:
     """Compile the document root.
 
@@ -40,7 +47,7 @@ def _compile_document_root(root: Component) -> str:
         The compiled document root.
     """
     document_root_imports = root._get_all_imports()
-    document_root_imports.setdefault("@emotion/react", []).append(ImportVar("jsx"))
+    _apply_common_imports(document_root_imports)
     return templates.DOCUMENT_ROOT.render(
         imports=utils.compile_imports(document_root_imports),
         document=root.render(),
@@ -77,7 +84,7 @@ def _compile_app(app_root: Component) -> str:
     ]
 
     app_root_imports = app_root._get_all_imports()
-    app_root_imports.setdefault("@emotion/react", []).append(ImportVar("jsx"))
+    _apply_common_imports(app_root_imports)
 
     return templates.APP_ROOT.render(
         imports=utils.compile_imports(app_root_imports),
@@ -148,7 +155,7 @@ def _compile_page(
         The compiled component.
     """
     imports = component._get_all_imports()
-    imports.setdefault("@emotion/react", []).append(ImportVar("jsx"))
+    _apply_common_imports(imports)
     imports = utils.compile_imports(imports)
 
     # Compile the code to render the component.
@@ -341,7 +348,7 @@ def _compile_components(
         component_renders.append(component_render)
         imports = utils.merge_imports(imports, component_imports)
 
-    imports.setdefault("@emotion/react", []).append(ImportVar("jsx"))
+    _apply_common_imports(imports)
 
     dynamic_imports = {
         comp_import: None
@@ -436,7 +443,7 @@ def get_shared_components_recursive(component: BaseComponent):
         f"$/{constants.Dirs.UTILS}/{constants.PageNames.STATEFUL_COMPONENTS}", None
     )
     if rendered_components:
-        all_imports.setdefault("@emotion/react", []).append(ImportVar("jsx"))
+        _apply_common_imports(all_imports)
 
     return templates.STATEFUL_COMPONENTS.render(
         imports=utils.compile_imports(all_imports),