From 8ae8688996338fb88c39255e336326c95f0f6961 Mon Sep 17 00:00:00 2001
From: Arik Chakma <arikchangma@gmail.com>
Date: Wed, 30 Apr 2025 00:07:08 +0600
Subject: [PATCH] wip

---
 .../GenerateCourse/GenerateAICourse.tsx       | 16 +++++-
 src/components/TopicDetail/TopicDetail.tsx    |  2 +-
 src/components/TopicDetail/TopicDetailAI.tsx  | 52 ++++++++++++++++---
 src/helper/generate-ai-course.ts              |  3 ++
 4 files changed, 62 insertions(+), 11 deletions(-)

diff --git a/src/components/GenerateCourse/GenerateAICourse.tsx b/src/components/GenerateCourse/GenerateAICourse.tsx
index 092cc0964..f7e57f8b2 100644
--- a/src/components/GenerateCourse/GenerateAICourse.tsx
+++ b/src/components/GenerateCourse/GenerateAICourse.tsx
@@ -54,6 +54,7 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
     const params = getUrlParams();
     const paramsTerm = params?.term;
     const paramsDifficulty = params?.difficulty;
+    const paramsSrc = params?.src || 'search';
     if (!paramsTerm || !paramsDifficulty) {
       return;
     }
@@ -87,6 +88,7 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
       instructions: paramsCustomInstructions,
       goal: paramsGoal,
       about: paramsAbout,
+      src: paramsSrc,
     });
   }, [term, difficulty]);
 
@@ -98,9 +100,18 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
     about?: string;
     isForce?: boolean;
     prompt?: string;
+    src?: string;
   }) => {
-    const { term, difficulty, isForce, prompt, instructions, goal, about } =
-      options;
+    const {
+      term,
+      difficulty,
+      isForce,
+      prompt,
+      instructions,
+      goal,
+      about,
+      src,
+    } = options;
 
     if (!isLoggedIn()) {
       window.location.href = '/ai';
@@ -121,6 +132,7 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
       about,
       isForce,
       prompt,
+      src,
     });
   };
 
diff --git a/src/components/TopicDetail/TopicDetail.tsx b/src/components/TopicDetail/TopicDetail.tsx
index d87f15d62..00d71807a 100644
--- a/src/components/TopicDetail/TopicDetail.tsx
+++ b/src/components/TopicDetail/TopicDetail.tsx
@@ -70,7 +70,7 @@ type PaidResourceType = {
 
 const paidResourcesCache: Record<string, PaidResourceType[]> = {};
 
-const defaultChatHistory: AIChatHistoryType[] = [
+export const defaultChatHistory: AIChatHistoryType[] = [
   {
     role: 'assistant',
     content: 'Hey, I am your AI instructor. How can I help you today? 🤖',
diff --git a/src/components/TopicDetail/TopicDetailAI.tsx b/src/components/TopicDetail/TopicDetailAI.tsx
index 18817ee76..36b488eaa 100644
--- a/src/components/TopicDetail/TopicDetailAI.tsx
+++ b/src/components/TopicDetail/TopicDetailAI.tsx
@@ -12,7 +12,13 @@ import { billingDetailsOptions } from '../../queries/billing';
 import { getAiCourseLimitOptions } from '../../queries/ai-course';
 import { queryClient } from '../../stores/query-client';
 import { isLoggedIn, removeAuthToken } from '../../lib/jwt';
-import { BotIcon, Loader2Icon, LockIcon, SendIcon } from 'lucide-react';
+import {
+  BotIcon,
+  Loader2Icon,
+  LockIcon,
+  RotateCcwIcon,
+  SendIcon,
+} from 'lucide-react';
 import { showLoginPopup } from '../../lib/popup';
 import { cn } from '../../lib/classname';
 import TextareaAutosize from 'react-textarea-autosize';
@@ -27,6 +33,7 @@ import { markdownToHtmlWithHighlighting } from '../../lib/markdown';
 import type { ResourceType } from '../../lib/resource-progress';
 import { getPercentage } from '../../lib/number';
 import { roadmapTreeMappingOptions } from '../../queries/roadmap-tree';
+import { defaultChatHistory } from './TopicDetail';
 
 type TopicDetailAIProps = {
   resourceId: string;
@@ -217,9 +224,16 @@ export function TopicDetailAI(props: TopicDetailAIProps) {
   );
   const hasSubjects =
     roadmapTreeMapping?.subjects && roadmapTreeMapping?.subjects?.length > 0;
+  const hasChatHistory = aiChatHistory.length > 1;
 
   return (
-    <div className="mt-4 flex grow flex-col overflow-hidden rounded-lg border border-gray-200">
+    <div className="relative mt-4 flex grow flex-col overflow-hidden rounded-lg border border-gray-200">
+      {isDataLoading && (
+        <div className="absolute inset-0 z-20 flex items-center justify-center gap-2 bg-white text-black">
+          <Loader2Icon className="size-8 animate-spin stroke-3 text-gray-500" />
+        </div>
+      )}
+
       {hasSubjects && (
         <div className="border-b border-gray-200 px-4 py-2">
           <h4 className="flex items-center gap-2 text-base">
@@ -232,7 +246,7 @@ export function TopicDetailAI(props: TopicDetailAIProps) {
                 <a
                   key={subject}
                   target="_blank"
-                  href={`/ai/search?term=${subject}&difficulty=beginner`}
+                  href={`/ai/search?term=${subject}&difficulty=beginner&src=topic`}
                   className="rounded-md border px-1.5"
                 >
                   {subject}
@@ -265,10 +279,33 @@ export function TopicDetailAI(props: TopicDetailAIProps) {
           </h4>
         )}
 
-        {!isDataLoading && !isPaidUser && (
-          <p className="text-sm text-gray-500">
-            <span className="font-medium">{usagePercentage}%</span> used
-          </p>
+        {!isDataLoading && (
+          <div className="flex items-center gap-2.5">
+            {hasChatHistory && (
+              <button
+                className="rounded-md bg-white p-1 text-xs font-medium text-black hover:bg-gray-200"
+                onClick={() => {
+                  setAiChatHistory(defaultChatHistory);
+                }}
+              >
+                <RotateCcwIcon className="size-3.5" />
+              </button>
+            )}
+
+            {!isPaidUser && (
+              <>
+                <button
+                  className="underline underline-offset-2 hover:no-underline"
+                  onClick={onUpgrade}
+                >
+                  Upgrade
+                </button>
+                <p className="text-sm text-gray-500">
+                  <span className="font-medium">{usagePercentage}%</span> used
+                </p>
+              </>
+            )}
+          </div>
         )}
       </div>
 
@@ -349,7 +386,6 @@ export function TopicDetailAI(props: TopicDetailAIProps) {
         <TextareaAutosize
           className={cn(
             'h-full min-h-[41px] grow resize-none bg-transparent px-4 py-2 focus:outline-hidden',
-            // isDisabled ? 'cursor-not-allowed opacity-50' : 'cursor-auto',
           )}
           placeholder="Ask AI anything about the lesson..."
           value={message}
diff --git a/src/helper/generate-ai-course.ts b/src/helper/generate-ai-course.ts
index 713e77e9a..fcd19150b 100644
--- a/src/helper/generate-ai-course.ts
+++ b/src/helper/generate-ai-course.ts
@@ -20,6 +20,7 @@ type GenerateCourseOptions = {
   onCourseChange?: (course: AiCourse, rawData: string) => void;
   onLoadingChange?: (isLoading: boolean) => void;
   onError?: (error: string) => void;
+  src?: string;
 };
 
 export async function generateCourse(options: GenerateCourseOptions) {
@@ -37,6 +38,7 @@ export async function generateCourse(options: GenerateCourseOptions) {
     instructions,
     goal,
     about,
+    src = 'search',
   } = options;
 
   onLoadingChange?.(true);
@@ -85,6 +87,7 @@ export async function generateCourse(options: GenerateCourseOptions) {
             instructions,
             goal,
             about,
+            src,
           }),
           credentials: 'include',
         },