From e973c1edbf42260f9643764e16fac864c70efbc4 Mon Sep 17 00:00:00 2001
From: Timothy Carambat <rambat1010@gmail.com>
Date: Tue, 16 Jan 2024 14:15:46 -0800
Subject: [PATCH] Preload onboarding (#606)

* no lazy onboarding

* no lazy onboarding steps

* Do not lazy load onboarding to prevent lazy-load white flash
---
 frontend/src/App.jsx                          |  3 +-
 .../src/pages/OnboardingFlow/Steps/index.jsx  | 30 ++++++++++++-------
 2 files changed, 22 insertions(+), 11 deletions(-)

diff --git a/frontend/src/App.jsx b/frontend/src/App.jsx
index fa74d434..7c14be4d 100644
--- a/frontend/src/App.jsx
+++ b/frontend/src/App.jsx
@@ -8,6 +8,8 @@ import PrivateRoute, {
 import { ToastContainer } from "react-toastify";
 import "react-toastify/dist/ReactToastify.css";
 import Login from "@/pages/Login";
+import OnboardingFlow from "@/pages/OnboardingFlow";
+
 import { PfpProvider } from "./PfpContext";
 import { LogoProvider } from "./LogoContext";
 
@@ -42,7 +44,6 @@ const DataConnectors = lazy(
 const DataConnectorSetup = lazy(
   () => import("@/pages/GeneralSettings/DataConnectors/Connectors")
 );
-const OnboardingFlow = lazy(() => import("@/pages/OnboardingFlow"));
 
 export default function App() {
   return (
diff --git a/frontend/src/pages/OnboardingFlow/Steps/index.jsx b/frontend/src/pages/OnboardingFlow/Steps/index.jsx
index 3f218d53..957d94a4 100644
--- a/frontend/src/pages/OnboardingFlow/Steps/index.jsx
+++ b/frontend/src/pages/OnboardingFlow/Steps/index.jsx
@@ -1,16 +1,26 @@
 import { ArrowLeft, ArrowRight } from "@phosphor-icons/react";
-import { lazy, useState } from "react";
+import { useState } from "react";
 import { isMobile } from "react-device-detect";
+import Home from "./Home";
+import LLMPreference from "./LLMPreference";
+import EmbeddingPreference from "./EmbeddingPreference";
+import VectorDatabaseConnection from "./VectorDatabaseConnection";
+import CustomLogo from "./CustomLogo";
+import UserSetup from "./UserSetup";
+import DataHandling from "./DataHandling";
+import Survey from "./Survey";
+import CreateWorkspace from "./CreateWorkspace";
+
 const OnboardingSteps = {
-  home: lazy(() => import("./Home")),
-  "llm-preference": lazy(() => import("./LLMPreference")),
-  "embedding-preference": lazy(() => import("./EmbeddingPreference")),
-  "vector-database": lazy(() => import("./VectorDatabaseConnection")),
-  "custom-logo": lazy(() => import("./CustomLogo")),
-  "user-setup": lazy(() => import("./UserSetup")),
-  "data-handling": lazy(() => import("./DataHandling")),
-  survey: lazy(() => import("./Survey")),
-  "create-workspace": lazy(() => import("./CreateWorkspace")),
+  home: Home,
+  "llm-preference": LLMPreference,
+  "embedding-preference": EmbeddingPreference,
+  "vector-database": VectorDatabaseConnection,
+  "custom-logo": CustomLogo,
+  "user-setup": UserSetup,
+  "data-handling": DataHandling,
+  survey: Survey,
+  "create-workspace": CreateWorkspace,
 };
 
 export default OnboardingSteps;