first commit

Author: sjosh88

.claude/settings.local.json

@@ -0,0 +1,16 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(pnpm dev)",
+      "WebSearch",
+      "WebFetch(domain:docs.x.ai)",
+      "Bash(npx tsc:*)",
+      "Bash(grep:*)",
+      "Bash(tree:*)",
+      "Bash(pnpm build:*)",
+      "Bash(psql:*)"
+    ],
+    "deny": [],
+    "ask": []
+  }
+}

.env.example

@@ -1,23 +1,30 @@
-# Generate a random secret: https://generate-secret.vercel.app/32 or `openssl rand -base64 32`
-AUTH_SECRET=****
+# ===========================================
+# REQUIRED VARIABLES
+# ===========================================
 
-# The following keys below are automatically created and
-# added to your environment when you deploy on Vercel
+# Generate a random secret: https://generate-secret.vercel.app/32
+# Or run: openssl rand -base64 32
+AUTH_SECRET=
 
-# Instructions to create an AI Gateway API key here: https://vercel.com/ai-gateway 
-# API key required for non-Vercel deployments
-# For Vercel deployments, OIDC tokens are used automatically
-# https://vercel.com/ai-gateway
-AI_GATEWAY_API_KEY=****
+# xAI API key (for direct Grok access)
+# Get it from: https://console.x.ai
+XAI_API_KEY=
 
+# PostgreSQL Database URL
+# For Docker setup (recommended): postgresql://chatbot:chatbot@localhost:5432/chatbot
+# For Vercel/Neon: Get from https://vercel.com/docs/postgres
+POSTGRES_URL=
 
-# Instructions to create a Vercel Blob Store here: https://vercel.com/docs/vercel-blob
-BLOB_READ_WRITE_TOKEN=****
+# ===========================================
+# OPTIONAL VARIABLES (not needed for dev)
+# ===========================================
 
-# Instructions to create a PostgreSQL database here: https://vercel.com/docs/postgres
-POSTGRES_URL=****
+# Vercel AI Gateway API key (only if you prefer routing through the gateway)
+# Get it from: https://vercel.com/ai-gateway
+AI_GATEWAY_API_KEY=
 
+# Vercel Blob storage (for file uploads)
+# BLOB_READ_WRITE_TOKEN=
 
-# Instructions to create a Redis store here:
-# https://vercel.com/docs/redis
-REDIS_URL=****
+# Redis (for resumable streams)
+# REDIS_URL=

README.md

@@ -36,15 +36,13 @@
 
 ## Model Providers
 
-This template uses the [Vercel AI Gateway](https://vercel.com/docs/ai-gateway) to access multiple AI models through a unified interface. The default configuration includes [xAI](https://x.ai) models (`grok-2-vision-1212`, `grok-3-mini`) routed through the gateway.
+The chatbot uses [xAI](https://x.ai) Grok models directly through the AI SDK (`grok-2-vision-1212`, `grok-3-mini`). Set `XAI_API_KEY` in `.env.local` to authenticate requests.
 
-### AI Gateway Authentication
+### Optional: Vercel AI Gateway
 
-**For Vercel deployments**: Authentication is handled automatically via OIDC tokens.
+If you prefer routing traffic through the [Vercel AI Gateway](https://vercel.com/docs/ai-gateway) for logging or usage limits, provide `AI_GATEWAY_API_KEY`. Vercel deployments automatically handle authentication via OIDC tokens; self-hosted deployments must set the key manually.
 
-**For non-Vercel deployments**: You need to provide an AI Gateway API key by setting the `AI_GATEWAY_API_KEY` environment variable in your `.env.local` file.
-
-With the [AI SDK](https://ai-sdk.dev/docs/introduction), you can also switch to direct LLM providers like [OpenAI](https://openai.com), [Anthropic](https://anthropic.com), [Cohere](https://cohere.com/), and [many more](https://ai-sdk.dev/providers/ai-sdk-providers) with just a few lines of code.
+With the [AI SDK](https://ai-sdk.dev/docs/introduction), you can also switch to other providers like [OpenAI](https://openai.com), [Anthropic](https://anthropic.com), [Cohere](https://cohere.com/), and [many more](https://ai-sdk.dev/providers/ai-sdk-providers) with just a few lines of code.
 
 ## Deploy Your Own
 

SETUP.md

@@ -0,0 +1,88 @@
+# Local Development Setup
+
+Quick setup guide for Windows and Mac.
+
+## Prerequisites
+
+- [Docker Desktop](https://www.docker.com/products/docker-desktop/) installed and running
+- [Node.js 18+](https://nodejs.org/)
+- [pnpm](https://pnpm.io/) (`npm install -g pnpm`)
+
+## Quick Start
+
+### 1. Start the Database
+
+```bash
+# Start PostgreSQL in Docker
+docker-compose up -d
+
+# Verify it's running
+docker-compose ps
+```
+
+### 2. Configure Environment
+
+```bash
+# Copy example env file
+cp .env.example .env.local
+
+# Edit .env.local and fill in:
+# - AUTH_SECRET (generate at https://generate-secret.vercel.app/32)
+# - XAI_API_KEY (from https://console.x.ai)
+# - POSTGRES_URL=postgresql://chatbot:chatbot@localhost:5432/chatbot
+# (Optional) AI_GATEWAY_API_KEY if you want to proxy through Vercel AI Gateway
+```
+
+### 3. Install Dependencies & Run Migrations
+
+```bash
+# Install packages
+pnpm install
+
+# Run database migrations
+pnpm db:migrate
+```
+
+### 4. Start Development Server
+
+```bash
+pnpm dev
+```
+
+Open [http://localhost:3000](http://localhost:3000)
+
+## Common Commands
+
+```bash
+# Stop database
+docker-compose down
+
+# Stop and remove all data
+docker-compose down -v
+
+# View database logs
+docker-compose logs -f postgres
+
+# Open Drizzle Studio (database UI)
+pnpm db:studio
+```
+
+## Troubleshooting
+
+### Port 5432 already in use
+Stop any existing PostgreSQL or change the port in docker-compose.yml:
+```yaml
+ports:
+  - "5433:5432"  # Use 5433 instead
+```
+Then update POSTGRES_URL to use port 5433.
+
+### Database connection refused
+Make sure Docker Desktop is running and the container is up:
+```bash
+docker-compose ps
+docker-compose up -d
+```
+
+### Migration fails
+Check your POSTGRES_URL is correct and the database is running.

app/(chat)/api/chat/route.ts

@@ -24,6 +24,7 @@ import { type RequestHints, systemPrompt } from "@/lib/ai/prompts";
 import { myProvider } from "@/lib/ai/providers";
 import { createDocument } from "@/lib/ai/tools/create-document";
 import { getWeather } from "@/lib/ai/tools/get-weather";
+import { analyzeMarketingMoment } from "@/lib/ai/tools/marketing-moment-analyzer";
 import { requestSuggestions } from "@/lib/ai/tools/request-suggestions";
 import { updateDocument } from "@/lib/ai/tools/update-document";
 import { isProductionEnvironment } from "@/lib/constants";
@@ -49,22 +50,6 @@ export const maxDuration = 60;
 
 let globalStreamContext: ResumableStreamContext | null = null;
 
-const getTokenlensCatalog = cache(
-  async (): Promise<ModelCatalog | undefined> => {
-    try {
-      return await fetchModels();
-    } catch (err) {
-      console.warn(
-        "TokenLens: catalog fetch failed, using default catalog",
-        err
-      );
-      return; // tokenlens helpers will fall back to defaultCatalog
-    }
-  },
-  ["tokenlens-catalog"],
-  { revalidate: 24 * 60 * 60 } // 24 hours
-);
-
 export function getStreamContext() {
   if (!globalStreamContext) {
     try {
@@ -73,18 +58,31 @@ export function getStreamContext() {
       });
     } catch (error: any) {
       if (error.message.includes("REDIS_URL")) {
-        console.log(
-          " > Resumable streams are disabled due to missing REDIS_URL"
-        );
+        console.log(" > Resumable streams disabled (missing REDIS_URL)");
       } else {
         console.error(error);
       }
     }
   }
-
   return globalStreamContext;
 }
 
+const getTokenlensCatalog = cache(
+  async (): Promise<ModelCatalog | undefined> => {
+    try {
+      return await fetchModels();
+    } catch (err) {
+      console.warn(
+        "TokenLens: catalog fetch failed, using default catalog",
+        err
+      );
+      return; // tokenlens helpers will fall back to defaultCatalog
+    }
+  },
+  ["tokenlens-catalog"],
+  { revalidate: 24 * 60 * 60 } // 24 hours
+);
+
 export async function POST(request: Request) {
   let requestBody: PostRequestBody;
 
@@ -179,20 +177,27 @@ export async function POST(request: Request) {
 
     const stream = createUIMessageStream({
       execute: ({ writer: dataStream }) => {
+        const activeTools:
+          | ("getWeather" | "createDocument" | "updateDocument" | "requestSuggestions" | "analyzeMarketingMoment")[]
+          | undefined =
+          selectedChatModel === "chat-model-reasoning"
+            ? undefined
+            : [
+                "getWeather",
+                "createDocument",
+                "updateDocument",
+                "requestSuggestions",
+                "analyzeMarketingMoment",
+              ];
+
+        const systemPromptContent = systemPrompt({ selectedChatModel, requestHints });
+
         const result = streamText({
           model: myProvider.languageModel(selectedChatModel),
-          system: systemPrompt({ selectedChatModel, requestHints }),
+          system: systemPromptContent,
           messages: convertToModelMessages(uiMessages),
           stopWhen: stepCountIs(5),
-          experimental_activeTools:
-            selectedChatModel === "chat-model-reasoning"
-              ? []
-              : [
-                  "getWeather",
-                  "createDocument",
-                  "updateDocument",
-                  "requestSuggestions",
-                ],
+          experimental_activeTools: activeTools,
           experimental_transform: smoothStream({ chunking: "word" }),
           tools: {
             getWeather,
@@ -202,6 +207,7 @@ export async function POST(request: Request) {
               session,
               dataStream,
             }),
+            analyzeMarketingMoment: analyzeMarketingMoment({ dataStream }),
           },
           experimental_telemetry: {
             isEnabled: isProductionEnvironment,
@@ -278,16 +284,6 @@ export async function POST(request: Request) {
       },
     });
 
-    // const streamContext = getStreamContext();
-
-    // if (streamContext) {
-    //   return new Response(
-    //     await streamContext.resumableStream(streamId, () =>
-    //       stream.pipeThrough(new JsonToSseTransformStream())
-    //     )
-    //   );
-    // }
-
     return new Response(stream.pipeThrough(new JsonToSseTransformStream()));
   } catch (error) {
     const vercelId = request.headers.get("x-vercel-id");

app/(chat)/layout.tsx

@@ -3,6 +3,7 @@ import Script from "next/script";
 import { AppSidebar } from "@/components/app-sidebar";
 import { DataStreamProvider } from "@/components/data-stream-provider";
 import { SidebarInset, SidebarProvider } from "@/components/ui/sidebar";
+import { MarketingProgressProvider } from "@/hooks/use-marketing-progress";
 import { auth } from "../(auth)/auth";
 
 export const experimental_ppr = true;
@@ -22,10 +23,12 @@ export default async function Layout({
         strategy="beforeInteractive"
       />
       <DataStreamProvider>
-        <SidebarProvider defaultOpen={!isCollapsed}>
-          <AppSidebar user={session?.user} />
-          <SidebarInset>{children}</SidebarInset>
-        </SidebarProvider>
+        <MarketingProgressProvider>
+          <SidebarProvider defaultOpen={!isCollapsed}>
+            <AppSidebar user={session?.user} />
+            <SidebarInset>{children}</SidebarInset>
+          </SidebarProvider>
+        </MarketingProgressProvider>
       </DataStreamProvider>
     </>
   );

app/test-marketing/page.tsx

@@ -0,0 +1,70 @@
+"use client";
+
+import { useState } from "react";
+
+export default function TestMarketingPage() {
+  const [status, setStatus] = useState("Ready to test");
+  const [response, setResponse] = useState<any>(null);
+
+  const testChat = async () => {
+    try {
+      setStatus("Testing...");
+      const res = await fetch("/api/chat", {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify({
+          id: "test-" + Date.now(),
+          message: {
+            id: "msg-" + Date.now(),
+            role: "user",
+            parts: [
+              {
+                type: "text",
+                text: "Is it a good time to launch a coffee shop in Seattle?",
+              },
+            ],
+          },
+          selectedChatModel: "chat-model-default",
+          selectedVisibilityType: "private",
+        }),
+      });
+
+      if (!res.ok) {
+        const errorText = await res.text();
+        setStatus(`Error: ${res.status} - ${errorText}`);
+        setResponse({ error: errorText });
+        return;
+      }
+
+      setStatus("Success!");
+      setResponse({ ok: true });
+    } catch (error: any) {
+      setStatus(`Exception: ${error.message}`);
+      setResponse({ error: error.message });
+    }
+  };
+
+  return (
+    <div className="p-8">
+      <h1 className="text-2xl font-bold mb-4">Marketing Tool Test</h1>
+      <div className="space-y-4">
+        <button
+          onClick={testChat}
+          className="px-4 py-2 bg-blue-500 text-white rounded hover:bg-blue-600"
+        >
+          Test Marketing Query
+        </button>
+        <div>
+          <strong>Status:</strong> {status}
+        </div>
+        {response && (
+          <pre className="bg-gray-100 p-4 rounded overflow-auto">
+            {JSON.stringify(response, null, 2)}
+          </pre>
+        )}
+      </div>
+    </div>
+  );
+}