WIP

agenta-web/src/components/Layout/Layout.tsx

@@ -170,7 +170,7 @@ const App: React.FC<LayoutProps> = ({children}) => {
         <NoSSRWrapper>
             {typeof window === "undefined" ? null : (
                 <ThemeProvider theme={{...token, isDark: isDarkTheme}}>
-                    {isAuthRoute || !isProjectId ? (
+                    {isAuthRoute ? (
                         <Layout className={classes.layout}>
                             <ErrorBoundary FallbackComponent={ErrorFallback}>
                                 {children}

agenta-web/src/components/ProtectedRoute/ProtectedRoute.tsx

@@ -1,22 +1,24 @@
 import {useSession} from "@/hooks/useSession"
 import {useRouter} from "next/router"
 import React, {PropsWithChildren, useEffect, useState} from "react"
+import {useProjectData} from "@/contexts/project.context"
 
 const ProtectedRoute: React.FC<PropsWithChildren> = ({children}) => {
     const router = useRouter()
     const {loading, doesSessionExist: isSignedIn} = useSession()
     const {pathname} = router
     const [shouldRender, setShouldRender] = useState(false)
+    const {isLoading, isProjectId} = useProjectData()
 
     useEffect(() => {
-        if (loading) {
+        if (loading || isLoading) {
             setShouldRender(false)
         } else {
             if (pathname.startsWith("/auth")) {
                 if (isSignedIn) {
                     router.push("/apps")
                 }
-                setShouldRender(!isSignedIn)
+                setShouldRender(true)
             } else {
                 if (!isSignedIn) {
                     router.push(
@@ -25,10 +27,10 @@ const ProtectedRoute: React.FC<PropsWithChildren> = ({children}) => {
                         )}`,
                     )
                 }
-                setShouldRender(isSignedIn)
+                setShouldRender(!!isProjectId)
             }
         }
-    }, [pathname, isSignedIn, loading])
+    }, [pathname, isSignedIn, loading, isProjectId, isLoading])
 
     return <>{shouldRender ? children : null}</>
 }

agenta-web/src/contexts/project.context.tsx

@@ -44,7 +44,7 @@ export const getCurrentProject = () => projectContextValues
 const ProjectContextProvider: React.FC<PropsWithChildren> = ({children}) => {
     const [project, setProject] = useStateCallback<Project | null>(null)
     const [useOrgData, setUseOrgData] = useState<Function>(() => () => "")
-    const [isLoading, setIsLoading] = useState(true)
+    const [isLoading, setIsLoading] = useState(false)
     const {doesSessionExist} = useSession()
 
     useEffect(() => {
@@ -103,7 +103,7 @@ const ProjectContextProvider: React.FC<PropsWithChildren> = ({children}) => {
                 refetch: fetcher,
             }}
         >
-            {isProjectId ? children : null}
+            {children}
         </ProjectContext.Provider>
     )
 }