From 9248766e4e51b5a07aa7bdb75b68197aeb0fe114 Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Tue, 3 Mar 2026 11:45:54 -0600
Subject: [PATCH] ui: keep first-run onboarding mounted during company creation

---
 ui/src/App.tsx | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/ui/src/App.tsx b/ui/src/App.tsx
index 5d7f2ceb..019ed3b5 100644
--- a/ui/src/App.tsx
+++ b/ui/src/App.tsx
@@ -134,11 +134,17 @@ function boardRoutes() {
 
 function CompanyRootRedirect() {
   const { companies, selectedCompany, loading } = useCompany();
+  const { onboardingOpen } = useDialog();
 
   if (loading) {
     return <div className="mx-auto max-w-xl py-10 text-sm text-muted-foreground">Loading...</div>;
   }
 
+  // Keep the first-run onboarding mounted until it completes.
+  if (onboardingOpen) {
+    return <NoCompaniesStartPage autoOpen={false} />;
+  }
+
   const targetCompany = selectedCompany ?? companies[0] ?? null;
   if (!targetCompany) {
     return <NoCompaniesStartPage />;
@@ -168,15 +174,16 @@ function UnprefixedBoardRedirect() {
   );
 }
 
-function NoCompaniesStartPage() {
+function NoCompaniesStartPage({ autoOpen = true }: { autoOpen?: boolean }) {
   const { openOnboarding } = useDialog();
   const opened = useRef(false);
 
   useEffect(() => {
+    if (!autoOpen) return;
     if (opened.current) return;
     opened.current = true;
     openOnboarding();
-  }, [openOnboarding]);
+  }, [autoOpen, openOnboarding]);
 
   return (
     <div className="mx-auto max-w-xl py-10">