From f72d4ddc40e122e76798973ad82fe594fc490ce5 Mon Sep 17 00:00:00 2001
From: Kamran Ahmed <kamranahmed.se@gmail.com>
Date: Fri, 14 Mar 2025 02:04:13 +0000
Subject: [PATCH] Add course regeneration

---
 .../GenerateCourse/AICourseContent.tsx        |  2 +-
 .../GenerateCourse/GenerateAICourse.tsx       | 13 +++-
 src/components/GenerateCourse/GetAICourse.tsx |  3 +-
 .../GenerateCourse/ModifyCoursePrompt.tsx     | 69 +++++++++++++++++++
 .../GenerateCourse/RegenerateOutline.tsx      | 27 +++++++-
 src/helper/generate-ai-course.ts              |  7 ++
 6 files changed, 114 insertions(+), 7 deletions(-)
 create mode 100644 src/components/GenerateCourse/ModifyCoursePrompt.tsx

diff --git a/src/components/GenerateCourse/AICourseContent.tsx b/src/components/GenerateCourse/AICourseContent.tsx
index 7e30708e7..90eb00f8b 100644
--- a/src/components/GenerateCourse/AICourseContent.tsx
+++ b/src/components/GenerateCourse/AICourseContent.tsx
@@ -27,7 +27,7 @@ type AICourseContentProps = {
   course: AiCourse;
   isLoading: boolean;
   error?: string;
-  onRegenerateOutline: () => void;
+  onRegenerateOutline: (prompt?: string) => void;
 };
 
 export function AICourseContent(props: AICourseContentProps) {
diff --git a/src/components/GenerateCourse/GenerateAICourse.tsx b/src/components/GenerateCourse/GenerateAICourse.tsx
index 7041bd0bf..7a054a3cb 100644
--- a/src/components/GenerateCourse/GenerateAICourse.tsx
+++ b/src/components/GenerateCourse/GenerateAICourse.tsx
@@ -43,8 +43,9 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
     term: string;
     difficulty: string;
     isForce?: boolean;
+    prompt?: string;
   }) => {
-    const { term, difficulty, isForce } = options;
+    const { term, difficulty, isForce, prompt } = options;
 
     if (!isLoggedIn()) {
       window.location.href = '/ai-tutor';
@@ -61,6 +62,7 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
       onLoadingChange: setIsLoading,
       onError: setError,
       isForce,
+      prompt,
     });
   };
 
@@ -95,8 +97,13 @@ export function GenerateAICourse(props: GenerateAICourseProps) {
       course={course}
       isLoading={isLoading}
       error={error}
-      onRegenerateOutline={() => {
-        handleGenerateCourse({ term, difficulty, isForce: true });
+      onRegenerateOutline={(prompt) => {
+        handleGenerateCourse({
+          term,
+          difficulty,
+          isForce: true,
+          prompt,
+        });
       }}
     />
   );
diff --git a/src/components/GenerateCourse/GetAICourse.tsx b/src/components/GenerateCourse/GetAICourse.tsx
index 3219b4eea..b3970d3fd 100644
--- a/src/components/GenerateCourse/GetAICourse.tsx
+++ b/src/components/GenerateCourse/GetAICourse.tsx
@@ -58,7 +58,7 @@ export function GetAICourse(props: GetAICourseProps) {
     setError(queryError.message);
   }, [queryError]);
 
-  const handleRegenerateCourse = async () => {
+  const handleRegenerateCourse = async (prompt?: string) => {
     if (!aiCourse) {
       return;
     }
@@ -67,6 +67,7 @@ export function GetAICourse(props: GetAICourseProps) {
       term: aiCourse.keyword,
       difficulty: aiCourse.difficulty,
       slug: courseSlug,
+      prompt,
       onCourseChange: (course, rawData) => {
         queryClient.setQueryData(
           getAiCourseOptions({ aiCourseSlug: courseSlug }).queryKey,
diff --git a/src/components/GenerateCourse/ModifyCoursePrompt.tsx b/src/components/GenerateCourse/ModifyCoursePrompt.tsx
new file mode 100644
index 000000000..35583635d
--- /dev/null
+++ b/src/components/GenerateCourse/ModifyCoursePrompt.tsx
@@ -0,0 +1,69 @@
+import { useState } from 'react';
+import { Modal } from '../Modal';
+
+export type ModifyCoursePromptProps = {
+  onClose: () => void;
+  onSubmit: (prompt: string) => void;
+};
+
+export function ModifyCoursePrompt(props: ModifyCoursePromptProps) {
+  const { onClose, onSubmit } = props;
+
+  const [prompt, setPrompt] = useState('');
+
+  const handleSubmit = (e: React.FormEvent<HTMLFormElement>) => {
+    e.preventDefault();
+    onSubmit(prompt);
+  };
+
+  return (
+    <Modal
+      onClose={onClose}
+      wrapperClassName="rounded-xl max-w-xl w-full h-auto"
+      bodyClassName="p-6"
+      overlayClassName="items-start md:items-center"
+    >
+      <div className="flex flex-col gap-4">
+        <div>
+          <h2 className="mb-2 text-left text-xl font-semibold">
+            Give AI more context
+          </h2>
+          <p className="text-sm text-gray-500">
+            Pass additional information to the AI to generate a course outline.
+          </p>
+        </div>
+        <form className="flex flex-col gap-2" onSubmit={handleSubmit}>
+          <textarea
+            id="prompt"
+            autoFocus
+            rows={3}
+            value={prompt}
+            onChange={(e) => setPrompt(e.target.value)}
+            className="w-full rounded-md border border-gray-200 p-2 placeholder:text-sm focus:outline-black"
+            placeholder="e.g. make sure to add a section on React hooks"
+          />
+
+          <p className="text-sm text-gray-500">
+            Complete the sentence: "I want AI to..."
+          </p>
+
+          <div className="flex justify-end gap-2">
+            <button
+              className="rounded-md bg-gray-200 px-4 py-2.5 text-sm text-black hover:opacity-80"
+              onClick={onClose}
+            >
+              Cancel
+            </button>
+            <button
+              type="submit"
+              disabled={!prompt.trim()}
+              className="rounded-md bg-black px-4 py-2.5 text-sm text-white hover:opacity-80 disabled:opacity-50"
+            >
+              Modify Prompt
+            </button>
+          </div>
+        </form>
+      </div>
+    </Modal>
+  );
+}
diff --git a/src/components/GenerateCourse/RegenerateOutline.tsx b/src/components/GenerateCourse/RegenerateOutline.tsx
index 43616ac9a..9aa4cb894 100644
--- a/src/components/GenerateCourse/RegenerateOutline.tsx
+++ b/src/components/GenerateCourse/RegenerateOutline.tsx
@@ -4,9 +4,10 @@ import { useOutsideClick } from '../../hooks/use-outside-click';
 import { cn } from '../../lib/classname';
 import { useIsPaidUser } from '../../queries/billing';
 import { UpgradeAccountModal } from '../Billing/UpgradeAccountModal';
+import { ModifyCoursePrompt } from './ModifyCoursePrompt';
 
 type RegenerateOutlineProps = {
-  onRegenerateOutline: () => void;
+  onRegenerateOutline: (prompt?: string) => void;
 };
 
 export function RegenerateOutline(props: RegenerateOutlineProps) {
@@ -14,6 +15,8 @@ export function RegenerateOutline(props: RegenerateOutlineProps) {
 
   const [isDropdownVisible, setIsDropdownVisible] = useState(false);
   const [showUpgradeModal, setShowUpgradeModal] = useState(false);
+  const [showPromptModal, setShowPromptModal] = useState(false);
+
   const ref = useRef<HTMLDivElement>(null);
 
   const { isPaidUser } = useIsPaidUser();
@@ -30,6 +33,16 @@ export function RegenerateOutline(props: RegenerateOutlineProps) {
         />
       )}
 
+      {showPromptModal && (
+        <ModifyCoursePrompt
+          onClose={() => setShowPromptModal(false)}
+          onSubmit={(prompt) => {
+            setShowPromptModal(false);
+            onRegenerateOutline(prompt);
+          }}
+        />
+      )}
+
       <div className="absolute right-3 top-3" ref={ref}>
         <button
           className={cn('text-gray-400 hover:text-black', {
@@ -59,7 +72,17 @@ export function RegenerateOutline(props: RegenerateOutlineProps) {
               />
               Regenerate
             </button>
-            <button className="flex w-full items-center gap-2.5 px-3 py-2 text-left text-sm text-gray-600 hover:bg-gray-100">
+            <button
+              onClick={() => {
+                setIsDropdownVisible(false);
+                if (!isPaidUser) {
+                  setShowUpgradeModal(true);
+                } else {
+                  setShowPromptModal(true);
+                }
+              }}
+              className="flex w-full items-center gap-2.5 px-3 py-2 text-left text-sm text-gray-600 hover:bg-gray-100"
+            >
               <PenSquare
                 size={16}
                 className="text-gray-400"
diff --git a/src/helper/generate-ai-course.ts b/src/helper/generate-ai-course.ts
index 9706b42e6..9aa03ad56 100644
--- a/src/helper/generate-ai-course.ts
+++ b/src/helper/generate-ai-course.ts
@@ -11,6 +11,7 @@ type GenerateCourseOptions = {
   difficulty: string;
   slug?: string;
   isForce?: boolean;
+  prompt?: string;
   onCourseIdChange?: (courseId: string) => void;
   onCourseSlugChange?: (courseSlug: string) => void;
   onCourseChange?: (course: AiCourse, rawData: string) => void;
@@ -29,6 +30,7 @@ export async function generateCourse(options: GenerateCourseOptions) {
     onLoadingChange,
     onError,
     isForce = false,
+    prompt,
   } = options;
 
   onLoadingChange?.(true);
@@ -54,6 +56,10 @@ export async function generateCourse(options: GenerateCourseOptions) {
             'Content-Type': 'application/json',
           },
           credentials: 'include',
+          body: JSON.stringify({
+            isForce,
+            customPrompt: prompt,
+          }),
         },
       );
     } else {
@@ -68,6 +74,7 @@ export async function generateCourse(options: GenerateCourseOptions) {
             keyword: term,
             difficulty,
             isForce,
+            customPrompt: prompt,
           }),
           credentials: 'include',
         },