From 9356369fdafeb145323b209cf7e135e51765a836 Mon Sep 17 00:00:00 2001
From: Kamran Ahmed <kamranahmed.se@gmail.com>
Date: Tue, 25 Mar 2025 22:10:43 +0000
Subject: [PATCH] Update header for course

---
 .../GenerateCourse/AICourseContent.tsx         | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/components/GenerateCourse/AICourseContent.tsx b/src/components/GenerateCourse/AICourseContent.tsx
index 4bf283436..fddf994d0 100644
--- a/src/components/GenerateCourse/AICourseContent.tsx
+++ b/src/components/GenerateCourse/AICourseContent.tsx
@@ -292,12 +292,18 @@ export function AICourseContent(props: AICourseContentProps) {
               onShowLimits={() => setShowAILimitsPopup(true)}
             />
 
-            <AIRoadmapViewSwitch
-              viewMode={viewMode}
-              setViewMode={setViewMode}
-              isLoading={isLoading}
-              variant="text"
-            />
+            {viewMode === 'module' && (
+              <button
+                onClick={() => {
+                  setExpandedModules({});
+                  setViewMode('outline');
+                }}
+                className="flex flex-shrink-0 items-center rounded-md border border-gray-300 bg-white px-4 py-2 text-sm font-medium text-gray-700 shadow-sm transition-colors hover:bg-gray-50 hover:text-gray-900 max-lg:hidden"
+              >
+                <BookOpenCheck size={18} className="mr-2" />
+                View Course Outline
+              </button>
+            )}
           </div>
         </div>
       </header>