From f1005e774e15b7aaaa81918620d934e219eee013 Mon Sep 17 00:00:00 2001
From: Nikhil Rao <nikhil@reflex.dev>
Date: Thu, 16 Nov 2023 13:20:48 -0800
Subject: [PATCH] Fix missing lazy imports (#2187)

---
 reflex/__init__.py    | 6 ++++++
 reflex/utils/build.py | 3 +++
 2 files changed, 9 insertions(+)

diff --git a/reflex/__init__.py b/reflex/__init__.py
index 2fd6cc766..03b13c527 100644
--- a/reflex/__init__.py
+++ b/reflex/__init__.py
@@ -62,6 +62,9 @@ _ALL_COMPONENTS = [
     "Container",
     "DataTable",
     "DataEditor",
+    "DataEditorTheme",
+    "DatePicker",
+    "DateTimePicker",
     "DebounceInput",
     "Divider",
     "Drawer",
@@ -227,6 +230,7 @@ _ALL_COMPONENTS = [
 _ALL_COMPONENTS += [to_snake_case(component) for component in _ALL_COMPONENTS]
 _ALL_COMPONENTS += [
     "components",
+    "color_mode_cond",
     "desktop_only",
     "mobile_only",
     "tablet_only",
@@ -234,7 +238,9 @@ _ALL_COMPONENTS += [
     "tablet_and_desktop",
     "selected_files",
     "clear_selected_files",
+    "EditorButtonList",
     "EditorOptions",
+    "NoSSRComponent",
 ]
 
 _MAPPING = {
diff --git a/reflex/utils/build.py b/reflex/utils/build.py
index ad3c3ca2c..d9cb7215c 100644
--- a/reflex/utils/build.py
+++ b/reflex/utils/build.py
@@ -214,6 +214,9 @@ def setup_frontend(
         root: The root path of the project.
         disable_telemetry: Whether to disable the Next telemetry.
     """
+    # Create the assets dir if it doesn't exist.
+    path_ops.mkdir(constants.Dirs.APP_ASSETS)
+
     # Copy asset files to public folder.
     path_ops.cp(
         src=str(root / constants.Dirs.APP_ASSETS),