Implement Domain-Based System Prompt Generation

.env.local.example

@@ -24,6 +24,9 @@ GEMINI_3_PRO_API_KEY=your_gemini_3_pro_api_key_here
 # If not set, the `search` tool is skipped and the LLM answers from its own knowledge.
 TAVILY_API_KEY=your_tavily_api_key
 
+# Firecrawl API (https://firecrawl.dev)
+FIRECRAWL_API_KEY=your_firecrawl_api_key
+
 # Supabase Credentials
 NEXT_PUBLIC_SUPABASE_URL=YOUR_SUPABASE_URL_HERE
 NEXT_PUBLIC_SUPABASE_ANON_KEY=YOUR_SUPABASE_ANON_KEY_HERE

components/settings/components/settings.tsx

@@ -46,6 +46,7 @@ const settingsFormSchema = z.object({
   ),
   newUserEmail: z.string().email().optional().or(z.literal('')),
   newUserRole: z.enum(["admin", "editor", "viewer"]).optional(),
+  domain: z.string().url().optional().or(z.literal('')),
 })
 
 export type SettingsFormValues = z.infer<typeof settingsFormSchema>
@@ -56,6 +57,7 @@ const defaultValues: Partial<SettingsFormValues> = {
     "You are a planetary copilot, an AI assistant designed to help users with information about planets, space exploration, and astronomy. Provide accurate, educational, and engaging responses about our solar system and beyond.",
   selectedModel: "Gemini 3.1 Pro",
   users: [],
+  domain: "",
 }
 
 interface SettingsProps {

components/settings/components/system-prompt-form.tsx

@@ -1,36 +1,166 @@
+import React, { useState, useEffect } from 'react'
 import type { UseFormReturn } from "react-hook-form"
 import { FormField, FormItem, FormLabel, FormControl, FormDescription, FormMessage } from "@/components/ui/form"
 import { Textarea } from "@/components/ui/textarea"
+import { Input } from "@/components/ui/input"
+import { Button } from "@/components/ui/button"
+import { Loader2, Sparkles } from "lucide-react"
+import { useToast } from "@/components/ui/hooks/use-toast"
+import { startSystemPromptGeneration, getSystemPromptGenerationJob } from "@/lib/actions/system-prompt"
 
 interface SystemPromptFormProps {
   form: UseFormReturn<any>
 }
 
 export function SystemPromptForm({ form }: SystemPromptFormProps) {
+  const { toast } = useToast()
   const systemPrompt = form.watch("systemPrompt")
+  const domain = form.watch("domain")
   const characterCount = systemPrompt?.length || 0
+  const [jobId, setJobId] = useState<string | null>(null)
+  const [isGenerating, setIsGenerating] = useState(false)
+
+  const handleGenerate = async () => {
+    if (!domain) {
+      toast({
+        title: "Domain required",
+        description: "Please enter a domain to generate a system prompt.",
+        variant: "destructive",
+      })
+      return
+    }
+
+    setIsGenerating(true)
+    const result = await startSystemPromptGeneration(domain)
+
+    if (result.error) {
+      toast({
+        title: "Generation failed",
+        description: result.error,
+        variant: "destructive",
+      })
+      setIsGenerating(false)
+    } else if (result.jobId) {
+      setJobId(result.jobId)
+    }
+  }
+
+  useEffect(() => {
+    let interval: NodeJS.Timeout | null = null
+
+    if (jobId) {
+      const pollStartTime = Date.now()
+      const MAX_POLL_DURATION = 120000 // 2 minutes
+
+      interval = setInterval(async () => {
+        const elapsed = Date.now() - pollStartTime
+        if (elapsed > MAX_POLL_DURATION) {
+          if (interval) clearInterval(interval)
+          setJobId(null)
+          setIsGenerating(false)
+          toast({
+            title: "Generation timeout",
+            description: "The prompt generation took too long. Please try again.",
+            variant: "destructive",
+          })
+          return
+        }
+
+        const job = await getSystemPromptGenerationJob(jobId)
+
+        if (job.error || job.status === 'error') {
+          if (interval) clearInterval(interval)
+          setJobId(null)
+          setIsGenerating(false)
+          toast({
+            title: "Generation error",
+            description: job.errorMessage || job.error || "An error occurred during generation.",
+            variant: "destructive",
+          })
+        } else if (job.status === 'complete') {
+          if (interval) clearInterval(interval)
+          setJobId(null)
+          setIsGenerating(false)
+          if (job.resultPrompt) {
+            form.setValue("systemPrompt", job.resultPrompt, { shouldValidate: true, shouldDirty: true })
+            toast({
+              title: "Prompt generated",
+              description: "Your system prompt has been generated based on the domain content.",
+            })
+          }
+        }
+      }, 3000)
+    }
+
+    return () => {
+      if (interval) clearInterval(interval)
+    }
+  }, [jobId, form, toast])
 
   return (
-    <FormField
-      control={form.control}
-      name="systemPrompt"
-      render={({ field, fieldState, formState }: { field: import("react-hook-form").ControllerRenderProps<any, "systemPrompt">; fieldState: import("react-hook-form").ControllerFieldState; formState: import("react-hook-form").UseFormStateReturn<any>; }) => (
-        <FormItem>
-          <FormLabel>System Prompt</FormLabel>
-          <FormControl>
-            <Textarea
-              placeholder="Enter the system prompt for your planetary copilot..."
-              className="min-h-[200px] resize-y"
-              {...field}
-            />
-          </FormControl>
-          <FormDescription className="flex justify-between">
-            <span>Define how your copilot should behave and respond to user queries.</span>
-            <span className={characterCount > 1800 ? "text-amber-500" : ""}>{characterCount}/2000</span>
-          </FormDescription>
-          <FormMessage />
-        </FormItem>
-      )}
-    />
+    <div className="space-y-6">
+      <FormField
+        control={form.control}
+        name="domain"
+        render={({ field }) => (
+          <FormItem>
+            <FormLabel>Business Domain</FormLabel>
+            <div className="flex gap-2">
+              <FormControl>
+                <Input
+                  placeholder="example.com"
+                  {...field}
+                  disabled={isGenerating}
+                />
+              </FormControl>
+              <Button
+                type="button"
+                onClick={handleGenerate}
+                disabled={isGenerating || !domain}
+                className="shrink-0"
+              >
+                {isGenerating ? (
+                  <>
+                    <Loader2 className="mr-2 h-4 w-4 animate-spin" />
+                    Generating...
+                  </>
+                ) : (
+                  <>
+                    <Sparkles className="mr-2 h-4 w-4" />
+                    Generate
+                  </>
+                )}
+              </Button>
+            </div>
+            <FormDescription>
+              Enter your business website to automatically generate a tailored system prompt.
+            </FormDescription>
+            <FormMessage />
+          </FormItem>
+        )}
+      />
+
+      <FormField
+        control={form.control}
+        name="systemPrompt"
+        render={({ field }) => (
+          <FormItem>
+            <FormLabel>System Prompt</FormLabel>
+            <FormControl>
+              <Textarea
+                placeholder="Enter the system prompt for your planetary copilot..."
+                className="min-h-[200px] resize-y"
+                {...field}
+              />
+            </FormControl>
+            <FormDescription className="flex justify-between">
+              <span>Define how your copilot should behave and respond to user queries.</span>
+              <span className={characterCount > 1800 ? "text-amber-500" : ""}>{characterCount}/2000</span>
+            </FormDescription>
+            <FormMessage />
+          </FormItem>
+        )}
+      />
+    </div>
   )
 }

drizzle/migrations/0002_lively_black_widow.sql

@@ -0,0 +1,12 @@
+CREATE TABLE "prompt_generation_jobs" (
+	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
+	"user_id" uuid NOT NULL,
+	"domain" text NOT NULL,
+	"status" text DEFAULT 'pending' NOT NULL,
+	"result_prompt" text,
+	"error_message" text,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+ALTER TABLE "prompt_generation_jobs" ADD CONSTRAINT "prompt_generation_jobs_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE cascade ON UPDATE no action;