├── .github
    └── ISSUE_TEMPLATE
    │   ├── config.yml
    │   └── issue_template.yml
├── demo.gif
├── app
    ├── favicon.ico
    ├── apple-icon.png
    ├── opengraph-image.png
    ├── report
    │   └── [id]
    │   │   ├── page.tsx
    │   │   └── report-content.tsx
    ├── layout.tsx
    ├── api
    │   ├── parse-document
    │   │   └── route.ts
    │   ├── fetch-content
    │   │   └── route.ts
    │   ├── download
    │   │   └── route.ts
    │   ├── generate-question
    │   │   └── route.ts
    │   ├── optimize-research
    │   │   └── route.ts
    │   ├── analyze-results
    │   │   └── route.ts
    │   ├── consolidate-report
    │   │   └── route.ts
    │   ├── report
    │   │   └── route.ts
    │   └── search
    │   │   └── route.ts
    └── globals.css
├── public
    ├── vercel.svg
    ├── window.svg
    ├── file.svg
    ├── globe.svg
    └── next.svg
├── next.config.ts
├── postcss.config.mjs
├── components
    ├── ui
    │   ├── skeleton.tsx
    │   ├── collapsible.tsx
    │   ├── input.tsx
    │   ├── separator.tsx
    │   ├── toaster.tsx
    │   ├── checkbox.tsx
    │   ├── tooltip.tsx
    │   ├── alert.tsx
    │   ├── scroll-area.tsx
    │   ├── tabs.tsx
    │   ├── button.tsx
    │   ├── card.tsx
    │   ├── dialog.tsx
    │   ├── sheet.tsx
    │   ├── toast.tsx
    │   ├── select.tsx
    │   └── dropdown-menu.tsx
    ├── analytics.tsx
    ├── flow
    │   ├── consolidated-edge.tsx
    │   ├── question-node.tsx
    │   ├── report-node.tsx
    │   ├── search-node.tsx
    │   ├── selection-node.tsx
    │   ├── project-actions.tsx
    │   └── project-selector.tsx
    ├── citations-footer.tsx
    ├── model-select.tsx
    ├── report-actions.tsx
    └── knowledge-base-sidebar.tsx
├── Dockerfile
├── components.json
├── eslint.config.mjs
├── hooks
    ├── use-mobile.tsx
    ├── use-knowledge-base.ts
    └── use-toast.ts
├── .gitignore
├── tsconfig.json
├── lib
    ├── redis.ts
    ├── gemini.ts
    ├── file-upload.ts
    ├── knowledge-base.ts
    ├── config.ts
    ├── utils.ts
    ├── models.ts
    ├── localStorage-utils.ts
    └── documents.ts
├── LICENSE
├── env.example
├── package.json
├── tailwind.config.ts
├── docs
    └── flow-projects.md
└── types
    └── index.ts


/.github/ISSUE_TEMPLATE/config.yml:
--------------------------------------------------------------------------------
1 | blank_issues_enabled: false
2 | 


--------------------------------------------------------------------------------
/demo.gif:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/btahir/open-deep-research/HEAD/demo.gif


--------------------------------------------------------------------------------
/app/favicon.ico:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/btahir/open-deep-research/HEAD/app/favicon.ico


--------------------------------------------------------------------------------
/app/apple-icon.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/btahir/open-deep-research/HEAD/app/apple-icon.png


--------------------------------------------------------------------------------
/app/opengraph-image.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/btahir/open-deep-research/HEAD/app/opengraph-image.png


--------------------------------------------------------------------------------
/public/vercel.svg:
--------------------------------------------------------------------------------
1 | <svg fill="none" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 1155 1000"><path d="m577.3 0 577.4 1000H0z" fill="#fff"/></svg>


--------------------------------------------------------------------------------
/next.config.ts:
--------------------------------------------------------------------------------
1 | import type { NextConfig } from "next";
2 | 
3 | const nextConfig: NextConfig = {
4 |   /* config options here */
5 | };
6 | 
7 | export default nextConfig;
8 | 


--------------------------------------------------------------------------------
/postcss.config.mjs:
--------------------------------------------------------------------------------
1 | /** @type {import('postcss-load-config').Config} */
2 | const config = {
3 |   plugins: {
4 |     tailwindcss: {},
5 |   },
6 | };
7 | 
8 | export default config;
9 | 


--------------------------------------------------------------------------------
/app/report/[id]/page.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react'
2 | import { ReportContent } from './report-content'
3 | 
4 | export default function ReportPage({ params }: any) {
5 |   return <ReportContent id={params.id} />
6 | }
7 | 


--------------------------------------------------------------------------------
/components/ui/skeleton.tsx:
--------------------------------------------------------------------------------
 1 | import { cn } from "@/lib/utils"
 2 | 
 3 | function Skeleton({
 4 |   className,
 5 |   ...props
 6 | }: React.HTMLAttributes<HTMLDivElement>) {
 7 |   return (
 8 |     <div
 9 |       className={cn("animate-pulse rounded-md bg-primary/10", className)}
10 |       {...props}
11 |     />
12 |   )
13 | }
14 | 
15 | export { Skeleton }
16 | 


--------------------------------------------------------------------------------
/public/window.svg:
--------------------------------------------------------------------------------
1 | <svg fill="none" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16"><path fill-rule="evenodd" clip-rule="evenodd" d="M1.5 2.5h13v10a1 1 0 0 1-1 1h-11a1 1 0 0 1-1-1zM0 1h16v11.5a2.5 2.5 0 0 1-2.5 2.5h-11A2.5 2.5 0 0 1 0 12.5zm3.75 4.5a.75.75 0 1 0 0-1.5.75.75 0 0 0 0 1.5M7 4.75a.75.75 0 1 1-1.5 0 .75.75 0 0 1 1.5 0m1.75.75a.75.75 0 1 0 0-1.5.75.75 0 0 0 0 1.5" fill="#666"/></svg>


--------------------------------------------------------------------------------
/public/file.svg:
--------------------------------------------------------------------------------
1 | <svg fill="none" viewBox="0 0 16 16" xmlns="http://www.w3.org/2000/svg"><path d="M14.5 13.5V5.41a1 1 0 0 0-.3-.7L9.8.29A1 1 0 0 0 9.08 0H1.5v13.5A2.5 2.5 0 0 0 4 16h8a2.5 2.5 0 0 0 2.5-2.5m-1.5 0v-7H8v-5H3v12a1 1 0 0 0 1 1h8a1 1 0 0 0 1-1M9.5 5V2.12L12.38 5zM5.13 5h-.62v1.25h2.12V5zm-.62 3h7.12v1.25H4.5zm.62 3h-.62v1.25h7.12V11z" clip-rule="evenodd" fill="#666" fill-rule="evenodd"/></svg>


--------------------------------------------------------------------------------
/components/ui/collapsible.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as CollapsiblePrimitive from "@radix-ui/react-collapsible"
 4 | 
 5 | const Collapsible = CollapsiblePrimitive.Root
 6 | 
 7 | const CollapsibleTrigger = CollapsiblePrimitive.CollapsibleTrigger
 8 | 
 9 | const CollapsibleContent = CollapsiblePrimitive.CollapsibleContent
10 | 
11 | export { Collapsible, CollapsibleTrigger, CollapsibleContent }
12 | 


--------------------------------------------------------------------------------
/Dockerfile:
--------------------------------------------------------------------------------
 1 | # Use Node.js Alpine as the base image
 2 | FROM node:20-alpine
 3 | 
 4 | # Set the working directory
 5 | WORKDIR /app
 6 | 
 7 | # Copy package.json and package-lock.json
 8 | COPY package.json package-lock.json ./
 9 | 
10 | # Install dependencies using npm
11 | RUN npm install
12 | 
13 | # Copy the rest of the application code
14 | COPY . .
15 | 
16 | # Build the application
17 | RUN npm run build
18 | 
19 | # Expose the port the app runs on
20 | EXPOSE 3000
21 | 
22 | # Start the application
23 | CMD ["npm", "start"]
24 | 


--------------------------------------------------------------------------------
/components.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "$schema": "https://ui.shadcn.com/schema.json",
 3 |   "style": "new-york",
 4 |   "rsc": true,
 5 |   "tsx": true,
 6 |   "tailwind": {
 7 |     "config": "tailwind.config.ts",
 8 |     "css": "app/globals.css",
 9 |     "baseColor": "neutral",
10 |     "cssVariables": true,
11 |     "prefix": ""
12 |   },
13 |   "aliases": {
14 |     "components": "@/components",
15 |     "utils": "@/lib/utils",
16 |     "ui": "@/components/ui",
17 |     "lib": "@/lib",
18 |     "hooks": "@/hooks"
19 |   },
20 |   "iconLibrary": "lucide"
21 | }


--------------------------------------------------------------------------------
/eslint.config.mjs:
--------------------------------------------------------------------------------
 1 | import { dirname } from 'path'
 2 | import { fileURLToPath } from 'url'
 3 | import { FlatCompat } from '@eslint/eslintrc'
 4 | 
 5 | const __filename = fileURLToPath(import.meta.url)
 6 | const __dirname = dirname(__filename)
 7 | 
 8 | const compat = new FlatCompat({
 9 |   baseDirectory: __dirname,
10 | })
11 | 
12 | const eslintConfig = [
13 |   ...compat.extends('next/core-web-vitals', 'next/typescript'),
14 |   {
15 |     rules: {
16 |       '@typescript-eslint/no-explicit-any': 'off',
17 |       '@typescript-eslint/ban-ts-comment': 'off',
18 |       'prefer-rest-params': 'off',
19 |     },
20 |   },
21 | ]
22 | 
23 | export default eslintConfig
24 | 


--------------------------------------------------------------------------------
/hooks/use-mobile.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | 
 3 | const MOBILE_BREAKPOINT = 768
 4 | 
 5 | export function useIsMobile() {
 6 |   const [isMobile, setIsMobile] = React.useState<boolean | undefined>(undefined)
 7 | 
 8 |   React.useEffect(() => {
 9 |     const mql = window.matchMedia(`(max-width: ${MOBILE_BREAKPOINT - 1}px)`)
10 |     const onChange = () => {
11 |       setIsMobile(window.innerWidth < MOBILE_BREAKPOINT)
12 |     }
13 |     mql.addEventListener("change", onChange)
14 |     setIsMobile(window.innerWidth < MOBILE_BREAKPOINT)
15 |     return () => mql.removeEventListener("change", onChange)
16 |   }, [])
17 | 
18 |   return !!isMobile
19 | }
20 | 


--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
 1 | # See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
 2 | 
 3 | # dependencies
 4 | /node_modules
 5 | /.pnp
 6 | .pnp.*
 7 | .yarn/*
 8 | !.yarn/patches
 9 | !.yarn/plugins
10 | !.yarn/releases
11 | !.yarn/versions
12 | 
13 | # testing
14 | /coverage
15 | 
16 | # next.js
17 | /.next/
18 | /out/
19 | 
20 | # production
21 | /build
22 | 
23 | # misc
24 | .DS_Store
25 | *.pem
26 | 
27 | # debug
28 | npm-debug.log*
29 | yarn-debug.log*
30 | yarn-error.log*
31 | .pnpm-debug.log*
32 | 
33 | # env files (can opt-in for committing if needed)
34 | .env*
35 | 
36 | # vercel
37 | .vercel
38 | 
39 | # typescript
40 | *.tsbuildinfo
41 | next-env.d.ts
42 | 


--------------------------------------------------------------------------------
/tsconfig.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "compilerOptions": {
 3 |     "target": "ES2017",
 4 |     "lib": ["dom", "dom.iterable", "esnext"],
 5 |     "allowJs": true,
 6 |     "skipLibCheck": true,
 7 |     "strict": true,
 8 |     "noEmit": true,
 9 |     "esModuleInterop": true,
10 |     "module": "esnext",
11 |     "moduleResolution": "bundler",
12 |     "resolveJsonModule": true,
13 |     "isolatedModules": true,
14 |     "jsx": "preserve",
15 |     "incremental": true,
16 |     "plugins": [
17 |       {
18 |         "name": "next"
19 |       }
20 |     ],
21 |     "paths": {
22 |       "@/*": ["./*"]
23 |     }
24 |   },
25 |   "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx", ".next/types/**/*.ts"],
26 |   "exclude": ["node_modules"]
27 | }
28 | 


--------------------------------------------------------------------------------
/components/analytics.tsx:
--------------------------------------------------------------------------------
 1 | 'use client'
 2 | 
 3 | import Script from 'next/script'
 4 | 
 5 | const GOOGLE_MEASUREMENT_ID =
 6 |   process.env.NEXT_PUBLIC_GOOGLE_MEASUREMENT_ID || ''
 7 | 
 8 | export function Analytics() {
 9 |   return (
10 |     <Script
11 |       src={`https://www.googletagmanager.com/gtag/js?id=${GOOGLE_MEASUREMENT_ID}`}
12 |       strategy='afterInteractive'
13 |       onLoad={() => {
14 |         // @ts-ignore
15 |         window.dataLayer = window.dataLayer || []
16 |         function gtag() {
17 |           // @ts-ignore
18 |           dataLayer.push(arguments)
19 |         }
20 |         // @ts-ignore
21 |         gtag('js', new Date())
22 |         // @ts-ignore
23 |         gtag('config', GOOGLE_MEASUREMENT_ID)
24 |       }}
25 |     />
26 |   )
27 | }
28 | 


--------------------------------------------------------------------------------
/components/ui/input.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | 
 3 | import { cn } from "@/lib/utils"
 4 | 
 5 | const Input = React.forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
 6 |   ({ className, type, ...props }, ref) => {
 7 |     return (
 8 |       <input
 9 |         type={type}
10 |         className={cn(
11 |           "flex h-9 w-full rounded-md border border-input bg-transparent px-3 py-1 text-base shadow-sm transition-colors file:border-0 file:bg-transparent file:text-sm file:font-medium file:text-foreground placeholder:text-muted-foreground focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:cursor-not-allowed disabled:opacity-50 md:text-sm",
12 |           className
13 |         )}
14 |         ref={ref}
15 |         {...props}
16 |       />
17 |     )
18 |   }
19 | )
20 | Input.displayName = "Input"
21 | 
22 | export { Input }
23 | 


--------------------------------------------------------------------------------
/components/ui/separator.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as SeparatorPrimitive from "@radix-ui/react-separator"
 5 | 
 6 | import { cn } from "@/lib/utils"
 7 | 
 8 | const Separator = React.forwardRef<
 9 |   React.ElementRef<typeof SeparatorPrimitive.Root>,
10 |   React.ComponentPropsWithoutRef<typeof SeparatorPrimitive.Root>
11 | >(
12 |   (
13 |     { className, orientation = "horizontal", decorative = true, ...props },
14 |     ref
15 |   ) => (
16 |     <SeparatorPrimitive.Root
17 |       ref={ref}
18 |       decorative={decorative}
19 |       orientation={orientation}
20 |       className={cn(
21 |         "shrink-0 bg-border",
22 |         orientation === "horizontal" ? "h-[1px] w-full" : "h-full w-[1px]",
23 |         className
24 |       )}
25 |       {...props}
26 |     />
27 |   )
28 | )
29 | Separator.displayName = SeparatorPrimitive.Root.displayName
30 | 
31 | export { Separator }
32 | 


--------------------------------------------------------------------------------
/lib/redis.ts:
--------------------------------------------------------------------------------
 1 | import { Redis } from '@upstash/redis'
 2 | import { Ratelimit } from '@upstash/ratelimit'
 3 | import { CONFIG } from './config'
 4 | 
 5 | export const redis = new Redis({
 6 |   url: process.env.UPSTASH_REDIS_REST_URL || '',
 7 |   token: process.env.UPSTASH_REDIS_REST_TOKEN || '',
 8 | })
 9 | 
10 | // Search: Configurable requests per minute
11 | export const searchRatelimit = new Ratelimit({
12 |   redis,
13 |   limiter: Ratelimit.slidingWindow(CONFIG.rateLimits.search, '1 m'),
14 | })
15 | 
16 | // Content fetching: Configurable requests per minute
17 | export const fetchContentRatelimit = new Ratelimit({
18 |   redis,
19 |   limiter: Ratelimit.slidingWindow(CONFIG.rateLimits.contentFetch, '1 m'),
20 | })
21 | 
22 | // Report generation: Configurable requests per minute
23 | export const reportContentRatelimit = new Ratelimit({
24 |   redis,
25 |   limiter: Ratelimit.slidingWindow(CONFIG.rateLimits.reportGeneration, '1 m'),
26 | })
27 | 


--------------------------------------------------------------------------------
/components/ui/toaster.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import { useToast } from "@/hooks/use-toast"
 4 | import {
 5 |   Toast,
 6 |   ToastClose,
 7 |   ToastDescription,
 8 |   ToastProvider,
 9 |   ToastTitle,
10 |   ToastViewport,
11 | } from "@/components/ui/toast"
12 | 
13 | export function Toaster() {
14 |   const { toasts } = useToast()
15 | 
16 |   return (
17 |     <ToastProvider>
18 |       {toasts.map(function ({ id, title, description, action, ...props }) {
19 |         return (
20 |           <Toast key={id} {...props}>
21 |             <div className="grid gap-1">
22 |               {title && <ToastTitle>{title}</ToastTitle>}
23 |               {description && (
24 |                 <ToastDescription>{description}</ToastDescription>
25 |               )}
26 |             </div>
27 |             {action}
28 |             <ToastClose />
29 |           </Toast>
30 |         )
31 |       })}
32 |       <ToastViewport />
33 |     </ToastProvider>
34 |   )
35 | }
36 | 


--------------------------------------------------------------------------------
/public/globe.svg:
--------------------------------------------------------------------------------
1 | <svg fill="none" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 16"><g clip-path="url(#a)"><path fill-rule="evenodd" clip-rule="evenodd" d="M10.27 14.1a6.5 6.5 0 0 0 3.67-3.45q-1.24.21-2.7.34-.31 1.83-.97 3.1M8 16A8 8 0 1 0 8 0a8 8 0 0 0 0 16m.48-1.52a7 7 0 0 1-.96 0H7.5a4 4 0 0 1-.84-1.32q-.38-.89-.63-2.08a40 40 0 0 0 3.92 0q-.25 1.2-.63 2.08a4 4 0 0 1-.84 1.31zm2.94-4.76q1.66-.15 2.95-.43a7 7 0 0 0 0-2.58q-1.3-.27-2.95-.43a18 18 0 0 1 0 3.44m-1.27-3.54a17 17 0 0 1 0 3.64 39 39 0 0 1-4.3 0 17 17 0 0 1 0-3.64 39 39 0 0 1 4.3 0m1.1-1.17q1.45.13 2.69.34a6.5 6.5 0 0 0-3.67-3.44q.65 1.26.98 3.1M8.48 1.5l.01.02q.41.37.84 1.31.38.89.63 2.08a40 40 0 0 0-3.92 0q.25-1.2.63-2.08a4 4 0 0 1 .85-1.32 7 7 0 0 1 .96 0m-2.75.4a6.5 6.5 0 0 0-3.67 3.44 29 29 0 0 1 2.7-.34q.31-1.83.97-3.1M4.58 6.28q-1.66.16-2.95.43a7 7 0 0 0 0 2.58q1.3.27 2.95.43a18 18 0 0 1 0-3.44m.17 4.71q-1.45-.12-2.69-.34a6.5 6.5 0 0 0 3.67 3.44q-.65-1.27-.98-3.1" fill="#666"/></g><defs><clipPath id="a"><path fill="#fff" d="M0 0h16v16H0z"/></clipPath></defs></svg>


--------------------------------------------------------------------------------
/app/layout.tsx:
--------------------------------------------------------------------------------
 1 | import type { Metadata } from 'next'
 2 | import { Geist, Zen_Dots } from 'next/font/google'
 3 | import { Toaster } from '@/components/ui/toaster'
 4 | import { Analytics } from '@/components/analytics'
 5 | 
 6 | import './globals.css'
 7 | 
 8 | const geistSans = Geist({
 9 |   variable: '--font-geist-sans',
10 |   subsets: ['latin'],
11 | })
12 | 
13 | const zenDots = Zen_Dots({
14 |   variable: '--font-zen-dots',
15 |   weight: ['400'],
16 |   style: ['normal'],
17 |   subsets: ['latin'],
18 |   display: 'swap',
19 | })
20 | 
21 | export const metadata: Metadata = {
22 |   title: 'Open Deep Research',
23 |   description:
24 |     'Open source alternative to Deep Research. Generate reports with AI based on search results.',
25 | }
26 | 
27 | export default function RootLayout({
28 |   children,
29 | }: Readonly<{
30 |   children: React.ReactNode
31 | }>) {
32 |   return (
33 |     <html lang='en'>
34 |       <Analytics />
35 |       <body className={`${geistSans.variable} ${zenDots.variable} antialiased`}>
36 |         {children}
37 |         <Toaster />
38 |       </body>
39 |     </html>
40 |   )
41 | }
42 | 


--------------------------------------------------------------------------------
/components/flow/consolidated-edge.tsx:
--------------------------------------------------------------------------------
 1 | import { getBezierPath, EdgeProps } from '@xyflow/react'
 2 | import { memo } from 'react'
 3 | 
 4 | // Define constants for edge styling
 5 | const EDGE_STYLES = {
 6 |   strokeWidth: 2,
 7 |   stroke: '#6366f1', // Indigo color for consolidated edges
 8 |   strokeDasharray: '5,5', // Dashed line pattern
 9 | }
10 | 
11 | export const ConsolidatedEdge = memo(function ConsolidatedEdge({
12 |   id,
13 |   sourceX,
14 |   sourceY,
15 |   targetX,
16 |   targetY,
17 |   sourcePosition,
18 |   targetPosition,
19 |   style = {},
20 |   markerEnd,
21 | }: EdgeProps) {
22 |   const [edgePath] = getBezierPath({
23 |     sourceX,
24 |     sourceY,
25 |     sourcePosition,
26 |     targetX,
27 |     targetY,
28 |     targetPosition,
29 |   })
30 | 
31 |   // Combine default edge styles with any custom styles
32 |   const combinedStyles = {
33 |     ...EDGE_STYLES,
34 |     ...style,
35 |   }
36 | 
37 |   return (
38 |     <path
39 |       id={id}
40 |       style={combinedStyles}
41 |       className='react-flow__edge-path'
42 |       d={edgePath}
43 |       markerEnd={markerEnd}
44 |     />
45 |   )
46 | })
47 | 


--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
 1 | MIT License
 2 | 
 3 | Copyright (c) 2024 Bilal
 4 | 
 5 | Permission is hereby granted, free of charge, to any person obtaining a copy
 6 | of this software and associated documentation files (the "Software"), to deal
 7 | in the Software without restriction, including without limitation the rights
 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 | 
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 | 
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.
22 | 


--------------------------------------------------------------------------------
/env.example:
--------------------------------------------------------------------------------
 1 | # Google Gemini Pro API key (required for AI report generation)
 2 | GEMINI_API_KEY=your_gemini_api_key
 3 | 
 4 | # OpenAI API key (optional - required only if OpenAI models are enabled)
 5 | OPENAI_API_KEY=your_openai_api_key
 6 | 
 7 | # Anthropic API key (optional - required only if Anthropic models are enabled)
 8 | ANTHROPIC_API_KEY=your_anthropic_api_key
 9 | 
10 | # DeepSeek API key (optional - required only if DeepSeek models are enabled)
11 | DEEPSEEK_API_KEY=your_deepseek_api_key
12 | 
13 | # Exa API key (Optional - if using Exa as search provider)
14 | EXA_API_KEY=your_exa_api_key
15 | 
16 | # Upstash Redis (required for rate limiting)
17 | UPSTASH_REDIS_REST_URL=your_upstash_redis_url
18 | UPSTASH_REDIS_REST_TOKEN=your_upstash_redis_token
19 | 
20 | # Bing Search API (Optional - if using Bing as search provider)
21 | AZURE_SUB_KEY=your-azure-subscription-key
22 | 
23 | # Google Custom Search API (Optional - if using Google as search provider)
24 | GOOGLE_SEARCH_API_KEY=your-google-search-api-key
25 | GOOGLE_SEARCH_CX=your-google-search-cx 
26 | 
27 | # Google Analytics Id
28 | NEXT_PUBLIC_GOOGLE_MEASUREMENT_ID=your-google-measurement-id
29 | 


--------------------------------------------------------------------------------
/components/ui/checkbox.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as CheckboxPrimitive from "@radix-ui/react-checkbox"
 5 | import { Check } from "lucide-react"
 6 | 
 7 | import { cn } from "@/lib/utils"
 8 | 
 9 | const Checkbox = React.forwardRef<
10 |   React.ElementRef<typeof CheckboxPrimitive.Root>,
11 |   React.ComponentPropsWithoutRef<typeof CheckboxPrimitive.Root>
12 | >(({ className, ...props }, ref) => (
13 |   <CheckboxPrimitive.Root
14 |     ref={ref}
15 |     className={cn(
16 |       "peer h-4 w-4 shrink-0 rounded-sm border border-primary shadow focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:cursor-not-allowed disabled:opacity-50 data-[state=checked]:bg-primary data-[state=checked]:text-primary-foreground",
17 |       className
18 |     )}
19 |     {...props}
20 |   >
21 |     <CheckboxPrimitive.Indicator
22 |       className={cn("flex items-center justify-center text-current")}
23 |     >
24 |       <Check className="h-4 w-4" />
25 |     </CheckboxPrimitive.Indicator>
26 |   </CheckboxPrimitive.Root>
27 | ))
28 | Checkbox.displayName = CheckboxPrimitive.Root.displayName
29 | 
30 | export { Checkbox }
31 | 


--------------------------------------------------------------------------------
/public/next.svg:
--------------------------------------------------------------------------------
1 | <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 394 80"><path fill="#000" d="M262 0h68.5v12.7h-27.2v66.6h-13.6V12.7H262V0ZM149 0v12.7H94v20.4h44.3v12.6H94v21h55v12.6H80.5V0h68.7zm34.3 0h-17.8l63.8 79.4h17.9l-32-39.7 32-39.6h-17.9l-23 28.6-23-28.6zm18.3 56.7-9-11-27.1 33.7h17.8l18.3-22.7z"/><path fill="#000" d="M81 79.3 17 0H0v79.3h13.6V17l50.2 62.3H81Zm252.6-.4c-1 0-1.8-.4-2.5-1s-1.1-1.6-1.1-2.6.3-1.8 1-2.5 1.6-1 2.6-1 1.8.3 2.5 1a3.4 3.4 0 0 1 .6 4.3 3.7 3.7 0 0 1-3 1.8zm23.2-33.5h6v23.3c0 2.1-.4 4-1.3 5.5a9.1 9.1 0 0 1-3.8 3.5c-1.6.8-3.5 1.3-5.7 1.3-2 0-3.7-.4-5.3-1s-2.8-1.8-3.7-3.2c-.9-1.3-1.4-3-1.4-5h6c.1.8.3 1.6.7 2.2s1 1.2 1.6 1.5c.7.4 1.5.5 2.4.5 1 0 1.8-.2 2.4-.6a4 4 0 0 0 1.6-1.8c.3-.8.5-1.8.5-3V45.5zm30.9 9.1a4.4 4.4 0 0 0-2-3.3 7.5 7.5 0 0 0-4.3-1.1c-1.3 0-2.4.2-3.3.5-.9.4-1.6 1-2 1.6a3.5 3.5 0 0 0-.3 4c.3.5.7.9 1.3 1.2l1.8 1 2 .5 3.2.8c1.3.3 2.5.7 3.7 1.2a13 13 0 0 1 3.2 1.8 8.1 8.1 0 0 1 3 6.5c0 2-.5 3.7-1.5 5.1a10 10 0 0 1-4.4 3.5c-1.8.8-4.1 1.2-6.8 1.2-2.6 0-4.9-.4-6.8-1.2-2-.8-3.4-2-4.5-3.5a10 10 0 0 1-1.7-5.6h6a5 5 0 0 0 3.5 4.6c1 .4 2.2.6 3.4.6 1.3 0 2.5-.2 3.5-.6 1-.4 1.8-1 2.4-1.7a4 4 0 0 0 .8-2.4c0-.9-.2-1.6-.7-2.2a11 11 0 0 0-2.1-1.4l-3.2-1-3.8-1c-2.8-.7-5-1.7-6.6-3.2a7.2 7.2 0 0 1-2.4-5.7 8 8 0 0 1 1.7-5 10 10 0 0 1 4.3-3.5c2-.8 4-1.2 6.4-1.2 2.3 0 4.4.4 6.2 1.2 1.8.8 3.2 2 4.3 3.4 1 1.4 1.5 3 1.5 5h-5.8z"/></svg>


--------------------------------------------------------------------------------
/components/ui/tooltip.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as TooltipPrimitive from "@radix-ui/react-tooltip"
 5 | 
 6 | import { cn } from "@/lib/utils"
 7 | 
 8 | const TooltipProvider = TooltipPrimitive.Provider
 9 | 
10 | const Tooltip = TooltipPrimitive.Root
11 | 
12 | const TooltipTrigger = TooltipPrimitive.Trigger
13 | 
14 | const TooltipContent = React.forwardRef<
15 |   React.ElementRef<typeof TooltipPrimitive.Content>,
16 |   React.ComponentPropsWithoutRef<typeof TooltipPrimitive.Content>
17 | >(({ className, sideOffset = 4, ...props }, ref) => (
18 |   <TooltipPrimitive.Portal>
19 |     <TooltipPrimitive.Content
20 |       ref={ref}
21 |       sideOffset={sideOffset}
22 |       className={cn(
23 |         "z-50 overflow-hidden rounded-md bg-primary px-3 py-1.5 text-xs text-primary-foreground animate-in fade-in-0 zoom-in-95 data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=closed]:zoom-out-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2",
24 |         className
25 |       )}
26 |       {...props}
27 |     />
28 |   </TooltipPrimitive.Portal>
29 | ))
30 | TooltipContent.displayName = TooltipPrimitive.Content.displayName
31 | 
32 | export { Tooltip, TooltipTrigger, TooltipContent, TooltipProvider }
33 | 


--------------------------------------------------------------------------------
/app/api/parse-document/route.ts:
--------------------------------------------------------------------------------
 1 | import { NextResponse } from 'next/server'
 2 | import { parseOfficeAsync } from 'officeparser'
 3 | 
 4 | export async function POST(request: Request) {
 5 |   try {
 6 |     const formData = await request.formData()
 7 |     const file = formData.get('file') as File
 8 | 
 9 |     if (!file) {
10 |       return NextResponse.json({ error: 'No file provided' }, { status: 400 })
11 |     }
12 | 
13 |     try {
14 |       // Convert the file to a Buffer
15 |       const arrayBuffer = await file.arrayBuffer()
16 |       const buffer = Buffer.from(arrayBuffer)
17 | 
18 |       // Configure officeparser
19 |       const config = {
20 |         outputErrorToConsole: false,
21 |         newlineDelimiter: '\n',
22 |         ignoreNotes: false,
23 |         putNotesAtLast: false,
24 |       }
25 | 
26 |       // Parse the document
27 |       const content = await parseOfficeAsync(buffer, config)
28 |       return NextResponse.json({ content })
29 |     } catch (error) {
30 |       console.error('Content extraction error:', error)
31 |       return NextResponse.json(
32 |         { error: 'Failed to extract content from document' },
33 |         { status: 500 }
34 |       )
35 |     }
36 |   } catch (error) {
37 |     console.error('Document parsing error:', error)
38 |     return NextResponse.json(
39 |       { error: 'Failed to parse document' },
40 |       { status: 500 }
41 |     )
42 |   }
43 | }
44 | 


--------------------------------------------------------------------------------
/.github/ISSUE_TEMPLATE/issue_template.yml:
--------------------------------------------------------------------------------
 1 | name: 'Issue with Pre-Submission Checklist'
 2 | description: "Before submitting your issue, please confirm you've followed these steps."
 3 | title: '[Issue] - '
 4 | labels: []
 5 | assignees: []
 6 | 
 7 | body:
 8 |   - type: markdown
 9 |     attributes:
10 |       value: |
11 |         **You are about to open an issue and interrupt the day of a stranger on the internet. And that's okay!**
12 | 
13 |         We all need help, and we should try to help each other as much as possible. But before you submit your issue, please make sure you've:
14 |         1. Read through the README.
15 |         2. Searched through existing GitHub issues to see if one of them answered your question.
16 |         3. Copied and pasted the README into ChatGPT or your favorite LLM of choice and asked it the question.
17 |         4. Spent 5 minutes thinking through this on your own.
18 | 
19 |         If all these failed, please feel free to open an issue. :)
20 | 
21 |   - type: checkboxes
22 |     id: confirmation
23 |     attributes:
24 |       label: "Please confirm that you've done the steps above:"
25 |       options:
26 |         - label: "Yes, I've read the README, searched existing issues, consulted an LLM, and thought it through."
27 |           required: true
28 | 
29 |   - type: textarea
30 |     id: issue_description
31 |     attributes:
32 |       label: 'Describe the issue'
33 |       description: 'Provide a clear and concise description of your issue.'
34 |       placeholder: 'Describe your issue here...'
35 | 


--------------------------------------------------------------------------------
/app/api/fetch-content/route.ts:
--------------------------------------------------------------------------------
 1 | import { NextResponse } from 'next/server'
 2 | import { fetchContentRatelimit } from '@/lib/redis'
 3 | import { CONFIG } from '@/lib/config'
 4 | 
 5 | export async function POST(request: Request) {
 6 |   try {
 7 |     const body = await request.json()
 8 |     const { url } = body
 9 | 
10 |     if (!url) {
11 |       return NextResponse.json({ error: 'URL is required' }, { status: 400 })
12 |     }
13 | 
14 |     // Only check rate limit if enabled
15 |     if (CONFIG.rateLimits.enabled) {
16 |       const { success } = await fetchContentRatelimit.limit(url)
17 |       if (!success) {
18 |         return NextResponse.json(
19 |           { error: 'Too many requests' },
20 |           { status: 429 }
21 |         )
22 |       }
23 |     }
24 | 
25 |     console.log('Fetching content for URL:', url)
26 | 
27 |     try {
28 |       const response = await fetch(
29 |         `https://r.jina.ai/${encodeURIComponent(url)}`
30 |       )
31 | 
32 |       if (!response.ok) {
33 |         console.warn(`Failed to fetch content for ${url}:`, response.status)
34 |         return NextResponse.json(
35 |           { error: 'Failed to fetch content' },
36 |           { status: response.status }
37 |         )
38 |       }
39 | 
40 |       const content = await response.text()
41 |       return NextResponse.json({ content })
42 |     } catch (error) {
43 |       console.warn(`Error fetching content for ${url}:`, error)
44 |       return NextResponse.json(
45 |         { error: 'Failed to fetch content' },
46 |         { status: 500 }
47 |       )
48 |     }
49 |   } catch (error) {
50 |     console.error('Content fetching error:', error)
51 |     return NextResponse.json({ error: 'Invalid request' }, { status: 400 })
52 |   }
53 | }
54 | 


--------------------------------------------------------------------------------
/components/citations-footer.tsx:
--------------------------------------------------------------------------------
 1 | import React from 'react'
 2 | import { type Report } from '@/types'
 3 | 
 4 | interface CitationsFooterProps {
 5 |   report: Report
 6 | }
 7 | 
 8 | export function CitationsFooter({ report }: CitationsFooterProps) {
 9 |   // Filter sources to only include those that were actually used
10 |   const filteredSources = React.useMemo(() => {
11 |     // If usedSources exists, filter the sources to only include those that were cited
12 |     if (report.usedSources && report.usedSources.length > 0 && report.sources) {
13 |       // Convert 1-based indices to 0-based for our array
14 |       const usedIndices = report.usedSources.map((num) => num - 1)
15 |       return report.sources.filter((_, index) =>
16 |         // Check if this source's index is in the usedSources array
17 |         usedIndices.includes(index)
18 |       )
19 |     }
20 |     // Default to all sources if no usedSources specified
21 |     return report.sources
22 |   }, [report.sources, report.usedSources])
23 | 
24 |   // If no filtered sources, don't render anything
25 |   if (!filteredSources || filteredSources.length === 0) {
26 |     return null
27 |   }
28 | 
29 |   return (
30 |     <div className='space-y-2 border-t pt-4 mt-6'>
31 |       <h3 className='text-xl font-semibold text-gray-700'>References</h3>
32 |       <ol className='list-decimal pl-5'>
33 |         {filteredSources.map((source) => (
34 |           <li key={source.id} className='mb-1'>
35 |             <a
36 |               href={source.url}
37 |               target='_blank'
38 |               rel='noopener noreferrer'
39 |               className='text-blue-600 hover:underline'
40 |             >
41 |               {source.name}
42 |             </a>
43 |           </li>
44 |         ))}
45 |       </ol>
46 |     </div>
47 |   )
48 | }
49 | 


--------------------------------------------------------------------------------
/lib/gemini.ts:
--------------------------------------------------------------------------------
 1 | import {
 2 |   GoogleGenerativeAI,
 3 |   HarmBlockThreshold,
 4 |   HarmCategory,
 5 | } from '@google/generative-ai'
 6 | 
 7 | const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY || '')
 8 | 
 9 | const generationJsonConfig = {
10 |   temperature: 1,
11 |   maxOutputTokens: 8192,
12 |   responseMimeType: 'application/json',
13 | }
14 | 
15 | const generationPlainTextConfig = {
16 |   temperature: 1,
17 |   maxOutputTokens: 8192,
18 |   responseMimeType: 'text/plain',
19 | }
20 | 
21 | const safetySettings = [
22 |   {
23 |     category: HarmCategory.HARM_CATEGORY_HARASSMENT,
24 |     threshold: HarmBlockThreshold.BLOCK_NONE,
25 |   },
26 |   {
27 |     category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
28 |     threshold: HarmBlockThreshold.BLOCK_NONE,
29 |   },
30 |   {
31 |     category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
32 |     threshold: HarmBlockThreshold.BLOCK_NONE,
33 |   },
34 |   {
35 |     category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
36 |     threshold: HarmBlockThreshold.BLOCK_NONE,
37 |   },
38 | ]
39 | 
40 | export const geminiFlashLiteModel = genAI.getGenerativeModel({
41 |   model: 'gemini-2.0-flash-lite-preview-02-05',
42 |   safetySettings,
43 |   generationConfig: generationJsonConfig,
44 | })
45 | 
46 | export const geminiFlashModel = genAI.getGenerativeModel({
47 |   model: 'gemini-2.0-flash',
48 |   safetySettings,
49 |   generationConfig: generationJsonConfig,
50 | })
51 | 
52 | export const geminiFlashThinkingModel = genAI.getGenerativeModel({
53 |   model: 'gemini-2.0-flash-thinking-exp-01-21',
54 |   safetySettings,
55 |   generationConfig: generationPlainTextConfig,
56 | })
57 | 
58 | export const geminiModel = genAI.getGenerativeModel({
59 |   model: 'gemini-2.0-pro-exp-02-05',
60 |   safetySettings,
61 |   generationConfig: generationJsonConfig,
62 | })
63 | 


--------------------------------------------------------------------------------
/components/ui/alert.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | import { cva, type VariantProps } from "class-variance-authority"
 3 | 
 4 | import { cn } from "@/lib/utils"
 5 | 
 6 | const alertVariants = cva(
 7 |   "relative w-full rounded-lg border px-4 py-3 text-sm [&>svg+div]:translate-y-[-3px] [&>svg]:absolute [&>svg]:left-4 [&>svg]:top-4 [&>svg]:text-foreground [&>svg~*]:pl-7",
 8 |   {
 9 |     variants: {
10 |       variant: {
11 |         default: "bg-background text-foreground",
12 |         destructive:
13 |           "border-destructive/50 text-destructive dark:border-destructive [&>svg]:text-destructive",
14 |       },
15 |     },
16 |     defaultVariants: {
17 |       variant: "default",
18 |     },
19 |   }
20 | )
21 | 
22 | const Alert = React.forwardRef<
23 |   HTMLDivElement,
24 |   React.HTMLAttributes<HTMLDivElement> & VariantProps<typeof alertVariants>
25 | >(({ className, variant, ...props }, ref) => (
26 |   <div
27 |     ref={ref}
28 |     role="alert"
29 |     className={cn(alertVariants({ variant }), className)}
30 |     {...props}
31 |   />
32 | ))
33 | Alert.displayName = "Alert"
34 | 
35 | const AlertTitle = React.forwardRef<
36 |   HTMLParagraphElement,
37 |   React.HTMLAttributes<HTMLHeadingElement>
38 | >(({ className, ...props }, ref) => (
39 |   <h5
40 |     ref={ref}
41 |     className={cn("mb-1 font-medium leading-none tracking-tight", className)}
42 |     {...props}
43 |   />
44 | ))
45 | AlertTitle.displayName = "AlertTitle"
46 | 
47 | const AlertDescription = React.forwardRef<
48 |   HTMLParagraphElement,
49 |   React.HTMLAttributes<HTMLParagraphElement>
50 | >(({ className, ...props }, ref) => (
51 |   <div
52 |     ref={ref}
53 |     className={cn("text-sm [&_p]:leading-relaxed", className)}
54 |     {...props}
55 |   />
56 | ))
57 | AlertDescription.displayName = "AlertDescription"
58 | 
59 | export { Alert, AlertTitle, AlertDescription }
60 | 


--------------------------------------------------------------------------------
/components/ui/scroll-area.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as ScrollAreaPrimitive from "@radix-ui/react-scroll-area"
 5 | 
 6 | import { cn } from "@/lib/utils"
 7 | 
 8 | const ScrollArea = React.forwardRef<
 9 |   React.ElementRef<typeof ScrollAreaPrimitive.Root>,
10 |   React.ComponentPropsWithoutRef<typeof ScrollAreaPrimitive.Root>
11 | >(({ className, children, ...props }, ref) => (
12 |   <ScrollAreaPrimitive.Root
13 |     ref={ref}
14 |     className={cn("relative overflow-hidden", className)}
15 |     {...props}
16 |   >
17 |     <ScrollAreaPrimitive.Viewport className="h-full w-full rounded-[inherit]">
18 |       {children}
19 |     </ScrollAreaPrimitive.Viewport>
20 |     <ScrollBar />
21 |     <ScrollAreaPrimitive.Corner />
22 |   </ScrollAreaPrimitive.Root>
23 | ))
24 | ScrollArea.displayName = ScrollAreaPrimitive.Root.displayName
25 | 
26 | const ScrollBar = React.forwardRef<
27 |   React.ElementRef<typeof ScrollAreaPrimitive.ScrollAreaScrollbar>,
28 |   React.ComponentPropsWithoutRef<typeof ScrollAreaPrimitive.ScrollAreaScrollbar>
29 | >(({ className, orientation = "vertical", ...props }, ref) => (
30 |   <ScrollAreaPrimitive.ScrollAreaScrollbar
31 |     ref={ref}
32 |     orientation={orientation}
33 |     className={cn(
34 |       "flex touch-none select-none transition-colors",
35 |       orientation === "vertical" &&
36 |         "h-full w-2.5 border-l border-l-transparent p-[1px]",
37 |       orientation === "horizontal" &&
38 |         "h-2.5 flex-col border-t border-t-transparent p-[1px]",
39 |       className
40 |     )}
41 |     {...props}
42 |   >
43 |     <ScrollAreaPrimitive.ScrollAreaThumb className="relative flex-1 rounded-full bg-border" />
44 |   </ScrollAreaPrimitive.ScrollAreaScrollbar>
45 | ))
46 | ScrollBar.displayName = ScrollAreaPrimitive.ScrollAreaScrollbar.displayName
47 | 
48 | export { ScrollArea, ScrollBar }
49 | 


--------------------------------------------------------------------------------
/package.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "name": "open-deep-research",
 3 |   "version": "0.1.0",
 4 |   "private": true,
 5 |   "scripts": {
 6 |     "dev": "next dev",
 7 |     "build": "next build",
 8 |     "start": "next start",
 9 |     "lint": "next lint"
10 |   },
11 |   "dependencies": {
12 |     "@anthropic-ai/sdk": "^0.18.0",
13 |     "@google/generative-ai": "^0.21.0",
14 |     "@radix-ui/react-checkbox": "^1.1.3",
15 |     "@radix-ui/react-collapsible": "^1.1.2",
16 |     "@radix-ui/react-dialog": "^1.1.4",
17 |     "@radix-ui/react-dropdown-menu": "^2.1.4",
18 |     "@radix-ui/react-scroll-area": "^1.2.2",
19 |     "@radix-ui/react-select": "^2.1.4",
20 |     "@radix-ui/react-separator": "^1.1.1",
21 |     "@radix-ui/react-slot": "^1.1.1",
22 |     "@radix-ui/react-tabs": "^1.1.2",
23 |     "@radix-ui/react-toast": "^1.2.4",
24 |     "@radix-ui/react-tooltip": "^1.1.6",
25 |     "@types/markdown-it": "^14.1.2",
26 |     "@upstash/ratelimit": "^2.0.5",
27 |     "@upstash/redis": "^1.34.3",
28 |     "@xyflow/react": "^12.4.3",
29 |     "class-variance-authority": "^0.7.1",
30 |     "clsx": "^2.1.1",
31 |     "date-fns": "^4.1.0",
32 |     "docx": "^9.1.0",
33 |     "jspdf": "^2.5.2",
34 |     "lucide-react": "^0.469.0",
35 |     "markdown-it": "^14.1.0",
36 |     "next": "15.1.11",
37 |     "officeparser": "^5.1.1",
38 |     "ollama": "^0.5.12",
39 |     "openai": "^4.80.1",
40 |     "react": "^19.0.0",
41 |     "react-dom": "^19.0.0",
42 |     "react-markdown": "^9.0.1",
43 |     "remark-gfm": "^4.0.0",
44 |     "tailwind-merge": "^2.6.0",
45 |     "tailwindcss-animate": "^1.0.7"
46 |   },
47 |   "devDependencies": {
48 |     "@eslint/eslintrc": "^3",
49 |     "@tailwindcss/typography": "^0.5.15",
50 |     "@types/node": "^20",
51 |     "@types/react": "^19",
52 |     "@types/react-dom": "^19",
53 |     "eslint": "^9",
54 |     "eslint-config-next": "15.1.2",
55 |     "postcss": "^8",
56 |     "tailwindcss": "^3.4.1",
57 |     "typescript": "^5"
58 |   }
59 | }
60 | 


--------------------------------------------------------------------------------
/components/model-select.tsx:
--------------------------------------------------------------------------------
 1 | 'use client'
 2 | 
 3 | import {
 4 |   Select,
 5 |   SelectContent,
 6 |   SelectItem,
 7 |   SelectTrigger,
 8 |   SelectValue,
 9 | } from '@/components/ui/select'
10 | import type { PlatformModel } from '@/types'
11 | import { CONFIG } from '@/lib/config'
12 | 
13 | const DEFAULT_MODEL = 'google__gemini-flash'
14 | 
15 | export const platformModels = Object.entries(CONFIG.platforms)
16 |   .flatMap(([platform, config]) => {
17 |     if (!config.enabled) return []
18 | 
19 |     return Object.entries(config.models).map(([modelId, modelConfig]) => {
20 |       return {
21 |         value: `${platform}__${modelId}`,
22 |         label: `${platform.charAt(0).toUpperCase() + platform.slice(1)} - ${
23 |           modelConfig.label
24 |         }`,
25 |         platform,
26 |         disabled: !modelConfig.enabled,
27 |       }
28 |     })
29 |   })
30 |   .filter(Boolean) as (PlatformModel & { disabled: boolean })[]
31 | 
32 | interface ModelSelectProps {
33 |   value: string
34 |   onValueChange: (value: string) => void
35 |   triggerClassName?: string
36 | }
37 | 
38 | export function ModelSelect({
39 |   value = DEFAULT_MODEL,
40 |   onValueChange,
41 |   triggerClassName,
42 | }: ModelSelectProps) {
43 |   return (
44 |     <Select
45 |       value={value}
46 |       onValueChange={onValueChange}
47 |       disabled={platformModels.length === 0}
48 |     >
49 |       <SelectTrigger className={triggerClassName}>
50 |         <SelectValue
51 |           placeholder={
52 |             platformModels.length === 0 ? 'No models available' : 'Select model'
53 |           }
54 |         />
55 |       </SelectTrigger>
56 |       <SelectContent>
57 |         {platformModels.map((model) => (
58 |           <SelectItem
59 |             key={model.value}
60 |             value={model.value}
61 |             disabled={model.disabled}
62 |             className={model.disabled ? 'text-gray-400 cursor-not-allowed' : ''}
63 |           >
64 |             {model.label}
65 |           </SelectItem>
66 |         ))}
67 |       </SelectContent>
68 |     </Select>
69 |   )
70 | }
71 | 
72 | export { DEFAULT_MODEL }
73 | 


--------------------------------------------------------------------------------
/components/ui/tabs.tsx:
--------------------------------------------------------------------------------
 1 | "use client"
 2 | 
 3 | import * as React from "react"
 4 | import * as TabsPrimitive from "@radix-ui/react-tabs"
 5 | 
 6 | import { cn } from "@/lib/utils"
 7 | 
 8 | const Tabs = TabsPrimitive.Root
 9 | 
10 | const TabsList = React.forwardRef<
11 |   React.ElementRef<typeof TabsPrimitive.List>,
12 |   React.ComponentPropsWithoutRef<typeof TabsPrimitive.List>
13 | >(({ className, ...props }, ref) => (
14 |   <TabsPrimitive.List
15 |     ref={ref}
16 |     className={cn(
17 |       "inline-flex h-9 items-center justify-center rounded-lg bg-muted p-1 text-muted-foreground",
18 |       className
19 |     )}
20 |     {...props}
21 |   />
22 | ))
23 | TabsList.displayName = TabsPrimitive.List.displayName
24 | 
25 | const TabsTrigger = React.forwardRef<
26 |   React.ElementRef<typeof TabsPrimitive.Trigger>,
27 |   React.ComponentPropsWithoutRef<typeof TabsPrimitive.Trigger>
28 | >(({ className, ...props }, ref) => (
29 |   <TabsPrimitive.Trigger
30 |     ref={ref}
31 |     className={cn(
32 |       "inline-flex items-center justify-center whitespace-nowrap rounded-md px-3 py-1 text-sm font-medium ring-offset-background transition-all focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 disabled:pointer-events-none disabled:opacity-50 data-[state=active]:bg-background data-[state=active]:text-foreground data-[state=active]:shadow",
33 |       className
34 |     )}
35 |     {...props}
36 |   />
37 | ))
38 | TabsTrigger.displayName = TabsPrimitive.Trigger.displayName
39 | 
40 | const TabsContent = React.forwardRef<
41 |   React.ElementRef<typeof TabsPrimitive.Content>,
42 |   React.ComponentPropsWithoutRef<typeof TabsPrimitive.Content>
43 | >(({ className, ...props }, ref) => (
44 |   <TabsPrimitive.Content
45 |     ref={ref}
46 |     className={cn(
47 |       "mt-2 ring-offset-background focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2",
48 |       className
49 |     )}
50 |     {...props}
51 |   />
52 | ))
53 | TabsContent.displayName = TabsPrimitive.Content.displayName
54 | 
55 | export { Tabs, TabsList, TabsTrigger, TabsContent }
56 | 


--------------------------------------------------------------------------------
/components/ui/button.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | import { Slot } from "@radix-ui/react-slot"
 3 | import { cva, type VariantProps } from "class-variance-authority"
 4 | 
 5 | import { cn } from "@/lib/utils"
 6 | 
 7 | const buttonVariants = cva(
 8 |   "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-colors focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0",
 9 |   {
10 |     variants: {
11 |       variant: {
12 |         default:
13 |           "bg-primary text-primary-foreground shadow hover:bg-primary/90",
14 |         destructive:
15 |           "bg-destructive text-destructive-foreground shadow-sm hover:bg-destructive/90",
16 |         outline:
17 |           "border border-input bg-background shadow-sm hover:bg-accent hover:text-accent-foreground",
18 |         secondary:
19 |           "bg-secondary text-secondary-foreground shadow-sm hover:bg-secondary/80",
20 |         ghost: "hover:bg-accent hover:text-accent-foreground",
21 |         link: "text-primary underline-offset-4 hover:underline",
22 |       },
23 |       size: {
24 |         default: "h-9 px-4 py-2",
25 |         sm: "h-8 rounded-md px-3 text-xs",
26 |         lg: "h-10 rounded-md px-8",
27 |         icon: "h-9 w-9",
28 |       },
29 |     },
30 |     defaultVariants: {
31 |       variant: "default",
32 |       size: "default",
33 |     },
34 |   }
35 | )
36 | 
37 | export interface ButtonProps
38 |   extends React.ButtonHTMLAttributes<HTMLButtonElement>,
39 |     VariantProps<typeof buttonVariants> {
40 |   asChild?: boolean
41 | }
42 | 
43 | const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
44 |   ({ className, variant, size, asChild = false, ...props }, ref) => {
45 |     const Comp = asChild ? Slot : "button"
46 |     return (
47 |       <Comp
48 |         className={cn(buttonVariants({ variant, size, className }))}
49 |         ref={ref}
50 |         {...props}
51 |       />
52 |     )
53 |   }
54 | )
55 | Button.displayName = "Button"
56 | 
57 | export { Button, buttonVariants }
58 | 


--------------------------------------------------------------------------------
/components/ui/card.tsx:
--------------------------------------------------------------------------------
 1 | import * as React from "react"
 2 | 
 3 | import { cn } from "@/lib/utils"
 4 | 
 5 | const Card = React.forwardRef<
 6 |   HTMLDivElement,
 7 |   React.HTMLAttributes<HTMLDivElement>
 8 | >(({ className, ...props }, ref) => (
 9 |   <div
10 |     ref={ref}
11 |     className={cn(
12 |       "rounded-xl border bg-card text-card-foreground shadow",
13 |       className
14 |     )}
15 |     {...props}
16 |   />
17 | ))
18 | Card.displayName = "Card"
19 | 
20 | const CardHeader = React.forwardRef<
21 |   HTMLDivElement,
22 |   React.HTMLAttributes<HTMLDivElement>
23 | >(({ className, ...props }, ref) => (
24 |   <div
25 |     ref={ref}
26 |     className={cn("flex flex-col space-y-1.5 p-6", className)}
27 |     {...props}
28 |   />
29 | ))
30 | CardHeader.displayName = "CardHeader"
31 | 
32 | const CardTitle = React.forwardRef<
33 |   HTMLDivElement,
34 |   React.HTMLAttributes<HTMLDivElement>
35 | >(({ className, ...props }, ref) => (
36 |   <div
37 |     ref={ref}
38 |     className={cn("font-semibold leading-none tracking-tight", className)}
39 |     {...props}
40 |   />
41 | ))
42 | CardTitle.displayName = "CardTitle"
43 | 
44 | const CardDescription = React.forwardRef<
45 |   HTMLDivElement,
46 |   React.HTMLAttributes<HTMLDivElement>
47 | >(({ className, ...props }, ref) => (
48 |   <div
49 |     ref={ref}
50 |     className={cn("text-sm text-muted-foreground", className)}
51 |     {...props}
52 |   />
53 | ))
54 | CardDescription.displayName = "CardDescription"
55 | 
56 | const CardContent = React.forwardRef<
57 |   HTMLDivElement,
58 |   React.HTMLAttributes<HTMLDivElement>
59 | >(({ className, ...props }, ref) => (
60 |   <div ref={ref} className={cn("p-6 pt-0", className)} {...props} />
61 | ))
62 | CardContent.displayName = "CardContent"
63 | 
64 | const CardFooter = React.forwardRef<
65 |   HTMLDivElement,
66 |   React.HTMLAttributes<HTMLDivElement>
67 | >(({ className, ...props }, ref) => (
68 |   <div
69 |     ref={ref}
70 |     className={cn("flex items-center p-6 pt-0", className)}
71 |     {...props}
72 |   />
73 | ))
74 | CardFooter.displayName = "CardFooter"
75 | 
76 | export { Card, CardHeader, CardFooter, CardTitle, CardDescription, CardContent }
77 | 


--------------------------------------------------------------------------------
/app/api/download/route.ts:
--------------------------------------------------------------------------------
 1 | import { NextResponse } from 'next/server'
 2 | import { generateDocx, generatePdf } from '@/lib/documents'
 3 | import { type Report } from '@/types'
 4 | 
 5 | export async function POST(request: Request) {
 6 |   try {
 7 |     const body = await request.json()
 8 |     console.log('Download request received:', {
 9 |       format: body.format,
10 |       reportTitle: body.report?.title
11 |     })
12 | 
13 |     const { report, format } = body as { report: Report; format: 'pdf' | 'docx' | 'txt' }
14 | 
15 |     let content: string | Buffer
16 |     const headers = new Headers()
17 | 
18 |     switch (format) {
19 |       case 'pdf':
20 |         console.log('Generating PDF')
21 |         content = await generatePdf(report)
22 |         headers.set('Content-Type', 'application/pdf')
23 |         break
24 | 
25 |       case 'docx':
26 |         console.log('Generating DOCX')
27 |         content = await generateDocx(report)
28 |         headers.set(
29 |           'Content-Type',
30 |           'application/vnd.openxmlformats-officedocument.wordprocessingml.document'
31 |         )
32 |         break
33 | 
34 |       case 'txt':
35 |       default:
36 |         console.log('Generating TXT')
37 |         content = `
38 | ${report.title}
39 | 
40 | ${report.summary}
41 | 
42 | ${report.sections
43 |   .map(
44 |     (section) => `
45 | ${section.title}
46 | ${section.content}
47 | `
48 |   )
49 |   .join('\n')}
50 | `.trim()
51 |         headers.set('Content-Type', 'text/plain')
52 |         break
53 |     }
54 | 
55 |     console.log(`Generated ${format} content, size:`, content.length)
56 |     headers.set('Content-Disposition', `attachment; filename=report.${format}`)
57 | 
58 |     return new Response(content, {
59 |       headers,
60 |       status: 200,
61 |     })
62 |   } catch (error) {
63 |     console.error('Download generation error:', error)
64 |     if (error instanceof Error) {
65 |       console.error('Error details:', {
66 |         message: error.message,
67 |         stack: error.stack,
68 |         name: error.name
69 |       })
70 |     }
71 |     return NextResponse.json(
72 |       { error: 'Failed to generate download' },
73 |       { status: 500 }
74 |     )
75 |   }
76 | } 


--------------------------------------------------------------------------------
/lib/file-upload.ts:
--------------------------------------------------------------------------------
 1 | import { type SearchResult } from '@/types'
 2 | 
 3 | export async function handleLocalFile(
 4 |   file: File,
 5 |   onStatusUpdate?: (loading: boolean) => void,
 6 |   onError?: (error: unknown, context: string) => void
 7 | ): Promise<SearchResult | null> {
 8 |   try {
 9 |     onStatusUpdate?.(true)
10 | 
11 |     let content = ''
12 |     if (file.type === 'text/plain' || file.name.endsWith('.txt')) {
13 |       content = await file.text()
14 |     } else if (
15 |       file.type === 'application/pdf' ||
16 |       file.name.endsWith('.pdf') ||
17 |       file.type ===
18 |         'application/vnd.openxmlformats-officedocument.wordprocessingml.document' ||
19 |       file.name.endsWith('.docx')
20 |     ) {
21 |       // Send the file to our parsing endpoint
22 |       const formData = new FormData()
23 |       formData.append('file', file)
24 | 
25 |       const response = await fetch('/api/parse-document', {
26 |         method: 'POST',
27 |         body: formData,
28 |       })
29 | 
30 |       if (!response.ok) {
31 |         const errorData = await response
32 |           .json()
33 |           .catch(() => ({ error: 'Failed to parse document' }))
34 |         throw new Error(errorData.error || 'Failed to parse document')
35 |       }
36 | 
37 |       const data = await response.json()
38 |       content = data.content
39 |     } else {
40 |       throw new Error(
41 |         'Unsupported file type. Only TXT, PDF, and DOCX files are supported.'
42 |       )
43 |     }
44 | 
45 |     // Truncate content to a reasonable snippet size
46 |     const snippet = content.slice(0, 500) + (content.length > 500 ? '...' : '')
47 | 
48 |     // Create a search result from the file
49 |     const timestamp = Date.now()
50 |     const newResult: SearchResult = {
51 |       id: `file-${timestamp}-${file.name}`,
52 |       url: URL.createObjectURL(file),
53 |       name: file.name,
54 |       snippet: snippet,
55 |       isCustomUrl: true,
56 |       content: content, // Store full content for report generation
57 |     }
58 | 
59 |     return newResult
60 |   } catch (error) {
61 |     onError?.(error, 'File Upload Error')
62 |     return null
63 |   } finally {
64 |     onStatusUpdate?.(false)
65 |   }
66 | }
67 | 
68 | export const SUPPORTED_FILE_TYPES =
69 |   '.txt,.pdf,.docx,text/plain,application/pdf,application/vnd.openxmlformats-officedocument.wordprocessingml.document'
70 | 


--------------------------------------------------------------------------------
/hooks/use-knowledge-base.ts:
--------------------------------------------------------------------------------
 1 | import { useState, useEffect } from 'react'
 2 | import { type KnowledgeBaseReport, type Report } from '@/types'
 3 | import {
 4 |   addToKnowledgeBase,
 5 |   getKnowledgeBaseReports,
 6 |   deleteFromKnowledgeBase,
 7 |   searchKnowledgeBase,
 8 | } from '@/lib/knowledge-base'
 9 | 
10 | export function useKnowledgeBase() {
11 |   const [reports, setReports] = useState<KnowledgeBaseReport[]>([])
12 | 
13 |   useEffect(() => {
14 |     // Load reports on mount
15 |     setReports(getKnowledgeBaseReports())
16 | 
17 |     // Listen for storage events from other tabs/windows
18 |     const handleStorageChange = (e: StorageEvent) => {
19 |       if (e.key === 'knowledge_base') {
20 |         setReports(getKnowledgeBaseReports())
21 |       }
22 |     }
23 | 
24 |     // Listen for custom knowledge base change events (same window)
25 |     const handleKnowledgeBaseChange = () => {
26 |       setReports(getKnowledgeBaseReports())
27 |     }
28 | 
29 |     // Listen for both storage and custom events
30 |     window.addEventListener('storage', handleStorageChange)
31 |     window.addEventListener('knowledge_base_change', handleKnowledgeBaseChange)
32 | 
33 |     return () => {
34 |       window.removeEventListener('storage', handleStorageChange)
35 |       window.removeEventListener('knowledge_base_change', handleKnowledgeBaseChange)
36 |     }
37 |   }, [])
38 | 
39 |   const addReport = (report: Report, query: string) => {
40 |     const newReport: KnowledgeBaseReport = {
41 |       id: `kb-${Date.now()}`,
42 |       timestamp: Date.now(),
43 |       query,
44 |       report,
45 |     }
46 | 
47 |     const success = addToKnowledgeBase(newReport)
48 |     if (success) {
49 |       setReports(getKnowledgeBaseReports())
50 |     }
51 |     return success
52 |   }
53 | 
54 |   const deleteReport = (reportId: string) => {
55 |     const success = deleteFromKnowledgeBase(reportId)
56 |     if (success) {
57 |       setReports(getKnowledgeBaseReports())
58 |     }
59 |     return success
60 |   }
61 | 
62 |   const searchReports = (query: string) => {
63 |     return searchKnowledgeBase(query)
64 |   }
65 | 
66 |   const clearAllReports = () => {
67 |     localStorage.removeItem('knowledge_base')
68 |     setReports([])
69 |     return true
70 |   }
71 | 
72 |   return {
73 |     reports,
74 |     addReport,
75 |     deleteReport,
76 |     searchReports,
77 |     clearAllReports,
78 |   }
79 | } 


--------------------------------------------------------------------------------
/lib/knowledge-base.ts:
--------------------------------------------------------------------------------
 1 | import { type KnowledgeBaseReport } from '@/types'
 2 | 
 3 | const KNOWLEDGE_BASE_KEY = 'knowledge_base'
 4 | 
 5 | // Helper to notify about knowledge base changes
 6 | const notifyKnowledgeBaseChange = () => {
 7 |   if (typeof window !== 'undefined') {
 8 |     window.dispatchEvent(new Event('knowledge_base_change'))
 9 |   }
10 | }
11 | 
12 | export function addToKnowledgeBase(report: KnowledgeBaseReport): boolean {
13 |   try {
14 |     const reports = getKnowledgeBaseReports()
15 |     reports.push(report)
16 |     localStorage.setItem(KNOWLEDGE_BASE_KEY, JSON.stringify(reports))
17 |     notifyKnowledgeBaseChange()
18 |     return true
19 |   } catch (error) {
20 |     console.error('Failed to add report to knowledge base:', error)
21 |     return false
22 |   }
23 | }
24 | 
25 | export function getKnowledgeBaseReports(): KnowledgeBaseReport[] {
26 |   try {
27 |     const reportsJson = localStorage.getItem(KNOWLEDGE_BASE_KEY)
28 |     if (!reportsJson) return []
29 |     
30 |     const reports = JSON.parse(reportsJson) as KnowledgeBaseReport[]
31 |     return reports.sort((a, b) => b.timestamp - a.timestamp)
32 |   } catch (error) {
33 |     console.error('Failed to get knowledge base reports:', error)
34 |     return []
35 |   }
36 | }
37 | 
38 | export function deleteFromKnowledgeBase(reportId: string): boolean {
39 |   try {
40 |     const reports = getKnowledgeBaseReports()
41 |     const filteredReports = reports.filter(report => report.id !== reportId)
42 |     localStorage.setItem(KNOWLEDGE_BASE_KEY, JSON.stringify(filteredReports))
43 |     notifyKnowledgeBaseChange()
44 |     return true
45 |   } catch (error) {
46 |     console.error('Failed to delete report from knowledge base:', error)
47 |     return false
48 |   }
49 | }
50 | 
51 | export function searchKnowledgeBase(query: string): KnowledgeBaseReport[] {
52 |   try {
53 |     const reports = getKnowledgeBaseReports()
54 |     
55 |     // Simple search implementation - can be enhanced with better search logic
56 |     const searchTerms = query.toLowerCase().split(' ')
57 |     return reports.filter(report => {
58 |       const searchText = `${report.query} ${report.report.title} ${report.report.summary}`.toLowerCase()
59 |       return searchTerms.every(term => searchText.includes(term))
60 |     })
61 |   } catch (error) {
62 |     console.error('Failed to search knowledge base:', error)
63 |     return []
64 |   }
65 | } 


--------------------------------------------------------------------------------
/tailwind.config.ts:
--------------------------------------------------------------------------------
 1 | import type { Config } from 'tailwindcss'
 2 | 
 3 | export default {
 4 |   darkMode: ['class'],
 5 |   content: [
 6 |     './pages/**/*.{js,ts,jsx,tsx,mdx}',
 7 |     './components/**/*.{js,ts,jsx,tsx,mdx}',
 8 |     './app/**/*.{js,ts,jsx,tsx,mdx}',
 9 |   ],
10 |   theme: {
11 |   	extend: {
12 |   		fontFamily: {
13 |   			sans: [
14 |   				'var(--font-geist-sans)'
15 |   			],
16 |   			heading: [
17 |   				'var(--font-zen-dots)'
18 |   			]
19 |   		},
20 |   		colors: {
21 |   			background: 'hsl(var(--background))',
22 |   			foreground: 'hsl(var(--foreground))',
23 |   			card: {
24 |   				DEFAULT: 'hsl(var(--card))',
25 |   				foreground: 'hsl(var(--card-foreground))'
26 |   			},
27 |   			popover: {
28 |   				DEFAULT: 'hsl(var(--popover))',
29 |   				foreground: 'hsl(var(--popover-foreground))'
30 |   			},
31 |   			primary: {
32 |   				DEFAULT: 'hsl(var(--primary))',
33 |   				foreground: 'hsl(var(--primary-foreground))'
34 |   			},
35 |   			secondary: {
36 |   				DEFAULT: 'hsl(var(--secondary))',
37 |   				foreground: 'hsl(var(--secondary-foreground))'
38 |   			},
39 |   			muted: {
40 |   				DEFAULT: 'hsl(var(--muted))',
41 |   				foreground: 'hsl(var(--muted-foreground))'
42 |   			},
43 |   			accent: {
44 |   				DEFAULT: 'hsl(var(--accent))',
45 |   				foreground: 'hsl(var(--accent-foreground))'
46 |   			},
47 |   			destructive: {
48 |   				DEFAULT: 'hsl(var(--destructive))',
49 |   				foreground: 'hsl(var(--destructive-foreground))'
50 |   			},
51 |   			border: 'hsl(var(--border))',
52 |   			input: 'hsl(var(--input))',
53 |   			ring: 'hsl(var(--ring))',
54 |   			chart: {
55 |   				'1': 'hsl(var(--chart-1))',
56 |   				'2': 'hsl(var(--chart-2))',
57 |   				'3': 'hsl(var(--chart-3))',
58 |   				'4': 'hsl(var(--chart-4))',
59 |   				'5': 'hsl(var(--chart-5))'
60 |   			},
61 |   			sidebar: {
62 |   				DEFAULT: 'hsl(var(--sidebar-background))',
63 |   				foreground: 'hsl(var(--sidebar-foreground))',
64 |   				primary: 'hsl(var(--sidebar-primary))',
65 |   				'primary-foreground': 'hsl(var(--sidebar-primary-foreground))',
66 |   				accent: 'hsl(var(--sidebar-accent))',
67 |   				'accent-foreground': 'hsl(var(--sidebar-accent-foreground))',
68 |   				border: 'hsl(var(--sidebar-border))',
69 |   				ring: 'hsl(var(--sidebar-ring))'
70 |   			}
71 |   		},
72 |   		borderRadius: {
73 |   			lg: 'var(--radius)',
74 |   			md: 'calc(var(--radius) - 2px)',
75 |   			sm: 'calc(var(--radius) - 4px)'
76 |   		}
77 |   	}
78 |   },
79 |   plugins: [require('tailwindcss-animate'), require('@tailwindcss/typography')],
80 | } satisfies Config
81 | 


--------------------------------------------------------------------------------
/lib/config.ts:
--------------------------------------------------------------------------------
  1 | export const CONFIG = {
  2 |   // Rate limits (requests per minute)
  3 |   rateLimits: {
  4 |     enabled: true, // Flag to enable/disable rate limiting
  5 |     search: 10,
  6 |     contentFetch: 20,
  7 |     reportGeneration: 5,
  8 |     agentOptimizations: 10,
  9 |   },
 10 | 
 11 |   // Search settings
 12 |   search: {
 13 |     resultsPerPage: 10,
 14 |     maxSelectableResults: 3,
 15 |     provider: 'google' as 'google' | 'bing' | 'exa', // Default search provider
 16 |     safeSearch: {
 17 |       google: 'active' as 'active' | 'off',
 18 |       bing: 'moderate' as 'moderate' | 'strict' | 'off',
 19 |     },
 20 |     market: 'en-US',
 21 |   },
 22 | 
 23 |   // AI Platform settings
 24 |   platforms: {
 25 |     google: {
 26 |       enabled: true,
 27 |       models: {
 28 |         'gemini-flash': {
 29 |           enabled: true,
 30 |           label: 'Gemini Flash',
 31 |         },
 32 |         'gemini-flash-thinking': {
 33 |           enabled: true,
 34 |           label: 'Gemini Flash Thinking',
 35 |         },
 36 |         'gemini-exp': {
 37 |           enabled: false,
 38 |           label: 'Gemini Exp',
 39 |         },
 40 |       },
 41 |     },
 42 |     ollama: {
 43 |       enabled: true,
 44 |       models: {
 45 |         'deepseek-r1:1.5b': {
 46 |           enabled: false,
 47 |           label: 'DeepSeek R1 1.5B',
 48 |         },
 49 |       },
 50 |     },
 51 |     openai: {
 52 |       enabled: true,
 53 |       models: {
 54 |         'gpt-4o': {
 55 |           enabled: false,
 56 |           label: 'GPT-4o',
 57 |         },
 58 |         'o1-mini': {
 59 |           enabled: false,
 60 |           label: 'o1-mini',
 61 |         },
 62 |         o1: {
 63 |           enabled: false,
 64 |           label: 'o1',
 65 |         },
 66 |       },
 67 |     },
 68 |     anthropic: {
 69 |       enabled: true,
 70 |       models: {
 71 |         'claude-3-7-sonnet-latest': {
 72 |           enabled: false,
 73 |           label: 'Claude 3.7 Sonnet',
 74 |         },
 75 |         'claude-3-5-haiku-latest': {
 76 |           enabled: false,
 77 |           label: 'Claude 3.5 Haiku',
 78 |         },
 79 |       },
 80 |     },
 81 |     deepseek: {
 82 |       enabled: true,
 83 |       models: {
 84 |         chat: {
 85 |           enabled: false,
 86 |           label: 'Chat',
 87 |         },
 88 |         reasoner: {
 89 |           enabled: false,
 90 |           label: 'Reasoner',
 91 |         },
 92 |       },
 93 |     },
 94 |     openrouter: {
 95 |       enabled: true,
 96 |       models: {
 97 |         'openrouter/auto': {
 98 |           enabled: false,
 99 |           label: 'Auto',
100 |         },
101 |       },
102 |     },
103 |   },
104 | } as const
105 | 


--------------------------------------------------------------------------------
/app/globals.css:
--------------------------------------------------------------------------------
 1 | @tailwind base;
 2 | @tailwind components;
 3 | @tailwind utilities;
 4 | 
 5 | body {
 6 |   font-family: Arial, Helvetica, sans-serif;
 7 | }
 8 | 
 9 | @layer base {
10 |   :root {
11 |     --background: 0 0% 100%;
12 |     --foreground: 0 0% 3.9%;
13 |     --card: 0 0% 100%;
14 |     --card-foreground: 0 0% 3.9%;
15 |     --popover: 0 0% 100%;
16 |     --popover-foreground: 0 0% 3.9%;
17 |     --primary: 0 0% 9%;
18 |     --primary-foreground: 0 0% 98%;
19 |     --secondary: 0 0% 96.1%;
20 |     --secondary-foreground: 0 0% 9%;
21 |     --muted: 0 0% 96.1%;
22 |     --muted-foreground: 0 0% 45.1%;
23 |     --accent: 0 0% 96.1%;
24 |     --accent-foreground: 0 0% 9%;
25 |     --destructive: 0 84.2% 60.2%;
26 |     --destructive-foreground: 0 0% 98%;
27 |     --border: 0 0% 89.8%;
28 |     --input: 0 0% 89.8%;
29 |     --ring: 0 0% 3.9%;
30 |     --chart-1: 12 76% 61%;
31 |     --chart-2: 173 58% 39%;
32 |     --chart-3: 197 37% 24%;
33 |     --chart-4: 43 74% 66%;
34 |     --chart-5: 27 87% 67%;
35 |     --radius: 0.5rem;
36 |     --sidebar-background: 0 0% 98%;
37 |     --sidebar-foreground: 240 5.3% 26.1%;
38 |     --sidebar-primary: 240 5.9% 10%;
39 |     --sidebar-primary-foreground: 0 0% 98%;
40 |     --sidebar-accent: 240 4.8% 95.9%;
41 |     --sidebar-accent-foreground: 240 5.9% 10%;
42 |     --sidebar-border: 220 13% 91%;
43 |     --sidebar-ring: 217.2 91.2% 59.8%;
44 |   }
45 |   .dark {
46 |     --background: 0 0% 3.9%;
47 |     --foreground: 0 0% 98%;
48 |     --card: 0 0% 3.9%;
49 |     --card-foreground: 0 0% 98%;
50 |     --popover: 0 0% 3.9%;
51 |     --popover-foreground: 0 0% 98%;
52 |     --primary: 0 0% 98%;
53 |     --primary-foreground: 0 0% 9%;
54 |     --secondary: 0 0% 14.9%;
55 |     --secondary-foreground: 0 0% 98%;
56 |     --muted: 0 0% 14.9%;
57 |     --muted-foreground: 0 0% 63.9%;
58 |     --accent: 0 0% 14.9%;
59 |     --accent-foreground: 0 0% 98%;
60 |     --destructive: 0 62.8% 30.6%;
61 |     --destructive-foreground: 0 0% 98%;
62 |     --border: 0 0% 14.9%;
63 |     --input: 0 0% 14.9%;
64 |     --ring: 0 0% 83.1%;
65 |     --chart-1: 220 70% 50%;
66 |     --chart-2: 160 60% 45%;
67 |     --chart-3: 30 80% 55%;
68 |     --chart-4: 280 65% 60%;
69 |     --chart-5: 340 75% 55%;
70 |     --sidebar-background: 240 5.9% 10%;
71 |     --sidebar-foreground: 240 4.8% 95.9%;
72 |     --sidebar-primary: 224.3 76.3% 48%;
73 |     --sidebar-primary-foreground: 0 0% 100%;
74 |     --sidebar-accent: 240 3.7% 15.9%;
75 |     --sidebar-accent-foreground: 240 4.8% 95.9%;
76 |     --sidebar-border: 240 3.7% 15.9%;
77 |     --sidebar-ring: 217.2 91.2% 59.8%;
78 |   }
79 | }
80 | 
81 | @layer base {
82 |   * {
83 |     @apply border-border;
84 |   }
85 |   body {
86 |     @apply bg-background text-foreground;
87 |   }
88 | }
89 | 


--------------------------------------------------------------------------------
/components/flow/question-node.tsx:
--------------------------------------------------------------------------------
 1 | import { memo } from 'react'
 2 | import { Handle, Position } from '@xyflow/react'
 3 | import { Card, CardContent } from '@/components/ui/card'
 4 | import { Button } from '@/components/ui/button'
 5 | import { Search, Loader2, AlertTriangle } from 'lucide-react'
 6 | import { SearchTermsNodeData } from '@/types'
 7 | 
 8 | export const QuestionNode = memo(function SearchTermsNode({
 9 |   data,
10 | }: {
11 |   data: SearchTermsNodeData
12 | }) {
13 |   const handleApprove = (term: string) => {
14 |     if (typeof data.onApprove !== 'function') {
15 |       console.error('No onApprove handler defined')
16 |       return
17 |     }
18 |     data.onApprove(term)
19 |   }
20 | 
21 |   const hasTerms =
22 |     Array.isArray(data.searchTerms) && data.searchTerms.length > 0
23 | 
24 |   return (
25 |     <div className='w-[400px]'>
26 |       <Card className='overflow-hidden'>
27 |         <Handle type='target' position={Position.Top} />
28 |         <CardContent className='p-4'>
29 |           {data.loading ? (
30 |             <div className='flex items-center gap-3'>
31 |               <Loader2 className='h-5 w-5 animate-spin text-blue-500' />
32 |               <p>Generating search terms...</p>
33 |             </div>
34 |           ) : data.error ? (
35 |             <div className='flex items-center gap-3 text-red-500'>
36 |               <AlertTriangle className='h-5 w-5' />
37 |               <p>{data.error}</p>
38 |             </div>
39 |           ) : hasTerms ? (
40 |             <div className='space-y-4'>
41 |               <div className='flex items-center gap-2'>
42 |                 <Search className='h-5 w-5 text-blue-500' />
43 |                 <h3 className='font-medium'>Follow-up Queries</h3>
44 |               </div>
45 |               <div className='space-y-2'>
46 |                 {data.searchTerms &&
47 |                   data.searchTerms.map((term, index) => (
48 |                     <div
49 |                       key={index}
50 |                       className='flex items-center justify-between gap-2 p-2 rounded bg-gray-50 hover:bg-gray-100 transition-colors'
51 |                     >
52 |                       <p className='text-sm text-gray-600 flex-1'>{term}</p>
53 |                       <Button
54 |                         size='sm'
55 |                         variant='ghost'
56 |                         onClick={() => handleApprove(term)}
57 |                         className='h-8 shrink-0'
58 |                         title='Search this term'
59 |                       >
60 |                         <Search className='h-4 w-4' />
61 |                       </Button>
62 |                     </div>
63 |                   ))}
64 |               </div>
65 |             </div>
66 |           ) : (
67 |             <div className='text-center text-gray-500 py-2'>
68 |               <Search className='h-5 w-5 mx-auto mb-2 opacity-40' />
69 |               <p>No suggestions available</p>
70 |             </div>
71 |           )}
72 |         </CardContent>
73 |         <Handle type='source' position={Position.Bottom} />
74 |       </Card>
75 |     </div>
76 |   )
77 | })
78 | 


--------------------------------------------------------------------------------
/app/api/generate-question/route.ts:
--------------------------------------------------------------------------------
 1 | import { NextResponse } from 'next/server'
 2 | import { CONFIG } from '@/lib/config'
 3 | import { generateWithModel } from '@/lib/models'
 4 | import { reportContentRatelimit } from '@/lib/redis'
 5 | export async function POST(request: Request) {
 6 |   try {
 7 |     const { report, platformModel } = await request.json()
 8 |     const platform = platformModel.split('__')[0]
 9 | 
10 |     if (CONFIG.rateLimits.enabled && platform !== 'ollama') {
11 |       const { success } = await reportContentRatelimit.limit(
12 |         'agentOptimizations'
13 |       )
14 |       if (!success) {
15 |         return NextResponse.json(
16 |           { error: 'Too many requests' },
17 |           { status: 429 }
18 |         )
19 |       }
20 |     }
21 | 
22 |     if (!report) {
23 |       return NextResponse.json({ error: 'Report is required' }, { status: 400 })
24 |     }
25 | 
26 |     const prompt = `Based on the following research report, generate 3 focused search terms or phrases for further research. These should be concise keywords or phrases that would help explore important aspects not fully covered in the current report.
27 | 
28 | Report Title: ${report.title}
29 | Summary: ${report.summary}
30 | 
31 | Key Sections:
32 | ${report.sections
33 |   ?.map(
34 |     (section: { title: string; content: string }) =>
35 |       `${section.title}: ${section.content}`
36 |   )
37 |   .join('\n')}
38 | 
39 | Generate exactly 3 search terms and return them in the following JSON format:
40 | {
41 |   "searchTerms": [
42 |     "first search term",
43 |     "second search term",
44 |     "third search term"
45 |   ]
46 | }
47 | 
48 | The search terms should be specific and focused on unexplored aspects of the topic.`
49 | 
50 |     try {
51 |       const response = await generateWithModel(prompt, platformModel)
52 | 
53 |       if (!response) {
54 |         throw new Error('No response from model')
55 |       }
56 | 
57 |       try {
58 |         // Parse the JSON response
59 |         const jsonResponse = JSON.parse(response)
60 |         if (
61 |           !Array.isArray(jsonResponse.searchTerms) ||
62 |           jsonResponse.searchTerms.length !== 3
63 |         ) {
64 |           throw new Error('Invalid search terms format')
65 |         }
66 | 
67 |         return NextResponse.json({ searchTerms: jsonResponse.searchTerms })
68 |       } catch (parseError) {
69 |         console.error('Failed to parse JSON response:', parseError)
70 |         // Fallback to line-based parsing if JSON parsing fails
71 |         const searchTerms = response
72 |           .split('\n')
73 |           .map((term) => term.trim())
74 |           .filter(
75 |             (term) =>
76 |               term.length > 0 && !term.includes('{') && !term.includes('}')
77 |           )
78 |           .slice(0, 3)
79 | 
80 |         return NextResponse.json({ searchTerms })
81 |       }
82 |     } catch (error) {
83 |       console.error('Model generation error:', error)
84 |       return NextResponse.json(
85 |         { error: 'Failed to generate search terms' },
86 |         { status: 500 }
87 |       )
88 |     }
89 |   } catch (error) {
90 |     console.error('Search terms generation error:', error)
91 |     return NextResponse.json(
92 |       { error: 'Failed to generate search terms' },
93 |       { status: 500 }
94 |     )
95 |   }
96 | }
97 | 


--------------------------------------------------------------------------------
/components/flow/report-node.tsx:
--------------------------------------------------------------------------------
 1 | import { memo } from 'react'
 2 | import { Handle, Position } from '@xyflow/react'
 3 | import { Card, CardContent } from '@/components/ui/card'
 4 | import { AlertTriangle, Loader2 } from 'lucide-react'
 5 | import type { ReportNodeData } from '@/types'
 6 | import ReactMarkdown from 'react-markdown'
 7 | import remarkGfm from 'remark-gfm'
 8 | import { ReportActions } from '@/components/report-actions'
 9 | import { Checkbox } from '@/components/ui/checkbox'
10 | import { CitationsFooter } from '@/components/citations-footer'
11 | 
12 | export const ReportNode = memo(function ReportNode({
13 |   id,
14 |   data,
15 | }: {
16 |   id: string
17 |   data: ReportNodeData
18 | }) {
19 |   const {
20 |     report,
21 |     loading,
22 |     error,
23 |     isSelected,
24 |     isConsolidated,
25 |     onSelect,
26 |     isConsolidating,
27 |   } = data
28 | 
29 |   return (
30 |     <div className='w-[600px]'>
31 |       <Handle type='target' position={Position.Top} />
32 |       <Card
33 |         className={`${isSelected ? 'ring-2 ring-blue-500' : ''} ${
34 |           isConsolidated ? 'border border-yellow-500' : ''
35 |         }`}
36 |       >
37 |         <CardContent className='p-6 space-y-4'>
38 |           {loading ? (
39 |             <div className='flex items-center justify-center p-4'>
40 |               <Loader2 className='h-6 w-6 animate-spin' />
41 |             </div>
42 |           ) : error ? (
43 |             <div className='flex items-center gap-2 text-red-500 text-center p-4'>
44 |               <AlertTriangle className='h-5 w-5' />
45 |               <span>{error}</span>
46 |             </div>
47 |           ) : report ? (
48 |             <div className='space-y-4'>
49 |               <div className='flex items-baseline gap-3'>
50 |                 {onSelect && (
51 |                   <Checkbox
52 |                     checked={isSelected}
53 |                     onCheckedChange={() => onSelect(id)}
54 |                     disabled={isConsolidating}
55 |                   />
56 |                 )}
57 |                 <h2 className='text-xl font-bold text-gray-800'>
58 |                   {report.title}
59 |                 </h2>
60 |               </div>
61 | 
62 |               <div className='flex justify-start'>
63 |                 <ReportActions report={report} size='sm' />
64 |               </div>
65 | 
66 |               <div className='max-h-[500px] overflow-y-auto pr-2 nowheel nodrag'>
67 |                 <p className='text-gray-700 mb-4'>{report.summary}</p>
68 | 
69 |                 {report.sections?.map((section, index) => (
70 |                   <div key={index} className='space-y-2 border-t pt-4 mb-4'>
71 |                     <h3 className='text-lg font-semibold text-gray-700'>
72 |                       {section.title}
73 |                     </h3>
74 |                     <div className='prose max-w-none text-gray-600'>
75 |                       <ReactMarkdown remarkPlugins={[remarkGfm]}>
76 |                         {section.content}
77 |                       </ReactMarkdown>
78 |                     </div>
79 |                   </div>
80 |                 ))}
81 | 
82 |                 {/* Citations Section */}
83 |                 <CitationsFooter report={report} />
84 |               </div>
85 |             </div>
86 |           ) : (
87 |             <div className='text-gray-500 text-center p-4'>
88 |               No report data available
89 |             </div>
90 |           )}
91 |         </CardContent>
92 |       </Card>
93 |       <Handle type='source' position={Position.Bottom} />
94 |     </div>
95 |   )
96 | })
97 | 


--------------------------------------------------------------------------------
/components/flow/search-node.tsx:
--------------------------------------------------------------------------------
  1 | import { memo, useState } from 'react'
  2 | import { Handle, Position } from '@xyflow/react'
  3 | import { Card, CardContent } from '@/components/ui/card'
  4 | import { Search, Loader2, Upload, AlertCircle } from 'lucide-react'
  5 | import { Button } from '@/components/ui/button'
  6 | import { Input } from '@/components/ui/input'
  7 | import { SUPPORTED_FILE_TYPES } from '@/lib/file-upload'
  8 | import { SearchNodeData } from '@/types'
  9 | 
 10 | export const SearchNode = memo(function SearchNode({
 11 |   data,
 12 | }: {
 13 |   data: SearchNodeData
 14 | }) {
 15 |   const [uploadError, setUploadError] = useState<string | null>(null)
 16 | 
 17 |   const handleFileUpload = (e: React.ChangeEvent<HTMLInputElement>) => {
 18 |     setUploadError(null)
 19 |     const file = e.target.files?.[0]
 20 | 
 21 |     if (!file) return
 22 | 
 23 |     if (!data.onFileUpload) {
 24 |       setUploadError('File upload is not available')
 25 |       return
 26 |     }
 27 | 
 28 |     try {
 29 |       data.onFileUpload(file)
 30 |       e.target.value = ''
 31 |     } catch (err) {
 32 |       setUploadError(
 33 |         `Upload failed: ${err instanceof Error ? err.message : 'Unknown error'}`
 34 |       )
 35 |     }
 36 |   }
 37 | 
 38 |   return (
 39 |     <div className='w-[400px]'>
 40 |       <Card>
 41 |         <CardContent className='p-4'>
 42 |           <div className='flex items-center justify-between gap-3'>
 43 |             {/* Icon and Query Section */}
 44 |             <div className='flex items-start gap-3 flex-1 min-w-0'>
 45 |               <div className='p-2 rounded-full bg-blue-50'>
 46 |                 {data.loading ? (
 47 |                   <Loader2 className='h-4 w-4 animate-spin text-blue-500' />
 48 |                 ) : (
 49 |                   <Search className='h-4 w-4 text-blue-500' />
 50 |                 )}
 51 |               </div>
 52 |               <div className='flex-1 min-w-0'>
 53 |                 <h3 className='font-medium text-sm text-gray-900'>
 54 |                   Search Query
 55 |                 </h3>
 56 |                 <p className='text-sm text-gray-600 truncate'>{data.query}</p>
 57 |               </div>
 58 |             </div>
 59 | 
 60 |             {/* Upload Button */}
 61 |             {data.onFileUpload && (
 62 |               <div className='relative flex-shrink-0'>
 63 |                 <Input
 64 |                   type='file'
 65 |                   onChange={handleFileUpload}
 66 |                   className='absolute inset-0 opacity-0 cursor-pointer'
 67 |                   accept={SUPPORTED_FILE_TYPES}
 68 |                 />
 69 |                 <Button
 70 |                   type='button'
 71 |                   variant='outline'
 72 |                   size='sm'
 73 |                   className='pointer-events-none'
 74 |                   disabled={data.loading}
 75 |                 >
 76 |                   <Upload className='h-4 w-4 mr-2' />
 77 |                   Upload
 78 |                 </Button>
 79 |               </div>
 80 |             )}
 81 |           </div>
 82 | 
 83 |           {uploadError && (
 84 |             <div className='mt-2 text-sm text-red-600 flex items-center gap-1'>
 85 |               <AlertCircle className='h-3 w-3' />
 86 |               <span>{uploadError}</span>
 87 |             </div>
 88 |           )}
 89 |         </CardContent>
 90 | 
 91 |         <Handle
 92 |           type='source'
 93 |           position={Position.Bottom}
 94 |           className='!bg-blue-500'
 95 |         />
 96 |       </Card>
 97 |     </div>
 98 |   )
 99 | })
100 | 


--------------------------------------------------------------------------------
/docs/flow-projects.md:
--------------------------------------------------------------------------------
 1 | # Flow Project Management
 2 | 
 3 | The Open Deep Research platform now includes a project management system that allows users to persist their research flows and organize them into separate projects.
 4 | 
 5 | ## Features
 6 | 
 7 | 1. **Project Persistence** - All research flows are automatically saved to the browser's localStorage, ensuring that research progress is not lost if the browser is closed.
 8 | 
 9 | 2. **Multiple Projects** - Users can create and switch between multiple research projects, each with its own set of nodes, edges, and research queries.
10 | 
11 | 3. **Project Management** - Create, rename, and delete projects as needed.
12 | 
13 | 4. **Import/Export** - Export projects to JSON files for backup and import them back when needed.
14 | 
15 | 5. **Storage Monitoring** - View localStorage usage to manage your saved projects effectively.
16 | 
17 | ## How It Works
18 | 
19 | - **Automatic Saving**: The flow state (nodes, edges, query) is automatically saved to localStorage as you work.
20 | - **Project Selector**: Use the dropdown menu in the top navigation bar to:
21 |   - Switch between existing projects
22 |   - Create new projects
23 |   - Rename or delete projects
24 | - **Data Management**: Use the database icon button to:
25 |   - Export projects as JSON files
26 |   - Import previously exported projects
27 |   - View storage usage information
28 | 
29 | ## Technical Implementation
30 | 
31 | The project management system is built on:
32 | 
33 | 1. `useFlowProjects` Hook:
34 | 
35 |    - Manages project state using localStorage
36 |    - Provides functions for creating, updating, and deleting projects
37 |    - Handles automatic saving of project state
38 |    - Monitors localStorage usage
39 | 
40 | 2. `ProjectSelector` Component:
41 | 
42 |    - Provides a user interface for project management
43 |    - Displays a list of available projects with timestamps
44 |    - Includes dialogs for creating and deleting projects
45 | 
46 | 3. `ProjectActions` Component:
47 | 
48 |    - Handles import and export functionality
49 |    - Shows storage usage information
50 |    - Provides warnings when storage is running low
51 | 
52 | 4. localStorage Keys:
53 |    - `open-deep-research-flow-projects`: Stores the array of all projects
54 |    - `open-deep-research-current-project`: Stores the ID of the currently active project
55 | 
56 | ## Data Structure
57 | 
58 | Each project contains:
59 | 
60 | ```typescript
61 | interface FlowProject {
62 |   id: string // Unique identifier
63 |   name: string // User-defined project name
64 |   createdAt: string // Creation timestamp
65 |   updatedAt: string // Last update timestamp
66 |   nodes: Node[] // ReactFlow nodes
67 |   edges: Edge[] // ReactFlow edges
68 |   query: string // Research query
69 | }
70 | ```
71 | 
72 | ## Limitations
73 | 
74 | - Project data is stored in the browser's localStorage, which has a size limit (typically 5MB)
75 | - Projects are not synced across devices or browsers
76 | - If localStorage is cleared, all projects will be lost (use export for backup)
77 | 
78 | ## Best Practices
79 | 
80 | 1. **Regular Exports**: Export your projects regularly to avoid data loss
81 | 2. **Monitor Storage**: Keep an eye on your storage usage to avoid hitting limits
82 | 3. **Project Organization**: Create separate projects for different research topics
83 | 4. **Clean Up**: Delete unnecessary projects to free up storage space
84 | 
85 | ## Future Improvements
86 | 
87 | - Cloud synchronization for projects across devices
88 | - More advanced project organization (folders, tags, etc.)
89 | - Version history and rollback capabilities
90 | - Selective import of specific projects
91 | 


--------------------------------------------------------------------------------
/lib/utils.ts:
--------------------------------------------------------------------------------
  1 | import { clsx, type ClassValue } from 'clsx'
  2 | import { twMerge } from 'tailwind-merge'
  3 | 
  4 | export function cn(...inputs: ClassValue[]) {
  5 |   return twMerge(clsx(inputs))
  6 | }
  7 | 
  8 | export function extractAndParseJSON(response: string) {
  9 |   function cleanJson(jsonStr: string): string {
 10 |     return (
 11 |       jsonStr
 12 |         // Remove YAML pipe characters
 13 |         .replace(/\|\n/g, '\n')
 14 |         // Remove YAML block scalar indicators (> and |) after colons
 15 |         .replace(/:\s*[>|](\s*\n|\s*$)/g, ': ')
 16 |         // Clean up any remaining YAML/Markdown artifacts
 17 |         .replace(/^\s*>/gm, '')
 18 |         // Remove trailing commas before closing braces/brackets
 19 |         .replace(/,(\s*[}\]])/g, '$1')
 20 |         // Normalize multiple newlines to single newlines
 21 |         .replace(/\n\s*\n/g, '\n')
 22 |         // Remove leading/trailing whitespace in multiline strings
 23 |         .replace(/:\s*"[\s\n]+/g, ': "')
 24 |         .replace(/[\s\n]+"/g, '"')
 25 |     )
 26 |   }
 27 | 
 28 |   // First attempt: Try to parse the entire response as JSON
 29 |   try {
 30 |     console.log(
 31 |       'Attempt 1 - Full parse, input:',
 32 |       response.slice(0, 100) + '...'
 33 |     )
 34 |     const result = JSON.parse(response)
 35 |     console.log('Attempt 1 succeeded')
 36 |     return result
 37 |   } catch (e) {
 38 |     console.log('Attempt 1 failed:', e)
 39 |   }
 40 | 
 41 |   // Second attempt: Look for JSON within code blocks
 42 |   const codeBlockRegex = /```(?:json)?\s*([\s\S]*?)```/
 43 |   const codeBlockMatch = response.match(codeBlockRegex)
 44 | 
 45 |   if (codeBlockMatch) {
 46 |     try {
 47 |       console.log(
 48 |         'Attempt 2 - Code block found, content:',
 49 |         codeBlockMatch[1].slice(0, 100) + '...'
 50 |       )
 51 |       const cleanedJson = cleanJson(codeBlockMatch[1])
 52 |       console.log(
 53 |         'Attempt 2 - Cleaned JSON:',
 54 |         cleanedJson.slice(0, 100) + '...'
 55 |       )
 56 |       const result = JSON.parse(cleanedJson)
 57 |       console.log('Attempt 2 succeeded')
 58 |       return result
 59 |     } catch (e) {
 60 |       console.log('Attempt 2 failed:', e)
 61 |     }
 62 |   } else {
 63 |     console.log('Attempt 2 - No code block found')
 64 |   }
 65 | 
 66 |   // Third attempt: Find the outermost matching braces
 67 |   console.log('Attempt 3 - Starting bracket matching')
 68 |   let bracketCount = 0
 69 |   let startIndex = -1
 70 |   let endIndex = -1
 71 |   let inString = false
 72 |   let escapeNext = false
 73 |   let foundStart = false
 74 | 
 75 |   for (let i = 0; i < response.length; i++) {
 76 |     // Handle string boundaries and escaped characters
 77 |     if (response[i] === '"' && !escapeNext) {
 78 |       inString = !inString
 79 |     } else if (response[i] === '\\' && !escapeNext) {
 80 |       escapeNext = true
 81 |       continue
 82 |     }
 83 | 
 84 |     escapeNext = false
 85 | 
 86 |     // Only count braces when not in a string
 87 |     if (!inString) {
 88 |       if (response[i] === '{') {
 89 |         if (bracketCount === 0) {
 90 |           startIndex = i
 91 |           foundStart = true
 92 |           console.log('Attempt 3 - Found opening brace at index:', i)
 93 |         }
 94 |         bracketCount++
 95 |       } else if (response[i] === '}') {
 96 |         bracketCount--
 97 |         if (bracketCount === 0 && foundStart) {
 98 |           endIndex = i + 1
 99 |           console.log('Attempt 3 - Found matching closing brace at index:', i)
100 |           // Try parsing this JSON substring with cleanup
101 |           try {
102 |             const jsonCandidate = cleanJson(
103 |               response.substring(startIndex, endIndex)
104 |             )
105 |             console.log(
106 |               'Attempt 3 - Trying to parse substring:',
107 |               jsonCandidate.slice(0, 100) + '...'
108 |             )
109 |             const result = JSON.parse(jsonCandidate)
110 |             console.log('Attempt 3 succeeded')
111 |             return result
112 |           } catch (e) {
113 |             console.log('Attempt 3 - Parse failed for this substring:', e)
114 |             foundStart = false // Reset to keep looking
115 |             continue
116 |           }
117 |         }
118 |       }
119 |     }
120 |   }
121 | 
122 |   console.log('All attempts failed - Final bracket count:', bracketCount)
123 |   throw new Error('No valid JSON found in response')
124 | }
125 | 


--------------------------------------------------------------------------------
/components/ui/dialog.tsx:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | import * as React from "react"
  4 | import * as DialogPrimitive from "@radix-ui/react-dialog"
  5 | import { X } from "lucide-react"
  6 | 
  7 | import { cn } from "@/lib/utils"
  8 | 
  9 | const Dialog = DialogPrimitive.Root
 10 | 
 11 | const DialogTrigger = DialogPrimitive.Trigger
 12 | 
 13 | const DialogPortal = DialogPrimitive.Portal
 14 | 
 15 | const DialogClose = DialogPrimitive.Close
 16 | 
 17 | const DialogOverlay = React.forwardRef<
 18 |   React.ElementRef<typeof DialogPrimitive.Overlay>,
 19 |   React.ComponentPropsWithoutRef<typeof DialogPrimitive.Overlay>
 20 | >(({ className, ...props }, ref) => (
 21 |   <DialogPrimitive.Overlay
 22 |     ref={ref}
 23 |     className={cn(
 24 |       "fixed inset-0 z-50 bg-black/80  data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0",
 25 |       className
 26 |     )}
 27 |     {...props}
 28 |   />
 29 | ))
 30 | DialogOverlay.displayName = DialogPrimitive.Overlay.displayName
 31 | 
 32 | const DialogContent = React.forwardRef<
 33 |   React.ElementRef<typeof DialogPrimitive.Content>,
 34 |   React.ComponentPropsWithoutRef<typeof DialogPrimitive.Content>
 35 | >(({ className, children, ...props }, ref) => (
 36 |   <DialogPortal>
 37 |     <DialogOverlay />
 38 |     <DialogPrimitive.Content
 39 |       ref={ref}
 40 |       className={cn(
 41 |         "fixed left-[50%] top-[50%] z-50 grid w-full max-w-lg translate-x-[-50%] translate-y-[-50%] gap-4 border bg-background p-6 shadow-lg duration-200 data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[state=closed]:slide-out-to-left-1/2 data-[state=closed]:slide-out-to-top-[48%] data-[state=open]:slide-in-from-left-1/2 data-[state=open]:slide-in-from-top-[48%] sm:rounded-lg",
 42 |         className
 43 |       )}
 44 |       {...props}
 45 |     >
 46 |       {children}
 47 |       <DialogPrimitive.Close className="absolute right-4 top-4 rounded-sm opacity-70 ring-offset-background transition-opacity hover:opacity-100 focus:outline-none focus:ring-2 focus:ring-ring focus:ring-offset-2 disabled:pointer-events-none data-[state=open]:bg-accent data-[state=open]:text-muted-foreground">
 48 |         <X className="h-4 w-4" />
 49 |         <span className="sr-only">Close</span>
 50 |       </DialogPrimitive.Close>
 51 |     </DialogPrimitive.Content>
 52 |   </DialogPortal>
 53 | ))
 54 | DialogContent.displayName = DialogPrimitive.Content.displayName
 55 | 
 56 | const DialogHeader = ({
 57 |   className,
 58 |   ...props
 59 | }: React.HTMLAttributes<HTMLDivElement>) => (
 60 |   <div
 61 |     className={cn(
 62 |       "flex flex-col space-y-1.5 text-center sm:text-left",
 63 |       className
 64 |     )}
 65 |     {...props}
 66 |   />
 67 | )
 68 | DialogHeader.displayName = "DialogHeader"
 69 | 
 70 | const DialogFooter = ({
 71 |   className,
 72 |   ...props
 73 | }: React.HTMLAttributes<HTMLDivElement>) => (
 74 |   <div
 75 |     className={cn(
 76 |       "flex flex-col-reverse sm:flex-row sm:justify-end sm:space-x-2",
 77 |       className
 78 |     )}
 79 |     {...props}
 80 |   />
 81 | )
 82 | DialogFooter.displayName = "DialogFooter"
 83 | 
 84 | const DialogTitle = React.forwardRef<
 85 |   React.ElementRef<typeof DialogPrimitive.Title>,
 86 |   React.ComponentPropsWithoutRef<typeof DialogPrimitive.Title>
 87 | >(({ className, ...props }, ref) => (
 88 |   <DialogPrimitive.Title
 89 |     ref={ref}
 90 |     className={cn(
 91 |       "text-lg font-semibold leading-none tracking-tight",
 92 |       className
 93 |     )}
 94 |     {...props}
 95 |   />
 96 | ))
 97 | DialogTitle.displayName = DialogPrimitive.Title.displayName
 98 | 
 99 | const DialogDescription = React.forwardRef<
100 |   React.ElementRef<typeof DialogPrimitive.Description>,
101 |   React.ComponentPropsWithoutRef<typeof DialogPrimitive.Description>
102 | >(({ className, ...props }, ref) => (
103 |   <DialogPrimitive.Description
104 |     ref={ref}
105 |     className={cn("text-sm text-muted-foreground", className)}
106 |     {...props}
107 |   />
108 | ))
109 | DialogDescription.displayName = DialogPrimitive.Description.displayName
110 | 
111 | export {
112 |   Dialog,
113 |   DialogPortal,
114 |   DialogOverlay,
115 |   DialogTrigger,
116 |   DialogClose,
117 |   DialogContent,
118 |   DialogHeader,
119 |   DialogFooter,
120 |   DialogTitle,
121 |   DialogDescription,
122 | }
123 | 


--------------------------------------------------------------------------------
/types/index.ts:
--------------------------------------------------------------------------------
  1 | export type Report = {
  2 |   title: string
  3 |   summary: string
  4 |   sections: {
  5 |     title: string
  6 |     content: string
  7 |   }[]
  8 |   sources: {
  9 |     id: string
 10 |     url: string
 11 |     name: string
 12 |   }[]
 13 |   usedSources?: number[] // Array of source indices that were actually used/cited
 14 | }
 15 | export interface Article {
 16 |   url: string
 17 |   title: string
 18 |   content: string
 19 | }
 20 | 
 21 | export type KnowledgeBaseReport = {
 22 |   id: string
 23 |   timestamp: number
 24 |   query: string
 25 |   report: Report
 26 | }
 27 | 
 28 | export type SearchResult = {
 29 |   id: string
 30 |   url: string
 31 |   name: string
 32 |   snippet: string
 33 |   isCustomUrl?: boolean
 34 |   score?: number
 35 |   content?: string
 36 | }
 37 | 
 38 | export type RankingResult = {
 39 |   url: string
 40 |   score: number
 41 |   reasoning: string
 42 | }
 43 | 
 44 | export type PlatformModel = {
 45 |   value: string
 46 |   label: string
 47 |   platform: string
 48 |   disabled: boolean
 49 | }
 50 | 
 51 | export type ModelVariant =
 52 |   | 'google__gemini-flash'
 53 |   | 'google__gemini-flash-thinking'
 54 |   | 'google__gemini-exp'
 55 |   | 'gpt-4o'
 56 |   | 'o1-mini'
 57 |   | 'o1'
 58 |   | 'claude-3-7-sonnet-latest'
 59 |   | 'claude-3-5-haiku-latest'
 60 |   | 'deepseek__chat'
 61 |   | 'deepseek__reasoner'
 62 |   | 'ollama__llama3.2'
 63 |   | 'ollama__deepseek-r1:14b'
 64 |   | 'openrouter__auto'
 65 | 
 66 | export type Status = {
 67 |   loading: boolean
 68 |   generatingReport: boolean
 69 |   agentStep: 'idle' | 'processing' | 'searching' | 'analyzing' | 'generating'
 70 |   fetchStatus: {
 71 |     total: number
 72 |     successful: number
 73 |     fallback: number
 74 |     sourceStatuses: Record<string, 'fetched' | 'preview'>
 75 |   }
 76 |   agentInsights: string[]
 77 |   searchQueries: string[]
 78 | }
 79 | 
 80 | export type State = {
 81 |   query: string
 82 |   timeFilter: string
 83 |   results: SearchResult[]
 84 |   selectedResults: string[]
 85 |   reportPrompt: string
 86 |   report: Report | null
 87 |   error: string | null
 88 |   newUrl: string
 89 |   isSourcesOpen: boolean
 90 |   selectedModel: string
 91 |   isAgentMode: boolean
 92 |   sidebarOpen: boolean
 93 |   activeTab: string
 94 |   status: Status
 95 | }
 96 | 
 97 | // Flow Component Types
 98 | export type BaseNodeData = {
 99 |   id?: string
100 |   loading?: boolean
101 |   error?: string
102 |   parentId?: string
103 |   childIds?: string[]
104 | }
105 | 
106 | export type SearchNodeData = BaseNodeData & {
107 |   query: string
108 |   onFileUpload?: (file: File) => void
109 | }
110 | 
111 | export type SelectionNodeData = BaseNodeData & {
112 |   results: SearchResult[]
113 |   onGenerateReport?: (
114 |     selectedResults: SearchResult[],
115 |     prompt: string
116 |   ) => Promise<
117 |     | { success: boolean; report: any; searchTerms: any; error?: undefined }
118 |     | {
119 |         success: boolean
120 |         error: string
121 |         report?: undefined
122 |         searchTerms?: undefined
123 |       }
124 |     | undefined
125 |   >
126 | }
127 | 
128 | export type ReportNodeData = BaseNodeData & {
129 |   report?: Report
130 |   isSelected?: boolean
131 |   onSelect?: (id: string) => void
132 |   isConsolidated?: boolean
133 |   isConsolidating?: boolean
134 | }
135 | 
136 | export type SearchTermsNodeData = BaseNodeData & {
137 |   searchTerms?: string[]
138 |   onApprove?: (term: string) => void
139 | }
140 | 
141 | // Combined interface for all node types with index signature for compatibility with xyflow
142 | export interface FlowNodeData extends BaseNodeData {
143 |   query?: string
144 |   results?: SearchResult[]
145 |   report?: Report
146 |   searchTerms?: string[]
147 |   question?: string
148 |   onGenerateReport?: (
149 |     selectedResults: SearchResult[],
150 |     prompt: string
151 |   ) => Promise<
152 |     | { success: boolean; report: any; searchTerms: any; error?: undefined }
153 |     | {
154 |         success: boolean
155 |         error: string
156 |         report?: undefined
157 |         searchTerms?: undefined
158 |       }
159 |     | undefined
160 |   >
161 |   onApprove?: (term?: string) => void
162 |   onConsolidate?: () => void
163 |   hasChildren?: boolean
164 |   isSelected?: boolean
165 |   onSelect?: (id: string) => void
166 |   isConsolidated?: boolean
167 |   isConsolidating?: boolean
168 |   onFileUpload?: (file: File) => void
169 |   [key: string]: any // This allows for dynamic properties required by xyflow
170 | }
171 | 
172 | // Configuration for different node types
173 | export interface NodeConfig {
174 |   zIndex: number
175 |   style?: React.CSSProperties
176 | }
177 | 


--------------------------------------------------------------------------------
/app/api/optimize-research/route.ts:
--------------------------------------------------------------------------------
  1 | import { NextResponse } from 'next/server'
  2 | import { reportContentRatelimit } from '@/lib/redis'
  3 | import { CONFIG } from '@/lib/config'
  4 | import { extractAndParseJSON } from '@/lib/utils'
  5 | import { generateWithModel } from '@/lib/models'
  6 | import { type ModelVariant } from '@/types'
  7 | 
  8 | export async function POST(request: Request) {
  9 |   try {
 10 |     const { prompt, platformModel = 'google__gemini-flash' } =
 11 |       (await request.json()) as {
 12 |         prompt: string
 13 |         platformModel: ModelVariant
 14 |       }
 15 | 
 16 |     if (!prompt) {
 17 |       return NextResponse.json({ error: 'Prompt is required' }, { status: 400 })
 18 |     }
 19 | 
 20 |     // Return test results for test queries
 21 |     if (prompt.toLowerCase() === 'test') {
 22 |       return NextResponse.json({
 23 |         query: 'test',
 24 |         optimizedPrompt:
 25 |           'Analyze and compare different research methodologies, focusing on scientific rigor, peer review processes, and validation techniques',
 26 |         explanation: 'Test optimization strategy',
 27 |         suggestedStructure: [
 28 |           'Test Structure 1',
 29 |           'Test Structure 2',
 30 |           'Test Structure 3',
 31 |         ],
 32 |       })
 33 |     }
 34 | 
 35 |     // Only check rate limit if enabled and not using Ollama (local model)
 36 |     const platform = platformModel.split('__')[0]
 37 |     const model = platformModel.split('__')[1]
 38 |     if (CONFIG.rateLimits.enabled && platform !== 'ollama') {
 39 |       const { success } = await reportContentRatelimit.limit(
 40 |         'agentOptimizations'
 41 |       )
 42 |       if (!success) {
 43 |         return NextResponse.json(
 44 |           { error: 'Too many requests' },
 45 |           { status: 429 }
 46 |         )
 47 |       }
 48 |     }
 49 | 
 50 |     // Check if selected platform is enabled
 51 |     const platformConfig =
 52 |       CONFIG.platforms[platform as keyof typeof CONFIG.platforms]
 53 |     if (!platformConfig?.enabled) {
 54 |       return NextResponse.json(
 55 |         { error: `${platform} platform is not enabled` },
 56 |         { status: 400 }
 57 |       )
 58 |     }
 59 | 
 60 |     // Check if selected model exists and is enabled
 61 |     const modelConfig = (platformConfig as any).models[model]
 62 |     if (!modelConfig) {
 63 |       return NextResponse.json(
 64 |         { error: `${model} model does not exist` },
 65 |         { status: 400 }
 66 |       )
 67 |     }
 68 |     if (!modelConfig.enabled) {
 69 |       return NextResponse.json(
 70 |         { error: `${model} model is disabled` },
 71 |         { status: 400 }
 72 |       )
 73 |     }
 74 | 
 75 |     const systemPrompt = `You are a research assistant tasked with optimizing a research topic into an effective search query.
 76 | 
 77 | Given this research topic: "${prompt}"
 78 | 
 79 | Your task is to:
 80 | 1. Generate ONE optimized search query that will help gather comprehensive information
 81 | 2. Create an optimized research prompt that will guide the final report generation
 82 | 3. Suggest a logical structure for organizing the research
 83 | 
 84 | The query should:
 85 | - Cover the core aspects of the topic
 86 | - Use relevant technical terms and synonyms
 87 | - Be specific enough to return high-quality results
 88 | - Be comprehensive yet concise
 89 | 
 90 | Format your response as a JSON object with this structure:
 91 | {
 92 |   "query": "the optimized search query",
 93 |   "optimizedPrompt": "The refined research prompt that will guide report generation",
 94 |   "explanation": "Brief explanation of the optimization strategy",
 95 |   "suggestedStructure": [
 96 |     "Key aspect 1 to cover",
 97 |     "Key aspect 2 to cover",
 98 |     "Key aspect 3 to cover"
 99 |   ]
100 | }
101 | 
102 | Make the query clear and focused, avoiding overly complex or lengthy constructions.`
103 | 
104 |     try {
105 |       const response = await generateWithModel(systemPrompt, platformModel)
106 | 
107 |       if (!response) {
108 |         throw new Error('No response from model')
109 |       }
110 | 
111 |       try {
112 |         const parsedResponse = extractAndParseJSON(response)
113 |         return NextResponse.json(parsedResponse)
114 |       } catch (parseError) {
115 |         console.error('Failed to parse optimization:', parseError)
116 |         return NextResponse.json(
117 |           { error: 'Failed to optimize research' },
118 |           { status: 500 }
119 |         )
120 |       }
121 |     } catch (error) {
122 |       console.error('Model generation error:', error)
123 |       return NextResponse.json(
124 |         { error: 'Failed to generate optimization' },
125 |         { status: 500 }
126 |       )
127 |     }
128 |   } catch (error) {
129 |     console.error('Research optimization failed:', error)
130 |     return NextResponse.json(
131 |       { error: 'Failed to optimize research' },
132 |       { status: 500 }
133 |     )
134 |   }
135 | }
136 | 


--------------------------------------------------------------------------------
/hooks/use-toast.ts:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | // Inspired by react-hot-toast library
  4 | import * as React from "react"
  5 | 
  6 | import type {
  7 |   ToastActionElement,
  8 |   ToastProps,
  9 | } from "@/components/ui/toast"
 10 | 
 11 | const TOAST_LIMIT = 1
 12 | const TOAST_REMOVE_DELAY = 1000000
 13 | 
 14 | type ToasterToast = ToastProps & {
 15 |   id: string
 16 |   title?: React.ReactNode
 17 |   description?: React.ReactNode
 18 |   action?: ToastActionElement
 19 | }
 20 | 
 21 | const actionTypes = {
 22 |   ADD_TOAST: "ADD_TOAST",
 23 |   UPDATE_TOAST: "UPDATE_TOAST",
 24 |   DISMISS_TOAST: "DISMISS_TOAST",
 25 |   REMOVE_TOAST: "REMOVE_TOAST",
 26 | } as const
 27 | 
 28 | let count = 0
 29 | 
 30 | function genId() {
 31 |   count = (count + 1) % Number.MAX_SAFE_INTEGER
 32 |   return count.toString()
 33 | }
 34 | 
 35 | type ActionType = typeof actionTypes
 36 | 
 37 | type Action =
 38 |   | {
 39 |       type: ActionType["ADD_TOAST"]
 40 |       toast: ToasterToast
 41 |     }
 42 |   | {
 43 |       type: ActionType["UPDATE_TOAST"]
 44 |       toast: Partial<ToasterToast>
 45 |     }
 46 |   | {
 47 |       type: ActionType["DISMISS_TOAST"]
 48 |       toastId?: ToasterToast["id"]
 49 |     }
 50 |   | {
 51 |       type: ActionType["REMOVE_TOAST"]
 52 |       toastId?: ToasterToast["id"]
 53 |     }
 54 | 
 55 | interface State {
 56 |   toasts: ToasterToast[]
 57 | }
 58 | 
 59 | const toastTimeouts = new Map<string, ReturnType<typeof setTimeout>>()
 60 | 
 61 | const addToRemoveQueue = (toastId: string) => {
 62 |   if (toastTimeouts.has(toastId)) {
 63 |     return
 64 |   }
 65 | 
 66 |   const timeout = setTimeout(() => {
 67 |     toastTimeouts.delete(toastId)
 68 |     dispatch({
 69 |       type: "REMOVE_TOAST",
 70 |       toastId: toastId,
 71 |     })
 72 |   }, TOAST_REMOVE_DELAY)
 73 | 
 74 |   toastTimeouts.set(toastId, timeout)
 75 | }
 76 | 
 77 | export const reducer = (state: State, action: Action): State => {
 78 |   switch (action.type) {
 79 |     case "ADD_TOAST":
 80 |       return {
 81 |         ...state,
 82 |         toasts: [action.toast, ...state.toasts].slice(0, TOAST_LIMIT),
 83 |       }
 84 | 
 85 |     case "UPDATE_TOAST":
 86 |       return {
 87 |         ...state,
 88 |         toasts: state.toasts.map((t) =>
 89 |           t.id === action.toast.id ? { ...t, ...action.toast } : t
 90 |         ),
 91 |       }
 92 | 
 93 |     case "DISMISS_TOAST": {
 94 |       const { toastId } = action
 95 | 
 96 |       // ! Side effects ! - This could be extracted into a dismissToast() action,
 97 |       // but I'll keep it here for simplicity
 98 |       if (toastId) {
 99 |         addToRemoveQueue(toastId)
100 |       } else {
101 |         state.toasts.forEach((toast) => {
102 |           addToRemoveQueue(toast.id)
103 |         })
104 |       }
105 | 
106 |       return {
107 |         ...state,
108 |         toasts: state.toasts.map((t) =>
109 |           t.id === toastId || toastId === undefined
110 |             ? {
111 |                 ...t,
112 |                 open: false,
113 |               }
114 |             : t
115 |         ),
116 |       }
117 |     }
118 |     case "REMOVE_TOAST":
119 |       if (action.toastId === undefined) {
120 |         return {
121 |           ...state,
122 |           toasts: [],
123 |         }
124 |       }
125 |       return {
126 |         ...state,
127 |         toasts: state.toasts.filter((t) => t.id !== action.toastId),
128 |       }
129 |   }
130 | }
131 | 
132 | const listeners: Array<(state: State) => void> = []
133 | 
134 | let memoryState: State = { toasts: [] }
135 | 
136 | function dispatch(action: Action) {
137 |   memoryState = reducer(memoryState, action)
138 |   listeners.forEach((listener) => {
139 |     listener(memoryState)
140 |   })
141 | }
142 | 
143 | type Toast = Omit<ToasterToast, "id">
144 | 
145 | function toast({ ...props }: Toast) {
146 |   const id = genId()
147 | 
148 |   const update = (props: ToasterToast) =>
149 |     dispatch({
150 |       type: "UPDATE_TOAST",
151 |       toast: { ...props, id },
152 |     })
153 |   const dismiss = () => dispatch({ type: "DISMISS_TOAST", toastId: id })
154 | 
155 |   dispatch({
156 |     type: "ADD_TOAST",
157 |     toast: {
158 |       ...props,
159 |       id,
160 |       open: true,
161 |       onOpenChange: (open) => {
162 |         if (!open) dismiss()
163 |       },
164 |     },
165 |   })
166 | 
167 |   return {
168 |     id: id,
169 |     dismiss,
170 |     update,
171 |   }
172 | }
173 | 
174 | function useToast() {
175 |   const [state, setState] = React.useState<State>(memoryState)
176 | 
177 |   React.useEffect(() => {
178 |     listeners.push(setState)
179 |     return () => {
180 |       const index = listeners.indexOf(setState)
181 |       if (index > -1) {
182 |         listeners.splice(index, 1)
183 |       }
184 |     }
185 |   }, [state])
186 | 
187 |   return {
188 |     ...state,
189 |     toast,
190 |     dismiss: (toastId?: string) => dispatch({ type: "DISMISS_TOAST", toastId }),
191 |   }
192 | }
193 | 
194 | export { useToast, toast }
195 | 


--------------------------------------------------------------------------------
/components/ui/sheet.tsx:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | import * as React from "react"
  4 | import * as SheetPrimitive from "@radix-ui/react-dialog"
  5 | import { cva, type VariantProps } from "class-variance-authority"
  6 | import { X } from "lucide-react"
  7 | 
  8 | import { cn } from "@/lib/utils"
  9 | 
 10 | const Sheet = SheetPrimitive.Root
 11 | 
 12 | const SheetTrigger = SheetPrimitive.Trigger
 13 | 
 14 | const SheetClose = SheetPrimitive.Close
 15 | 
 16 | const SheetPortal = SheetPrimitive.Portal
 17 | 
 18 | const SheetOverlay = React.forwardRef<
 19 |   React.ElementRef<typeof SheetPrimitive.Overlay>,
 20 |   React.ComponentPropsWithoutRef<typeof SheetPrimitive.Overlay>
 21 | >(({ className, ...props }, ref) => (
 22 |   <SheetPrimitive.Overlay
 23 |     className={cn(
 24 |       "fixed inset-0 z-50 bg-black/80  data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0",
 25 |       className
 26 |     )}
 27 |     {...props}
 28 |     ref={ref}
 29 |   />
 30 | ))
 31 | SheetOverlay.displayName = SheetPrimitive.Overlay.displayName
 32 | 
 33 | const sheetVariants = cva(
 34 |   "fixed z-50 gap-4 bg-background p-6 shadow-lg transition ease-in-out data-[state=closed]:duration-300 data-[state=open]:duration-500 data-[state=open]:animate-in data-[state=closed]:animate-out",
 35 |   {
 36 |     variants: {
 37 |       side: {
 38 |         top: "inset-x-0 top-0 border-b data-[state=closed]:slide-out-to-top data-[state=open]:slide-in-from-top",
 39 |         bottom:
 40 |           "inset-x-0 bottom-0 border-t data-[state=closed]:slide-out-to-bottom data-[state=open]:slide-in-from-bottom",
 41 |         left: "inset-y-0 left-0 h-full w-3/4 border-r data-[state=closed]:slide-out-to-left data-[state=open]:slide-in-from-left sm:max-w-sm",
 42 |         right:
 43 |           "inset-y-0 right-0 h-full w-3/4 border-l data-[state=closed]:slide-out-to-right data-[state=open]:slide-in-from-right sm:max-w-sm",
 44 |       },
 45 |     },
 46 |     defaultVariants: {
 47 |       side: "right",
 48 |     },
 49 |   }
 50 | )
 51 | 
 52 | interface SheetContentProps
 53 |   extends React.ComponentPropsWithoutRef<typeof SheetPrimitive.Content>,
 54 |     VariantProps<typeof sheetVariants> {}
 55 | 
 56 | const SheetContent = React.forwardRef<
 57 |   React.ElementRef<typeof SheetPrimitive.Content>,
 58 |   SheetContentProps
 59 | >(({ side = "right", className, children, ...props }, ref) => (
 60 |   <SheetPortal>
 61 |     <SheetOverlay />
 62 |     <SheetPrimitive.Content
 63 |       ref={ref}
 64 |       className={cn(sheetVariants({ side }), className)}
 65 |       {...props}
 66 |     >
 67 |       <SheetPrimitive.Close className="absolute right-4 top-4 rounded-sm opacity-70 ring-offset-background transition-opacity hover:opacity-100 focus:outline-none focus:ring-2 focus:ring-ring focus:ring-offset-2 disabled:pointer-events-none data-[state=open]:bg-secondary">
 68 |         <X className="h-4 w-4" />
 69 |         <span className="sr-only">Close</span>
 70 |       </SheetPrimitive.Close>
 71 |       {children}
 72 |     </SheetPrimitive.Content>
 73 |   </SheetPortal>
 74 | ))
 75 | SheetContent.displayName = SheetPrimitive.Content.displayName
 76 | 
 77 | const SheetHeader = ({
 78 |   className,
 79 |   ...props
 80 | }: React.HTMLAttributes<HTMLDivElement>) => (
 81 |   <div
 82 |     className={cn(
 83 |       "flex flex-col space-y-2 text-center sm:text-left",
 84 |       className
 85 |     )}
 86 |     {...props}
 87 |   />
 88 | )
 89 | SheetHeader.displayName = "SheetHeader"
 90 | 
 91 | const SheetFooter = ({
 92 |   className,
 93 |   ...props
 94 | }: React.HTMLAttributes<HTMLDivElement>) => (
 95 |   <div
 96 |     className={cn(
 97 |       "flex flex-col-reverse sm:flex-row sm:justify-end sm:space-x-2",
 98 |       className
 99 |     )}
100 |     {...props}
101 |   />
102 | )
103 | SheetFooter.displayName = "SheetFooter"
104 | 
105 | const SheetTitle = React.forwardRef<
106 |   React.ElementRef<typeof SheetPrimitive.Title>,
107 |   React.ComponentPropsWithoutRef<typeof SheetPrimitive.Title>
108 | >(({ className, ...props }, ref) => (
109 |   <SheetPrimitive.Title
110 |     ref={ref}
111 |     className={cn("text-lg font-semibold text-foreground", className)}
112 |     {...props}
113 |   />
114 | ))
115 | SheetTitle.displayName = SheetPrimitive.Title.displayName
116 | 
117 | const SheetDescription = React.forwardRef<
118 |   React.ElementRef<typeof SheetPrimitive.Description>,
119 |   React.ComponentPropsWithoutRef<typeof SheetPrimitive.Description>
120 | >(({ className, ...props }, ref) => (
121 |   <SheetPrimitive.Description
122 |     ref={ref}
123 |     className={cn("text-sm text-muted-foreground", className)}
124 |     {...props}
125 |   />
126 | ))
127 | SheetDescription.displayName = SheetPrimitive.Description.displayName
128 | 
129 | export {
130 |   Sheet,
131 |   SheetPortal,
132 |   SheetOverlay,
133 |   SheetTrigger,
134 |   SheetClose,
135 |   SheetContent,
136 |   SheetHeader,
137 |   SheetFooter,
138 |   SheetTitle,
139 |   SheetDescription,
140 | }
141 | 


--------------------------------------------------------------------------------
/components/ui/toast.tsx:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | import * as React from "react"
  4 | import * as ToastPrimitives from "@radix-ui/react-toast"
  5 | import { cva, type VariantProps } from "class-variance-authority"
  6 | import { X } from "lucide-react"
  7 | 
  8 | import { cn } from "@/lib/utils"
  9 | 
 10 | const ToastProvider = ToastPrimitives.Provider
 11 | 
 12 | const ToastViewport = React.forwardRef<
 13 |   React.ElementRef<typeof ToastPrimitives.Viewport>,
 14 |   React.ComponentPropsWithoutRef<typeof ToastPrimitives.Viewport>
 15 | >(({ className, ...props }, ref) => (
 16 |   <ToastPrimitives.Viewport
 17 |     ref={ref}
 18 |     className={cn(
 19 |       "fixed top-0 z-[100] flex max-h-screen w-full flex-col-reverse p-4 sm:bottom-0 sm:right-0 sm:top-auto sm:flex-col md:max-w-[420px]",
 20 |       className
 21 |     )}
 22 |     {...props}
 23 |   />
 24 | ))
 25 | ToastViewport.displayName = ToastPrimitives.Viewport.displayName
 26 | 
 27 | const toastVariants = cva(
 28 |   "group pointer-events-auto relative flex w-full items-center justify-between space-x-2 overflow-hidden rounded-md border p-4 pr-6 shadow-lg transition-all data-[swipe=cancel]:translate-x-0 data-[swipe=end]:translate-x-[var(--radix-toast-swipe-end-x)] data-[swipe=move]:translate-x-[var(--radix-toast-swipe-move-x)] data-[swipe=move]:transition-none data-[state=open]:animate-in data-[state=closed]:animate-out data-[swipe=end]:animate-out data-[state=closed]:fade-out-80 data-[state=closed]:slide-out-to-right-full data-[state=open]:slide-in-from-top-full data-[state=open]:sm:slide-in-from-bottom-full",
 29 |   {
 30 |     variants: {
 31 |       variant: {
 32 |         default: "border bg-background text-foreground",
 33 |         destructive:
 34 |           "destructive group border-destructive bg-destructive text-destructive-foreground",
 35 |       },
 36 |     },
 37 |     defaultVariants: {
 38 |       variant: "default",
 39 |     },
 40 |   }
 41 | )
 42 | 
 43 | const Toast = React.forwardRef<
 44 |   React.ElementRef<typeof ToastPrimitives.Root>,
 45 |   React.ComponentPropsWithoutRef<typeof ToastPrimitives.Root> &
 46 |     VariantProps<typeof toastVariants>
 47 | >(({ className, variant, ...props }, ref) => {
 48 |   return (
 49 |     <ToastPrimitives.Root
 50 |       ref={ref}
 51 |       className={cn(toastVariants({ variant }), className)}
 52 |       {...props}
 53 |     />
 54 |   )
 55 | })
 56 | Toast.displayName = ToastPrimitives.Root.displayName
 57 | 
 58 | const ToastAction = React.forwardRef<
 59 |   React.ElementRef<typeof ToastPrimitives.Action>,
 60 |   React.ComponentPropsWithoutRef<typeof ToastPrimitives.Action>
 61 | >(({ className, ...props }, ref) => (
 62 |   <ToastPrimitives.Action
 63 |     ref={ref}
 64 |     className={cn(
 65 |       "inline-flex h-8 shrink-0 items-center justify-center rounded-md border bg-transparent px-3 text-sm font-medium transition-colors hover:bg-secondary focus:outline-none focus:ring-1 focus:ring-ring disabled:pointer-events-none disabled:opacity-50 group-[.destructive]:border-muted/40 group-[.destructive]:hover:border-destructive/30 group-[.destructive]:hover:bg-destructive group-[.destructive]:hover:text-destructive-foreground group-[.destructive]:focus:ring-destructive",
 66 |       className
 67 |     )}
 68 |     {...props}
 69 |   />
 70 | ))
 71 | ToastAction.displayName = ToastPrimitives.Action.displayName
 72 | 
 73 | const ToastClose = React.forwardRef<
 74 |   React.ElementRef<typeof ToastPrimitives.Close>,
 75 |   React.ComponentPropsWithoutRef<typeof ToastPrimitives.Close>
 76 | >(({ className, ...props }, ref) => (
 77 |   <ToastPrimitives.Close
 78 |     ref={ref}
 79 |     className={cn(
 80 |       "absolute right-1 top-1 rounded-md p-1 text-foreground/50 opacity-0 transition-opacity hover:text-foreground focus:opacity-100 focus:outline-none focus:ring-1 group-hover:opacity-100 group-[.destructive]:text-red-300 group-[.destructive]:hover:text-red-50 group-[.destructive]:focus:ring-red-400 group-[.destructive]:focus:ring-offset-red-600",
 81 |       className
 82 |     )}
 83 |     toast-close=""
 84 |     {...props}
 85 |   >
 86 |     <X className="h-4 w-4" />
 87 |   </ToastPrimitives.Close>
 88 | ))
 89 | ToastClose.displayName = ToastPrimitives.Close.displayName
 90 | 
 91 | const ToastTitle = React.forwardRef<
 92 |   React.ElementRef<typeof ToastPrimitives.Title>,
 93 |   React.ComponentPropsWithoutRef<typeof ToastPrimitives.Title>
 94 | >(({ className, ...props }, ref) => (
 95 |   <ToastPrimitives.Title
 96 |     ref={ref}
 97 |     className={cn("text-sm font-semibold [&+div]:text-xs", className)}
 98 |     {...props}
 99 |   />
100 | ))
101 | ToastTitle.displayName = ToastPrimitives.Title.displayName
102 | 
103 | const ToastDescription = React.forwardRef<
104 |   React.ElementRef<typeof ToastPrimitives.Description>,
105 |   React.ComponentPropsWithoutRef<typeof ToastPrimitives.Description>
106 | >(({ className, ...props }, ref) => (
107 |   <ToastPrimitives.Description
108 |     ref={ref}
109 |     className={cn("text-sm opacity-90", className)}
110 |     {...props}
111 |   />
112 | ))
113 | ToastDescription.displayName = ToastPrimitives.Description.displayName
114 | 
115 | type ToastProps = React.ComponentPropsWithoutRef<typeof Toast>
116 | 
117 | type ToastActionElement = React.ReactElement<typeof ToastAction>
118 | 
119 | export {
120 |   type ToastProps,
121 |   type ToastActionElement,
122 |   ToastProvider,
123 |   ToastViewport,
124 |   Toast,
125 |   ToastTitle,
126 |   ToastDescription,
127 |   ToastClose,
128 |   ToastAction,
129 | }
130 | 


--------------------------------------------------------------------------------
/app/api/analyze-results/route.ts:
--------------------------------------------------------------------------------
  1 | import { NextResponse } from 'next/server'
  2 | import { reportContentRatelimit } from '@/lib/redis'
  3 | import { CONFIG } from '@/lib/config'
  4 | import { extractAndParseJSON } from '@/lib/utils'
  5 | import { generateWithModel } from '@/lib/models'
  6 | import { type ModelVariant } from '@/types'
  7 | 
  8 | type SearchResultInput = {
  9 |   title: string
 10 |   snippet: string
 11 |   url: string
 12 |   content?: string
 13 | }
 14 | 
 15 | export async function POST(request: Request) {
 16 |   try {
 17 |     const {
 18 |       prompt,
 19 |       results,
 20 |       isTestQuery = false,
 21 |       platformModel = 'google__gemini-flash',
 22 |     } = (await request.json()) as {
 23 |       prompt: string
 24 |       results: SearchResultInput[]
 25 |       isTestQuery?: boolean
 26 |       platformModel: ModelVariant
 27 |     }
 28 | 
 29 |     if (!prompt || !results?.length) {
 30 |       return NextResponse.json(
 31 |         { error: 'Prompt and results are required' },
 32 |         { status: 400 }
 33 |       )
 34 |     }
 35 | 
 36 |     // Return test results for test queries
 37 |     if (
 38 |       isTestQuery ||
 39 |       results.some((r) => r.url.includes('example.com/test'))
 40 |     ) {
 41 |       return NextResponse.json({
 42 |         rankings: results.map((result, index) => ({
 43 |           url: result.url,
 44 |           score: index === 0 ? 1 : 0.5, // Give first result highest score
 45 |           reasoning: 'Test ranking result',
 46 |         })),
 47 |         analysis: 'Test analysis of search results',
 48 |       })
 49 |     }
 50 | 
 51 |     // Only check rate limit if enabled and not using Ollama (local model)
 52 |     const platform = platformModel.split('__')[0]
 53 |     const model = platformModel.split('__')[1]
 54 |     if (CONFIG.rateLimits.enabled && platform !== 'ollama') {
 55 |       const { success } = await reportContentRatelimit.limit(
 56 |         'agentOptimizations'
 57 |       )
 58 |       if (!success) {
 59 |         return NextResponse.json(
 60 |           { error: 'Too many requests' },
 61 |           { status: 429 }
 62 |         )
 63 |       }
 64 |     }
 65 | 
 66 |     // Check if selected platform is enabled
 67 |     const platformConfig =
 68 |       CONFIG.platforms[platform as keyof typeof CONFIG.platforms]
 69 |     if (!platformConfig?.enabled) {
 70 |       return NextResponse.json(
 71 |         { error: `${platform} platform is not enabled` },
 72 |         { status: 400 }
 73 |       )
 74 |     }
 75 | 
 76 |     // Check if selected model exists and is enabled
 77 |     const modelConfig = (platformConfig as any).models[model]
 78 |     if (!modelConfig) {
 79 |       return NextResponse.json(
 80 |         { error: `${model} model does not exist` },
 81 |         { status: 400 }
 82 |       )
 83 |     }
 84 |     if (!modelConfig.enabled) {
 85 |       return NextResponse.json(
 86 |         { error: `${model} model is disabled` },
 87 |         { status: 400 }
 88 |       )
 89 |     }
 90 | 
 91 |     const systemPrompt = `You are a research assistant tasked with analyzing search results for relevance to a research topic.
 92 | 
 93 | Research Topic: "${prompt}"
 94 | 
 95 | Analyze these search results and score them based on:
 96 | 1. Relevance to the research topic
 97 | 2. Information quality and depth
 98 | 3. Source credibility
 99 | 4. Uniqueness of perspective
100 | 
101 | For each result, assign a score from 0 to 1, where:
102 | - 1.0: Highly relevant, authoritative, and comprehensive
103 | - 0.7-0.9: Very relevant with good information
104 | - 0.4-0.6: Moderately relevant or basic information
105 | - 0.1-0.3: Tangentially relevant
106 | - 0.0: Not relevant or unreliable
107 | 
108 | Here are the results to analyze:
109 | 
110 | ${results
111 |   .map(
112 |     (result, index) => `
113 | Result ${index + 1}:
114 | Title: ${result.title}
115 | URL: ${result.url}
116 | Snippet: ${result.snippet}
117 | ${result.content ? `Full Content: ${result.content}` : ''}
118 | ---`
119 |   )
120 |   .join('\n')}
121 | 
122 | Format your response as a JSON object with this structure:
123 | {
124 |   "rankings": [
125 |     {
126 |       "url": "result url",
127 |       "score": 0.85,
128 |       "reasoning": "Brief explanation of the score"
129 |     }
130 |   ],
131 |   "analysis": "Brief overall analysis of the result set"
132 | }
133 | 
134 | Focus on finding results that provide unique, high-quality information relevant to the research topic.`
135 | 
136 |     try {
137 |       const response = await generateWithModel(systemPrompt, platformModel)
138 | 
139 |       if (!response) {
140 |         throw new Error('No response from model')
141 |       }
142 | 
143 |       try {
144 |         const parsedResponse = extractAndParseJSON(response)
145 |         return NextResponse.json(parsedResponse)
146 |       } catch (parseError) {
147 |         console.error('Failed to parse analysis:', parseError)
148 |         return NextResponse.json(
149 |           { error: 'Failed to analyze results' },
150 |           { status: 500 }
151 |         )
152 |       }
153 |     } catch (error) {
154 |       console.error('Model generation error:', error)
155 |       return NextResponse.json(
156 |         { error: 'Failed to generate analysis' },
157 |         { status: 500 }
158 |       )
159 |     }
160 |   } catch (error) {
161 |     console.error('Result analysis failed:', error)
162 |     return NextResponse.json(
163 |       { error: 'Failed to analyze results' },
164 |       { status: 500 }
165 |     )
166 |   }
167 | }
168 | 


--------------------------------------------------------------------------------
/components/report-actions.tsx:
--------------------------------------------------------------------------------
  1 | import { Button } from '@/components/ui/button'
  2 | import { Brain, Download, Copy } from 'lucide-react'
  3 | import {
  4 |   DropdownMenu,
  5 |   DropdownMenuContent,
  6 |   DropdownMenuItem,
  7 |   DropdownMenuTrigger,
  8 | } from '@/components/ui/dropdown-menu'
  9 | import { useToast } from '@/hooks/use-toast'
 10 | import { useKnowledgeBase } from '@/hooks/use-knowledge-base'
 11 | import type { Report } from '@/types'
 12 | 
 13 | interface ReportActionsProps {
 14 |   report: Report
 15 |   prompt?: string
 16 |   size?: 'default' | 'sm'
 17 |   variant?: 'default' | 'outline'
 18 |   className?: string
 19 |   hideKnowledgeBase?: boolean
 20 | }
 21 | 
 22 | export function ReportActions({
 23 |   report,
 24 |   prompt,
 25 |   size = 'sm',
 26 |   variant = 'outline',
 27 |   className = '',
 28 |   hideKnowledgeBase = false,
 29 | }: ReportActionsProps) {
 30 |   const { addReport, reports } = useKnowledgeBase()
 31 |   const { toast } = useToast()
 32 | 
 33 |   // Check if report is already saved by comparing title and summary
 34 |   const isReportSaved = reports.some(
 35 |     (savedReport) =>
 36 |       savedReport.report.title === report.title &&
 37 |       savedReport.report.summary === report.summary
 38 |   )
 39 | 
 40 |   const handleDownload = async (format: 'pdf' | 'docx' | 'txt') => {
 41 |     try {
 42 |       const response = await fetch('/api/download', {
 43 |         method: 'POST',
 44 |         headers: { 'Content-Type': 'application/json' },
 45 |         body: JSON.stringify({
 46 |           report,
 47 |           format,
 48 |         }),
 49 |       })
 50 | 
 51 |       const blob = await response.blob()
 52 |       const url = window.URL.createObjectURL(blob)
 53 |       const a = document.createElement('a')
 54 |       a.href = url
 55 |       a.download = `report.${format}`
 56 |       document.body.appendChild(a)
 57 |       a.click()
 58 |       window.URL.revokeObjectURL(url)
 59 |       document.body.removeChild(a)
 60 |     } catch (error) {
 61 |       toast({
 62 |         title: 'Download failed',
 63 |         description: error instanceof Error ? error.message : 'Download failed',
 64 |         variant: 'destructive',
 65 |       })
 66 |     }
 67 |   }
 68 | 
 69 |   const handleSaveToKnowledgeBase = () => {
 70 |     if (isReportSaved) return
 71 |     const success = addReport(report, prompt || '')
 72 |     if (success) {
 73 |       toast({
 74 |         title: 'Saved to Knowledge Base',
 75 |         description: 'The report has been saved for future reference',
 76 |       })
 77 |     }
 78 |   }
 79 | 
 80 |   const handleCopy = async () => {
 81 |     try {
 82 |       let formattedContent = `${report.title}\n\n${
 83 |         report.summary
 84 |       }\n\n${report.sections
 85 |         .map((section) => `${section.title}\n${section.content}`)
 86 |         .join('\n\n')}`
 87 | 
 88 |       // Filter sources if usedSources is available
 89 |       const filteredSources =
 90 |         report.usedSources && report.usedSources.length > 0 && report.sources
 91 |           ? report.sources.filter((_, index) =>
 92 |               report.usedSources!.map((num) => num - 1).includes(index)
 93 |             )
 94 |           : report.sources
 95 | 
 96 |       // Add citations if filtered sources are available
 97 |       if (filteredSources && filteredSources.length > 0) {
 98 |         formattedContent +=
 99 |           '\n\nReferences:\n' +
100 |           filteredSources
101 |             .map(
102 |               (source, index) => `${index + 1}. ${source.name} - ${source.url}`
103 |             )
104 |             .join('\n')
105 |       }
106 | 
107 |       await navigator.clipboard.writeText(formattedContent)
108 |       toast({
109 |         title: 'Copied',
110 |         description: 'Report content copied to clipboard',
111 |       })
112 |     } catch (error) {
113 |       toast({
114 |         title: 'Copy failed',
115 |         description:
116 |           error instanceof Error ? error.message : 'Failed to copy content',
117 |         variant: 'destructive',
118 |       })
119 |     }
120 |   }
121 | 
122 |   return (
123 |     <div className={`flex gap-2 ${className}`}>
124 |       <Button
125 |         variant={variant}
126 |         size={size}
127 |         className='gap-2'
128 |         onClick={handleCopy}
129 |         title='Copy report'
130 |       >
131 |         <Copy className='h-4 w-4' />
132 |         <span className='hidden sm:inline'>Copy</span>
133 |       </Button>
134 |       {!hideKnowledgeBase && (
135 |         <Button
136 |           variant={variant}
137 |           size={size}
138 |           className='gap-2'
139 |           onClick={handleSaveToKnowledgeBase}
140 |           disabled={isReportSaved}
141 |           title={isReportSaved ? 'Already saved' : 'Save to Knowledge Base'}
142 |         >
143 |           <Brain className='h-4 w-4' />
144 |           <span className='hidden sm:inline'>
145 |             {isReportSaved ? 'Saved' : 'Save to Knowledge Base'}
146 |           </span>
147 |         </Button>
148 |       )}
149 |       <DropdownMenu>
150 |         <DropdownMenuTrigger asChild>
151 |           <Button
152 |             variant={variant}
153 |             size={size}
154 |             className='gap-2'
155 |             title='Download'
156 |           >
157 |             <Download className='h-4 w-4' />
158 |             <span className='hidden sm:inline'>Download</span>
159 |           </Button>
160 |         </DropdownMenuTrigger>
161 |         <DropdownMenuContent align='end'>
162 |           <DropdownMenuItem onClick={() => handleDownload('pdf')}>
163 |             Download as PDF
164 |           </DropdownMenuItem>
165 |           <DropdownMenuItem onClick={() => handleDownload('docx')}>
166 |             Download as Word
167 |           </DropdownMenuItem>
168 |           <DropdownMenuItem onClick={() => handleDownload('txt')}>
169 |             Download as Text
170 |           </DropdownMenuItem>
171 |         </DropdownMenuContent>
172 |       </DropdownMenu>
173 |     </div>
174 |   )
175 | }
176 | 


--------------------------------------------------------------------------------
/app/report/[id]/report-content.tsx:
--------------------------------------------------------------------------------
  1 | 'use client'
  2 | 
  3 | import React, { useEffect, useState } from 'react'
  4 | import { useRouter } from 'next/navigation'
  5 | import { Button } from '@/components/ui/button'
  6 | import { Card } from '@/components/ui/card'
  7 | import { ArrowLeft, Trash2, AlertTriangle, Brain } from 'lucide-react'
  8 | import { type KnowledgeBaseReport } from '@/types'
  9 | import ReactMarkdown from 'react-markdown'
 10 | import remarkGfm from 'remark-gfm'
 11 | import { formatDistanceToNow } from 'date-fns'
 12 | import { useKnowledgeBase } from '@/hooks/use-knowledge-base'
 13 | import { Alert, AlertDescription, AlertTitle } from '@/components/ui/alert'
 14 | import { ReportActions } from '@/components/report-actions'
 15 | import { CitationsFooter } from '@/components/citations-footer'
 16 | 
 17 | export function ReportContent({ id }: any) {
 18 |   const router = useRouter()
 19 |   const { reports, deleteReport } = useKnowledgeBase()
 20 |   const [report, setReport] = useState<KnowledgeBaseReport | null>(null)
 21 |   const [showDeleteConfirm, setShowDeleteConfirm] = useState(false)
 22 | 
 23 |   useEffect(() => {
 24 |     const foundReport = reports.find((r) => r.id === id)
 25 |     if (foundReport) {
 26 |       setReport(foundReport)
 27 |     }
 28 |   }, [id, reports])
 29 | 
 30 |   const handleDelete = () => {
 31 |     if (!report) return
 32 |     deleteReport(report.id)
 33 |     router.push('/')
 34 |   }
 35 | 
 36 |   if (!report) {
 37 |     return (
 38 |       <div className='min-h-screen bg-white p-4 sm:p-8'>
 39 |         <div className='max-w-4xl mx-auto'>
 40 |           <Alert variant='destructive'>
 41 |             <AlertTriangle className='h-4 w-4' />
 42 |             <AlertTitle>Report Not Found</AlertTitle>
 43 |             <AlertDescription>
 44 |               The report you&apos;re looking for doesn&apos;t exist or has been
 45 |               deleted.
 46 |             </AlertDescription>
 47 |           </Alert>
 48 |           <div className='mt-4 text-center'>
 49 |             <Button
 50 |               variant='ghost'
 51 |               onClick={() => router.push('/')}
 52 |               className='gap-2'
 53 |             >
 54 |               <ArrowLeft className='h-4 w-4' />
 55 |               Back to Home
 56 |             </Button>
 57 |           </div>
 58 |         </div>
 59 |       </div>
 60 |     )
 61 |   }
 62 | 
 63 |   return (
 64 |     <div className='min-h-screen bg-white p-4 sm:p-8'>
 65 |       <div className='max-w-4xl mx-auto'>
 66 |         <div className='mb-8 flex items-center justify-between'>
 67 |           <Button
 68 |             variant='ghost'
 69 |             onClick={() => router.back()}
 70 |             className='gap-2'
 71 |           >
 72 |             <ArrowLeft className='h-4 w-4' />
 73 |             Back
 74 |           </Button>
 75 |           <div className='flex gap-2'>
 76 |             <ReportActions
 77 |               report={report?.report}
 78 |               prompt={report?.query}
 79 |               size='sm'
 80 |               hideKnowledgeBase={true}
 81 |             />
 82 |             <Button
 83 |               variant='destructive'
 84 |               size='sm'
 85 |               onClick={() => setShowDeleteConfirm(true)}
 86 |               className='gap-2'
 87 |             >
 88 |               <Trash2 className='h-4 w-4' />
 89 |               Delete Report
 90 |             </Button>
 91 |           </div>
 92 |         </div>
 93 | 
 94 |         {showDeleteConfirm && (
 95 |           <Alert variant='destructive' className='mb-6'>
 96 |             <AlertTriangle className='h-4 w-4' />
 97 |             <AlertTitle>Delete Report?</AlertTitle>
 98 |             <AlertDescription className='space-y-2'>
 99 |               <p>
100 |                 This will permanently delete this report from your knowledge
101 |                 base.
102 |               </p>
103 |               <div className='flex gap-2 mt-2'>
104 |                 <Button
105 |                   variant='outline'
106 |                   size='sm'
107 |                   onClick={() => setShowDeleteConfirm(false)}
108 |                 >
109 |                   Cancel
110 |                 </Button>
111 |                 <Button variant='destructive' size='sm' onClick={handleDelete}>
112 |                   Yes, Delete
113 |                 </Button>
114 |               </div>
115 |             </AlertDescription>
116 |           </Alert>
117 |         )}
118 | 
119 |         <Alert className='mb-6'>
120 |           <Brain className='h-4 w-4' />
121 |           <AlertTitle>Knowledge Base Report</AlertTitle>
122 |           <AlertDescription>
123 |             This report was saved{' '}
124 |             {formatDistanceToNow(report.timestamp, { addSuffix: true })}
125 |             in response to the query: &apos;{report.query}&apos;
126 |           </AlertDescription>
127 |         </Alert>
128 | 
129 |         <Card className='p-6'>
130 |           <div className='mb-6'>
131 |             <h1 className='text-3xl font-bold text-gray-800 mb-2'>
132 |               {report.report.title}
133 |             </h1>
134 |           </div>
135 | 
136 |           <div className='prose max-w-none'>
137 |             <h2>Summary</h2>
138 | 
139 |             {/* Scrollable content area */}
140 |             <div
141 |               className='max-h-[700px] overflow-y-auto pr-2'
142 |               style={{ scrollbarWidth: 'thin' }}
143 |             >
144 |               <p className='mb-6'>{report.report.summary}</p>
145 | 
146 |               {report.report.sections.map((section, index) => (
147 |                 <div key={index} className='mb-6'>
148 |                   <h2>{section.title}</h2>
149 |                   <ReactMarkdown remarkPlugins={[remarkGfm]}>
150 |                     {section.content}
151 |                   </ReactMarkdown>
152 |                 </div>
153 |               ))}
154 | 
155 |               {/* Citations Section */}
156 |               <CitationsFooter report={report.report} />
157 |             </div>
158 |           </div>
159 |         </Card>
160 |       </div>
161 |     </div>
162 |   )
163 | }
164 | 


--------------------------------------------------------------------------------
/lib/models.ts:
--------------------------------------------------------------------------------
  1 | import OpenAI from 'openai'
  2 | import Anthropic from '@anthropic-ai/sdk'
  3 | import ollama from 'ollama'
  4 | import {
  5 |   geminiFlashThinkingModel,
  6 |   geminiModel,
  7 |   geminiFlashModel,
  8 | } from './gemini'
  9 | 
 10 | const openai = new OpenAI({
 11 |   apiKey: process.env.OPENAI_API_KEY || '',
 12 | })
 13 | 
 14 | const deepseek = new OpenAI({
 15 |   baseURL: 'https://api.deepseek.com',
 16 |   apiKey: process.env.DEEPSEEK_API_KEY || '',
 17 | })
 18 | 
 19 | const anthropic = new Anthropic({
 20 |   apiKey: process.env.ANTHROPIC_API_KEY || '',
 21 | })
 22 | 
 23 | type DeepSeekMessage = {
 24 |   role: 'user' | 'assistant' | 'system'
 25 |   content: string
 26 | }
 27 | 
 28 | export async function generateWithGemini(
 29 |   systemPrompt: string,
 30 |   model: string
 31 | ): Promise<string> {
 32 |   let result
 33 |   if (model === 'gemini-flash-thinking') {
 34 |     result = await geminiFlashThinkingModel.generateContent(systemPrompt)
 35 |   } else if (model === 'gemini-exp') {
 36 |     result = await geminiModel.generateContent(systemPrompt)
 37 |   } else {
 38 |     result = await geminiFlashModel.generateContent(systemPrompt)
 39 |   }
 40 |   const text = result.response.text()
 41 |   if (!text) {
 42 |     throw new Error('No response content from Gemini')
 43 |   }
 44 |   return text
 45 | }
 46 | 
 47 | export async function generateWithOpenAI(
 48 |   systemPrompt: string,
 49 |   model: string
 50 | ): Promise<string> {
 51 |   const response = await openai.chat.completions.create({
 52 |     model,
 53 |     messages: [
 54 |       {
 55 |         role: 'user',
 56 |         content: systemPrompt,
 57 |       },
 58 |     ],
 59 |   })
 60 |   const content = response.choices[0].message.content
 61 |   if (!content) {
 62 |     throw new Error('No response content from OpenAI')
 63 |   }
 64 |   return content
 65 | }
 66 | 
 67 | export async function generateWithDeepSeek(
 68 |   systemPrompt: string,
 69 |   model: string
 70 | ): Promise<string> {
 71 |   try {
 72 |     const messages: DeepSeekMessage[] = [
 73 |       {
 74 |         role: 'user',
 75 |         content: systemPrompt,
 76 |       },
 77 |     ]
 78 | 
 79 |     const response = await deepseek.chat.completions.create({
 80 |       model: `deepseek-${model}`,
 81 |       messages: messages as any,
 82 |       max_tokens: 4000,
 83 |     })
 84 | 
 85 |     const content = response.choices[0].message.content
 86 |     if (!content) {
 87 |       throw new Error('No response content from DeepSeek')
 88 |     }
 89 | 
 90 |     // For the reasoner model, we can get additional reasoning content
 91 |     if (
 92 |       model === 'deepseek-reasoner' &&
 93 |       (response.choices[0].message as any).reasoning_content
 94 |     ) {
 95 |       console.log(
 96 |         'DeepSeek reasoning:',
 97 |         (response.choices[0].message as any).reasoning_content
 98 |       )
 99 |     }
100 | 
101 |     return content
102 |   } catch (error) {
103 |     console.error('DeepSeek API error:', error)
104 |     throw error
105 |   }
106 | }
107 | 
108 | export async function generateWithAnthropic(
109 |   systemPrompt: string,
110 |   model: string
111 | ): Promise<string> {
112 |   const response = await anthropic.messages.create({
113 |     model,
114 |     max_tokens: 3500,
115 |     temperature: 0.9,
116 |     messages: [
117 |       {
118 |         role: 'user',
119 |         content: systemPrompt,
120 |       },
121 |     ],
122 |   })
123 |   const content = response.content[0].text
124 |   if (!content) {
125 |     throw new Error('No response content from Anthropic')
126 |   }
127 |   return content
128 | }
129 | 
130 | export async function generateWithOllama(
131 |   systemPrompt: string,
132 |   model: string
133 | ): Promise<string> {
134 |   const response = await ollama.chat({
135 |     model: model.replace('ollama__', ''),
136 |     messages: [{ role: 'user', content: systemPrompt }],
137 |   })
138 |   const content = response.message.content
139 |   if (!content) {
140 |     throw new Error('No response content from Ollama')
141 |   }
142 |   return content
143 | }
144 | 
145 | export async function generateWithOpenRouter(
146 |   systemPrompt: string,
147 |   model: string
148 | ): Promise<string> {
149 |   const response = await fetch(
150 |     'https://openrouter.ai/api/v1/chat/completions',
151 |     {
152 |       method: 'POST',
153 |       headers: {
154 |         Authorization: `Bearer ${process.env.OPENROUTER_API_KEY}`,
155 |         'Content-Type': 'application/json',
156 |       },
157 |       body: JSON.stringify({
158 |         model: model.replace('openrouter__', ''),
159 |         messages: [
160 |           {
161 |             role: 'user',
162 |             content: systemPrompt,
163 |           },
164 |         ],
165 |       }),
166 |     }
167 |   )
168 | 
169 |   const responseData = await response.text()
170 | 
171 |   if (!response.ok) {
172 |     throw new Error(`OpenRouter API error: ${responseData}`)
173 |   }
174 | 
175 |   let data
176 |   try {
177 |     data = JSON.parse(responseData)
178 |   } catch (e) {
179 |     console.log('error', e)
180 |     throw new Error(`Failed to parse OpenRouter response: ${responseData}`)
181 |   }
182 | 
183 |   if (!data.choices?.[0]?.message?.content) {
184 |     throw new Error(
185 |       `Invalid OpenRouter response format: ${JSON.stringify(data)}`
186 |     )
187 |   }
188 | 
189 |   return data.choices[0].message.content
190 | }
191 | 
192 | export async function generateWithModel(
193 |   systemPrompt: string,
194 |   platformModel: string
195 | ): Promise<string> {
196 |   const [platform, model] = platformModel.split('__')
197 | 
198 |   switch (platform) {
199 |     case 'google':
200 |       return generateWithGemini(systemPrompt, model)
201 |     case 'openai':
202 |       return generateWithOpenAI(systemPrompt, model)
203 |     case 'deepseek':
204 |       return generateWithDeepSeek(systemPrompt, model)
205 |     case 'anthropic':
206 |       return generateWithAnthropic(systemPrompt, model)
207 |     case 'ollama':
208 |       return generateWithOllama(systemPrompt, model)
209 |     case 'openrouter':
210 |       return generateWithOpenRouter(systemPrompt, model)
211 |     default:
212 |       throw new Error('Invalid platform specified')
213 |   }
214 | }
215 | 


--------------------------------------------------------------------------------
/components/ui/select.tsx:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | import * as React from "react"
  4 | import * as SelectPrimitive from "@radix-ui/react-select"
  5 | import { Check, ChevronDown, ChevronUp } from "lucide-react"
  6 | 
  7 | import { cn } from "@/lib/utils"
  8 | 
  9 | const Select = SelectPrimitive.Root
 10 | 
 11 | const SelectGroup = SelectPrimitive.Group
 12 | 
 13 | const SelectValue = SelectPrimitive.Value
 14 | 
 15 | const SelectTrigger = React.forwardRef<
 16 |   React.ElementRef<typeof SelectPrimitive.Trigger>,
 17 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.Trigger>
 18 | >(({ className, children, ...props }, ref) => (
 19 |   <SelectPrimitive.Trigger
 20 |     ref={ref}
 21 |     className={cn(
 22 |       "flex h-9 w-full items-center justify-between whitespace-nowrap rounded-md border border-input bg-transparent px-3 py-2 text-sm shadow-sm ring-offset-background placeholder:text-muted-foreground focus:outline-none focus:ring-1 focus:ring-ring disabled:cursor-not-allowed disabled:opacity-50 [&>span]:line-clamp-1",
 23 |       className
 24 |     )}
 25 |     {...props}
 26 |   >
 27 |     {children}
 28 |     <SelectPrimitive.Icon asChild>
 29 |       <ChevronDown className="h-4 w-4 opacity-50" />
 30 |     </SelectPrimitive.Icon>
 31 |   </SelectPrimitive.Trigger>
 32 | ))
 33 | SelectTrigger.displayName = SelectPrimitive.Trigger.displayName
 34 | 
 35 | const SelectScrollUpButton = React.forwardRef<
 36 |   React.ElementRef<typeof SelectPrimitive.ScrollUpButton>,
 37 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.ScrollUpButton>
 38 | >(({ className, ...props }, ref) => (
 39 |   <SelectPrimitive.ScrollUpButton
 40 |     ref={ref}
 41 |     className={cn(
 42 |       "flex cursor-default items-center justify-center py-1",
 43 |       className
 44 |     )}
 45 |     {...props}
 46 |   >
 47 |     <ChevronUp className="h-4 w-4" />
 48 |   </SelectPrimitive.ScrollUpButton>
 49 | ))
 50 | SelectScrollUpButton.displayName = SelectPrimitive.ScrollUpButton.displayName
 51 | 
 52 | const SelectScrollDownButton = React.forwardRef<
 53 |   React.ElementRef<typeof SelectPrimitive.ScrollDownButton>,
 54 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.ScrollDownButton>
 55 | >(({ className, ...props }, ref) => (
 56 |   <SelectPrimitive.ScrollDownButton
 57 |     ref={ref}
 58 |     className={cn(
 59 |       "flex cursor-default items-center justify-center py-1",
 60 |       className
 61 |     )}
 62 |     {...props}
 63 |   >
 64 |     <ChevronDown className="h-4 w-4" />
 65 |   </SelectPrimitive.ScrollDownButton>
 66 | ))
 67 | SelectScrollDownButton.displayName =
 68 |   SelectPrimitive.ScrollDownButton.displayName
 69 | 
 70 | const SelectContent = React.forwardRef<
 71 |   React.ElementRef<typeof SelectPrimitive.Content>,
 72 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.Content>
 73 | >(({ className, children, position = "popper", ...props }, ref) => (
 74 |   <SelectPrimitive.Portal>
 75 |     <SelectPrimitive.Content
 76 |       ref={ref}
 77 |       className={cn(
 78 |         "relative z-50 max-h-96 min-w-[8rem] overflow-hidden rounded-md border bg-popover text-popover-foreground shadow-md data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2",
 79 |         position === "popper" &&
 80 |           "data-[side=bottom]:translate-y-1 data-[side=left]:-translate-x-1 data-[side=right]:translate-x-1 data-[side=top]:-translate-y-1",
 81 |         className
 82 |       )}
 83 |       position={position}
 84 |       {...props}
 85 |     >
 86 |       <SelectScrollUpButton />
 87 |       <SelectPrimitive.Viewport
 88 |         className={cn(
 89 |           "p-1",
 90 |           position === "popper" &&
 91 |             "h-[var(--radix-select-trigger-height)] w-full min-w-[var(--radix-select-trigger-width)]"
 92 |         )}
 93 |       >
 94 |         {children}
 95 |       </SelectPrimitive.Viewport>
 96 |       <SelectScrollDownButton />
 97 |     </SelectPrimitive.Content>
 98 |   </SelectPrimitive.Portal>
 99 | ))
100 | SelectContent.displayName = SelectPrimitive.Content.displayName
101 | 
102 | const SelectLabel = React.forwardRef<
103 |   React.ElementRef<typeof SelectPrimitive.Label>,
104 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.Label>
105 | >(({ className, ...props }, ref) => (
106 |   <SelectPrimitive.Label
107 |     ref={ref}
108 |     className={cn("px-2 py-1.5 text-sm font-semibold", className)}
109 |     {...props}
110 |   />
111 | ))
112 | SelectLabel.displayName = SelectPrimitive.Label.displayName
113 | 
114 | const SelectItem = React.forwardRef<
115 |   React.ElementRef<typeof SelectPrimitive.Item>,
116 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.Item>
117 | >(({ className, children, ...props }, ref) => (
118 |   <SelectPrimitive.Item
119 |     ref={ref}
120 |     className={cn(
121 |       "relative flex w-full cursor-default select-none items-center rounded-sm py-1.5 pl-2 pr-8 text-sm outline-none focus:bg-accent focus:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50",
122 |       className
123 |     )}
124 |     {...props}
125 |   >
126 |     <span className="absolute right-2 flex h-3.5 w-3.5 items-center justify-center">
127 |       <SelectPrimitive.ItemIndicator>
128 |         <Check className="h-4 w-4" />
129 |       </SelectPrimitive.ItemIndicator>
130 |     </span>
131 |     <SelectPrimitive.ItemText>{children}</SelectPrimitive.ItemText>
132 |   </SelectPrimitive.Item>
133 | ))
134 | SelectItem.displayName = SelectPrimitive.Item.displayName
135 | 
136 | const SelectSeparator = React.forwardRef<
137 |   React.ElementRef<typeof SelectPrimitive.Separator>,
138 |   React.ComponentPropsWithoutRef<typeof SelectPrimitive.Separator>
139 | >(({ className, ...props }, ref) => (
140 |   <SelectPrimitive.Separator
141 |     ref={ref}
142 |     className={cn("-mx-1 my-1 h-px bg-muted", className)}
143 |     {...props}
144 |   />
145 | ))
146 | SelectSeparator.displayName = SelectPrimitive.Separator.displayName
147 | 
148 | export {
149 |   Select,
150 |   SelectGroup,
151 |   SelectValue,
152 |   SelectTrigger,
153 |   SelectContent,
154 |   SelectLabel,
155 |   SelectItem,
156 |   SelectSeparator,
157 |   SelectScrollUpButton,
158 |   SelectScrollDownButton,
159 | }
160 | 


--------------------------------------------------------------------------------
/components/flow/selection-node.tsx:
--------------------------------------------------------------------------------
  1 | import { memo, useState, useEffect } from 'react'
  2 | import { Handle, Position } from '@xyflow/react'
  3 | import { Card, CardContent } from '@/components/ui/card'
  4 | import { Button } from '@/components/ui/button'
  5 | import { Checkbox } from '@/components/ui/checkbox'
  6 | import { FileText, AlertTriangle, Loader2 } from 'lucide-react'
  7 | import { Input } from '@/components/ui/input'
  8 | import type { SearchResult, SelectionNodeData } from '@/types'
  9 | import { CONFIG } from '@/lib/config'
 10 | 
 11 | const MAX_SELECTIONS = CONFIG.search.maxSelectableResults
 12 | 
 13 | export const SelectionNode = memo(function SelectionNode({
 14 |   data,
 15 | }: {
 16 |   data: SelectionNodeData
 17 | }) {
 18 |   const [selectedResults, setSelectedResults] = useState<SearchResult[]>([])
 19 |   const [prompt, setPrompt] = useState('')
 20 |   const [error, setError] = useState<string | null>(null)
 21 |   const [isGenerating, setIsGenerating] = useState(false)
 22 | 
 23 |   // Reset selection when results change
 24 |   useEffect(() => {
 25 |     setSelectedResults([])
 26 |     setError(null)
 27 |   }, [data.results])
 28 | 
 29 |   const handleSelect = (result: SearchResult) => {
 30 |     setSelectedResults((prev) => {
 31 |       if (prev.find((r) => r.id === result.id)) {
 32 |         return prev.filter((r) => r.id !== result.id)
 33 |       }
 34 |       if (prev.length >= MAX_SELECTIONS) return prev
 35 |       return [...prev, result]
 36 |     })
 37 |   }
 38 | 
 39 |   const handleGenerateReport = () => {
 40 |     if (selectedResults.length === 0) {
 41 |       setError('Please select at least one result')
 42 |       return
 43 |     }
 44 | 
 45 |     // Clear any previous errors
 46 |     setError(null)
 47 | 
 48 |     // Check if onGenerateReport is defined
 49 |     if (typeof data.onGenerateReport !== 'function') {
 50 |       setError('Report generation is not available')
 51 |       return
 52 |     }
 53 | 
 54 |     setIsGenerating(true)
 55 | 
 56 |     try {
 57 |       // Call onGenerateReport and capture the result
 58 |       const result = data.onGenerateReport(selectedResults, prompt)
 59 | 
 60 |       // Check if the result is a Promise using Promise.resolve
 61 |       Promise.resolve(result)
 62 |         .then(() => {
 63 |           // Report generation completed successfully
 64 |         })
 65 |         .catch((err) => {
 66 |           setError(
 67 |             `Error generating report: ${
 68 |               err instanceof Error ? err.message : 'Unknown error'
 69 |             }`
 70 |           )
 71 |         })
 72 |         .finally(() => {
 73 |           setIsGenerating(false)
 74 |         })
 75 |     } catch (err) {
 76 |       setIsGenerating(false)
 77 |       setError(
 78 |         `Error generating report: ${
 79 |           err instanceof Error ? err.message : 'Unknown error'
 80 |         }`
 81 |       )
 82 |     }
 83 |   }
 84 | 
 85 |   const isSelectionValid = selectedResults.length > 0 && !error
 86 | 
 87 |   return (
 88 |     <div className='w-auto max-w-[600px] mx-auto'>
 89 |       <Card>
 90 |         <Handle type='target' position={Position.Top} />
 91 |         <CardContent className='p-6'>
 92 |           <div className='space-y-6'>
 93 |             <div className='flex items-center justify-between'>
 94 |               <h3 className='font-medium text-lg flex items-center gap-3'>
 95 |                 <FileText className='h-6 w-6 text-blue-500' />
 96 |                 Search Results
 97 |               </h3>
 98 |               <Button
 99 |                 size='default'
100 |                 disabled={!isSelectionValid || isGenerating}
101 |                 onClick={handleGenerateReport}
102 |                 className='gap-2'
103 |               >
104 |                 {isGenerating ? (
105 |                   <Loader2 className='h-4 w-4 animate-spin' />
106 |                 ) : (
107 |                   <FileText className='h-4 w-4' />
108 |                 )}
109 |                 {isGenerating
110 |                   ? 'Generating...'
111 |                   : `Generate Report (${selectedResults.length})`}
112 |               </Button>
113 |             </div>
114 | 
115 |             {error && (
116 |               <div className='bg-red-50 border border-red-200 p-3 rounded-md flex items-center gap-2 text-red-700'>
117 |                 <AlertTriangle className='h-4 w-4' />
118 |                 <p className='text-sm'>{error}</p>
119 |               </div>
120 |             )}
121 | 
122 |             <p className='text-gray-600'>
123 |               Select up to {MAX_SELECTIONS} results to analyze (
124 |               {selectedResults.length} selected)
125 |             </p>
126 | 
127 |             {selectedResults.length > 0 && (
128 |               <Input
129 |                 value={prompt}
130 |                 onChange={(e) => setPrompt(e.target.value)}
131 |                 placeholder='What would you like to know about these sources?'
132 |               />
133 |             )}
134 | 
135 |             <div className='space-y-4 max-h-[400px] overflow-y-auto pr-4 nowheel nodrag'>
136 |               {data.results.map((result) => (
137 |                 <div
138 |                   key={result.id}
139 |                   className='flex items-start gap-4 bg-gray-50 p-4 rounded-lg'
140 |                 >
141 |                   <div className='pt-1'>
142 |                     <Checkbox
143 |                       checked={selectedResults.some((r) => r.id === result.id)}
144 |                       onCheckedChange={() => handleSelect(result)}
145 |                       disabled={
146 |                         (!selectedResults.some((r) => r.id === result.id) &&
147 |                           selectedResults.length >= MAX_SELECTIONS) ||
148 |                         isGenerating
149 |                       }
150 |                     />
151 |                   </div>
152 |                   <div className='space-y-2 flex-1'>
153 |                     <a
154 |                       href={result.url}
155 |                       target='_blank'
156 |                       rel='noopener noreferrer'
157 |                       className='text-blue-600 hover:underline font-medium block'
158 |                     >
159 |                       {result.name}
160 |                     </a>
161 |                     <p className='text-sm text-green-700 truncate'>
162 |                       {result.url}
163 |                     </p>
164 |                     <p className='text-gray-600 line-clamp-3'>
165 |                       {result.snippet}
166 |                     </p>
167 |                   </div>
168 |                 </div>
169 |               ))}
170 |             </div>
171 |           </div>
172 |         </CardContent>
173 |         <Handle type='source' position={Position.Bottom} />
174 |       </Card>
175 |     </div>
176 |   )
177 | })
178 | 


--------------------------------------------------------------------------------
/app/api/consolidate-report/route.ts:
--------------------------------------------------------------------------------
  1 | import { NextResponse } from 'next/server'
  2 | import { CONFIG } from '@/lib/config'
  3 | import { generateWithModel } from '@/lib/models'
  4 | import type { Report } from '@/types'
  5 | import { reportContentRatelimit } from '@/lib/redis'
  6 | 
  7 | export async function POST(request: Request) {
  8 |   try {
  9 |     const { reports, platformModel } = await request.json()
 10 |     const [platform, model] = platformModel.split('__')
 11 | 
 12 |     if (CONFIG.rateLimits.enabled && platform !== 'ollama') {
 13 |       const { success } = await reportContentRatelimit.limit('report')
 14 |       if (!success) {
 15 |         return NextResponse.json(
 16 |           { error: 'Too many requests' },
 17 |           { status: 429 }
 18 |         )
 19 |       }
 20 |     }
 21 | 
 22 |     console.log('Consolidating reports:', {
 23 |       numReports: reports.length,
 24 |       reportTitles: reports.map((r: Report) => r.title),
 25 |       platform,
 26 |       model,
 27 |     })
 28 | 
 29 |     if (!reports?.length) {
 30 |       return NextResponse.json(
 31 |         { error: 'Reports are required' },
 32 |         { status: 400 }
 33 |       )
 34 |     }
 35 | 
 36 |     // Collect all unique sources from all reports
 37 |     const allSources: { id: string; url: string; name: string }[] = []
 38 |     const sourceMap = new Map<string, number>() // Maps source id to index in allSources
 39 | 
 40 |     reports.forEach((report: Report) => {
 41 |       if (report.sources && report.sources.length > 0) {
 42 |         report.sources.forEach((source) => {
 43 |           if (!sourceMap.has(source.id)) {
 44 |             sourceMap.set(source.id, allSources.length)
 45 |             allSources.push(source)
 46 |           }
 47 |         })
 48 |       }
 49 |     })
 50 | 
 51 |     // Create source index for citations
 52 |     const sourceIndex = allSources
 53 |       .map(
 54 |         (source, index) =>
 55 |           `[${index + 1}] Source: ${source.name} - ${source.url}`
 56 |       )
 57 |       .join('\n')
 58 | 
 59 |     const prompt = `Create a comprehensive consolidated report that synthesizes the following research reports:
 60 | 
 61 | ${reports
 62 |   .map(
 63 |     (report: Report, index: number) => `
 64 | Report ${index + 1} Title: ${report.title}
 65 | Report ${index + 1} Summary: ${report.summary}
 66 | Key Findings:
 67 | ${report.sections
 68 |   ?.map((section) => `- ${section.title}: ${section.content}`)
 69 |   .join('\n')}
 70 | `
 71 |   )
 72 |   .join('\n\n')}
 73 | 
 74 | Sources for citation:
 75 | ${sourceIndex}
 76 | 
 77 | Analyze and synthesize these reports to create a comprehensive consolidated report that:
 78 | 1. Identifies common themes and patterns across the reports
 79 | 2. Highlights key insights and findings
 80 | 3. Shows how different reports complement or contrast each other
 81 | 4. Draws overarching conclusions
 82 | 5. Suggests potential areas for further research
 83 | 6. Uses citations only when necessary to reference specific claims, statistics, or quotes from sources
 84 | 
 85 | Format the response as a structured report with:
 86 | - A clear title that encompasses the overall research topic
 87 | - An executive summary of the consolidated findings
 88 | - Detailed sections that analyze different aspects
 89 | - A conclusion that ties everything together
 90 | - Judicious use of citations in superscript format [¹], [²], etc. ONLY when necessary
 91 | 
 92 | Return the response in the following JSON format:
 93 | {
 94 |   "title": "Overall Research Topic Title",
 95 |   "summary": "Executive summary of findings",
 96 |   "sections": [
 97 |     {
 98 |       "title": "Section Title",
 99 |       "content": "Section content with selective citations"
100 |     }
101 |   ],
102 |   "usedSources": [1, 2] // Array of source numbers that were actually cited in the report
103 | }
104 | 
105 | CITATION GUIDELINES:
106 | 1. Only use citations when truly necessary - specifically for:
107 |    - Direct quotes from sources
108 |    - Specific statistics, figures, or data points
109 |    - Non-obvious facts or claims that need verification
110 |    - Controversial statements
111 |    
112 | 2. DO NOT use citations for:
113 |    - General knowledge
114 |    - Your own analysis or synthesis of information
115 |    - Widely accepted facts
116 |    - Every sentence or paragraph
117 | 
118 | 3. When needed, use superscript citation numbers in square brackets [¹], [²], etc. at the end of the relevant sentence
119 |    
120 | 4. The citation numbers correspond directly to the source numbers provided in the list
121 |    
122 | 5. Be judicious and selective with citations - a well-written report should flow naturally with citations only where they truly add credibility
123 | 
124 | 6. You DO NOT need to cite every source provided. Only cite the sources that contain information directly relevant to the report. Track which sources you actually cite and include their numbers in the "usedSources" array in the output JSON.
125 | 
126 | 7. It's completely fine if some sources aren't cited at all - this means they weren't needed for the specific analysis requested.`
127 | 
128 |     console.log('Generated prompt:', prompt)
129 | 
130 |     try {
131 |       const response = await generateWithModel(prompt, platformModel)
132 | 
133 |       if (!response) {
134 |         throw new Error('No response from model')
135 |       }
136 | 
137 |       console.log('Model response:', response)
138 | 
139 |       // Try to parse the response as JSON, if it's not already
140 |       let parsedResponse
141 |       try {
142 |         parsedResponse =
143 |           typeof response === 'string' ? JSON.parse(response) : response
144 |         console.log('Parsed response:', parsedResponse)
145 |       } catch (parseError) {
146 |         console.error('Failed to parse response as JSON:', parseError)
147 |         // If it's not JSON, create a basic report structure
148 |         parsedResponse = {
149 |           title: 'Consolidated Research Report',
150 |           summary: response.split('\n\n')[0] || 'Summary not available',
151 |           sections: [
152 |             {
153 |               title: 'Findings',
154 |               content: response,
155 |             },
156 |           ],
157 |         }
158 |       }
159 | 
160 |       // Add sources to the response
161 |       parsedResponse.sources = allSources
162 | 
163 |       return NextResponse.json(parsedResponse)
164 |     } catch (error) {
165 |       console.error('Model generation error:', error)
166 |       return NextResponse.json(
167 |         { error: 'Failed to generate consolidated report' },
168 |         { status: 500 }
169 |       )
170 |     }
171 |   } catch (error) {
172 |     console.error('Consolidation error:', error)
173 |     return NextResponse.json(
174 |       { error: 'Failed to consolidate reports' },
175 |       { status: 500 }
176 |     )
177 |   }
178 | }
179 | 


--------------------------------------------------------------------------------
/app/api/report/route.ts:
--------------------------------------------------------------------------------
  1 | import { NextResponse } from 'next/server'
  2 | import { reportContentRatelimit } from '@/lib/redis'
  3 | import { type Article, type ModelVariant } from '@/types'
  4 | import { CONFIG } from '@/lib/config'
  5 | import { extractAndParseJSON } from '@/lib/utils'
  6 | import { generateWithModel } from '@/lib/models'
  7 | 
  8 | export const maxDuration = 60
  9 | 
 10 | export async function POST(request: Request) {
 11 |   try {
 12 |     const body = await request.json()
 13 |     const {
 14 |       selectedResults,
 15 |       sources,
 16 |       prompt,
 17 |       platformModel = 'google-gemini-flash',
 18 |     } = body as {
 19 |       selectedResults: Article[]
 20 |       sources: any[]
 21 |       prompt: string
 22 |       platformModel: ModelVariant
 23 |     }
 24 | 
 25 |     // Only check rate limit if enabled and not using Ollama (local model)
 26 |     const platform = platformModel.split('__')[0]
 27 |     const model = platformModel.split('__')[1]
 28 |     if (CONFIG.rateLimits.enabled && platform !== 'ollama') {
 29 |       const { success } = await reportContentRatelimit.limit('report')
 30 |       if (!success) {
 31 |         return NextResponse.json(
 32 |           { error: 'Too many requests' },
 33 |           { status: 429 }
 34 |         )
 35 |       }
 36 |     }
 37 | 
 38 |     // Check if selected platform is enabled
 39 |     const platformConfig =
 40 |       CONFIG.platforms[platform as keyof typeof CONFIG.platforms]
 41 |     if (!platformConfig?.enabled) {
 42 |       return NextResponse.json(
 43 |         { error: `${platform} platform is not enabled` },
 44 |         { status: 400 }
 45 |       )
 46 |     }
 47 | 
 48 |     // Check if selected model exists and is enabled
 49 |     const modelConfig = (platformConfig as any).models[model]
 50 |     if (!modelConfig) {
 51 |       return NextResponse.json(
 52 |         { error: `${model} model does not exist` },
 53 |         { status: 400 }
 54 |       )
 55 |     }
 56 |     if (!modelConfig.enabled) {
 57 |       return NextResponse.json(
 58 |         { error: `${model} model is disabled` },
 59 |         { status: 400 }
 60 |       )
 61 |     }
 62 | 
 63 |     const generateSystemPrompt = (articles: Article[], userPrompt: string) => {
 64 |       return `You are a research assistant tasked with creating a comprehensive report based on multiple sources. 
 65 | The report should specifically address this request: "${userPrompt}"
 66 | 
 67 | Your report should:
 68 | 1. Have a clear title that reflects the specific analysis requested
 69 | 2. Begin with a concise executive summary
 70 | 3. Be organized into relevant sections based on the analysis requested
 71 | 4. Use markdown formatting for emphasis, lists, and structure
 72 | 5. Use citations ONLY when necessary for specific claims, statistics, direct quotes, or important facts
 73 | 6. Maintain objectivity while addressing the specific aspects requested in the prompt
 74 | 7. Compare and contrast the information from sources, noting areas of consensus or points of contention
 75 | 8. Showcase key insights, important data, or innovative ideas
 76 | 
 77 | Here are the source articles to analyze (numbered for citation purposes):
 78 | 
 79 | ${articles
 80 |   .map(
 81 |     (article, index) => `
 82 | [${index + 1}] Title: ${article.title}
 83 | URL: ${article.url}
 84 | Content: ${article.content}
 85 | ---
 86 | `
 87 |   )
 88 |   .join('\n')}
 89 | 
 90 | Format the report as a JSON object with the following structure:
 91 | {
 92 |   "title": "Report title",
 93 |   "summary": "Executive summary (can include markdown)",
 94 |   "sections": [
 95 |     {
 96 |       "title": "Section title",
 97 |       "content": "Section content with markdown formatting and selective citations"
 98 |     }
 99 |   ],
100 |   "usedSources": [1, 2] // Array of source numbers that were actually cited in the report
101 | }
102 | 
103 | Use markdown formatting in the content to improve readability:
104 | - Use **bold** for emphasis
105 | - Use bullet points and numbered lists where appropriate
106 | - Use headings and subheadings with # syntax
107 | - Include code blocks if relevant
108 | - Use > for quotations
109 | - Use --- for horizontal rules where appropriate
110 | 
111 | CITATION GUIDELINES:
112 | 1. Only use citations when truly necessary - specifically for:
113 |    - Direct quotes from sources
114 |    - Specific statistics, figures, or data points
115 |    - Non-obvious facts or claims that need verification
116 |    - Controversial statements
117 |    
118 | 2. DO NOT use citations for:
119 |    - General knowledge
120 |    - Your own analysis or synthesis of information
121 |    - Widely accepted facts
122 |    - Every sentence or paragraph
123 | 
124 | 3. When needed, use superscript citation numbers in square brackets [¹], [²], etc. at the end of the relevant sentence
125 |    
126 | 4. The citation numbers correspond directly to the source numbers provided in the list
127 |    
128 | 5. Be judicious and selective with citations - a well-written report should flow naturally with citations only where they truly add credibility
129 | 
130 | 6. You DO NOT need to cite every source provided. Only cite the sources that contain information directly relevant to the report. Track which sources you actually cite and include their numbers in the "usedSources" array in the output JSON.
131 | 
132 | 7. It's completely fine if some sources aren't cited at all - this means they weren't needed for the specific analysis requested.`
133 |     }
134 | 
135 |     const systemPrompt = generateSystemPrompt(selectedResults, prompt)
136 | 
137 |     // console.log('Sending prompt to model:', systemPrompt)
138 |     console.log('Model:', model)
139 | 
140 |     try {
141 |       const response = await generateWithModel(systemPrompt, platformModel)
142 | 
143 |       if (!response) {
144 |         throw new Error('No response from model')
145 |       }
146 | 
147 |       try {
148 |         const reportData = extractAndParseJSON(response)
149 |         // Add sources to the report data
150 |         reportData.sources = sources
151 |         console.log('Parsed report data:', reportData)
152 |         return NextResponse.json(reportData)
153 |       } catch (parseError) {
154 |         console.error('JSON parsing error:', parseError)
155 |         return NextResponse.json(
156 |           { error: 'Failed to parse report format' },
157 |           { status: 500 }
158 |         )
159 |       }
160 |     } catch (error) {
161 |       console.error('Model generation error:', error)
162 |       return NextResponse.json(
163 |         { error: 'Failed to generate report content' },
164 |         { status: 500 }
165 |       )
166 |     }
167 |   } catch (error) {
168 |     console.error('Report generation error:', error)
169 |     return NextResponse.json(
170 |       { error: 'Failed to generate report' },
171 |       { status: 500 }
172 |     )
173 |   }
174 | }
175 | 


--------------------------------------------------------------------------------
/components/knowledge-base-sidebar.tsx:
--------------------------------------------------------------------------------
  1 | import { Brain, Search, Trash2, AlertTriangle } from 'lucide-react'
  2 | import { Button } from '@/components/ui/button'
  3 | import { Input } from '@/components/ui/input'
  4 | import { Card } from '@/components/ui/card'
  5 | import { ScrollArea } from '@/components/ui/scroll-area'
  6 | import { formatDistanceToNow } from 'date-fns'
  7 | import { useKnowledgeBase } from '@/hooks/use-knowledge-base'
  8 | import { useState } from 'react'
  9 | import {
 10 |   Sheet,
 11 |   SheetContent,
 12 |   SheetHeader,
 13 |   SheetTitle,
 14 | } from '@/components/ui/sheet'
 15 | import Link from 'next/link'
 16 | import { Alert, AlertDescription, AlertTitle } from '@/components/ui/alert'
 17 | 
 18 | interface KnowledgeBaseSidebarProps {
 19 |   open: boolean
 20 |   onOpenChange: (open: boolean) => void
 21 | }
 22 | 
 23 | export function KnowledgeBaseSidebar({
 24 |   open,
 25 |   onOpenChange,
 26 | }: KnowledgeBaseSidebarProps) {
 27 |   const { reports, searchReports, clearAllReports } = useKnowledgeBase()
 28 |   const [searchQuery, setSearchQuery] = useState('')
 29 |   const [showClearConfirm, setShowClearConfirm] = useState(false)
 30 | 
 31 |   const filteredReports = searchQuery ? searchReports(searchQuery) : reports
 32 | 
 33 |   return (
 34 |     <Sheet open={open} onOpenChange={onOpenChange}>
 35 |       <SheetContent side='left' className='w-[400px] sm:w-[540px]'>
 36 |         <SheetHeader>
 37 |           <div className='flex items-center justify-between'>
 38 |             <SheetTitle className='flex items-center gap-2'>
 39 |               <Brain className='h-6 w-6' />
 40 |               Knowledge Base
 41 |             </SheetTitle>
 42 |           </div>
 43 |         </SheetHeader>
 44 |         <div className='mt-8 space-y-4'>
 45 |           {showClearConfirm && (
 46 |             <Alert variant='destructive'>
 47 |               <AlertTriangle className='h-4 w-4' />
 48 |               <AlertTitle>Clear All Reports?</AlertTitle>
 49 |               <AlertDescription className='space-y-2'>
 50 |                 <p>
 51 |                   This will permanently delete all {reports.length} report
 52 |                   {reports.length === 1 ? '' : 's'} from your knowledge base.
 53 |                 </p>
 54 |                 <div className='flex gap-2 mt-2'>
 55 |                   <Button
 56 |                     variant='outline'
 57 |                     size='sm'
 58 |                     onClick={() => setShowClearConfirm(false)}
 59 |                   >
 60 |                     Cancel
 61 |                   </Button>
 62 |                   <Button
 63 |                     variant='destructive'
 64 |                     size='sm'
 65 |                     onClick={() => {
 66 |                       clearAllReports()
 67 |                       setShowClearConfirm(false)
 68 |                     }}
 69 |                   >
 70 |                     Yes, Clear All
 71 |                   </Button>
 72 |                 </div>
 73 |               </AlertDescription>
 74 |             </Alert>
 75 |           )}
 76 | 
 77 |           {reports.length > 0 && !showClearConfirm && (
 78 |             <Alert>
 79 |               <AlertTitle className='font-semibold'>Current Status</AlertTitle>
 80 |               <AlertDescription>
 81 |                 You have {reports.length} saved report
 82 |                 {reports.length === 1 ? '' : 's'}. Click on a report to view
 83 |                 details or use the search to find specific reports.
 84 |               </AlertDescription>
 85 |             </Alert>
 86 |           )}
 87 |           <div className='relative'>
 88 |             <Input
 89 |               type='text'
 90 |               placeholder='Search saved reports...'
 91 |               value={searchQuery}
 92 |               onChange={(e) => setSearchQuery(e.target.value)}
 93 |               className='pr-8'
 94 |             />
 95 |             <Search className='absolute right-2 top-2.5 h-4 w-4 text-gray-400' />
 96 |           </div>
 97 |           <ScrollArea className='h-[calc(100vh-280px)]'>
 98 |             <div className='space-y-2'>
 99 |               {filteredReports.map((savedReport) => (
100 |                 <Card
101 |                   key={savedReport.id}
102 |                   className='p-4 hover:bg-gray-50 transition-colors'
103 |                 >
104 |                   <div className='flex justify-between items-start gap-2'>
105 |                     <Link
106 |                       href={`/report/${savedReport.id}`}
107 |                       className='flex-1 min-w-0 cursor-pointer'
108 |                     >
109 |                       <h3 className='font-semibold truncate'>
110 |                         {savedReport.report.title}
111 |                       </h3>
112 |                       <p className='text-sm text-gray-500 truncate'>
113 |                         {savedReport.query}
114 |                       </p>
115 |                       <p className='text-xs text-gray-400'>
116 |                         {formatDistanceToNow(savedReport.timestamp, {
117 |                           addSuffix: true,
118 |                         })}
119 |                       </p>
120 |                     </Link>
121 |                   </div>
122 |                 </Card>
123 |               ))}
124 |               {filteredReports.length === 0 && (
125 |                 <Alert variant='destructive'>
126 |                   <AlertTriangle className='h-4 w-4' />
127 |                   <AlertTitle>No Reports Found</AlertTitle>
128 |                   <AlertDescription>
129 |                     {searchQuery
130 |                       ? 'No reports match your search query. Try a different search term.'
131 |                       : 'Your knowledge base is empty. Save some reports to get started.'}
132 |                   </AlertDescription>
133 |                 </Alert>
134 |               )}
135 |             </div>
136 |           </ScrollArea>
137 |         </div>
138 |         <div className='absolute bottom-0 left-0 right-0 border-t bg-background/95 backdrop-blur supports-[backdrop-filter]:bg-background/60'>
139 |           <Button
140 |             variant='ghost'
141 |             onClick={() => setShowClearConfirm(true)}
142 |             className='relative w-full rounded-none h-16 hover:bg-destructive/10'
143 |           >
144 |             <div className='absolute left-4 top-1/2 -translate-y-1/2'>
145 |               <div className='relative'>
146 |                 <Trash2 className='h-5 w-5 text-muted-foreground group-hover:text-destructive transition-colors' />
147 |               </div>
148 |             </div>
149 |             <span className='text-sm font-medium text-muted-foreground group-hover:text-destructive transition-colors'>
150 |               Clear All Reports
151 |             </span>
152 |           </Button>
153 |         </div>
154 |       </SheetContent>
155 |     </Sheet>
156 |   )
157 | }
158 | 


--------------------------------------------------------------------------------
/lib/localStorage-utils.ts:
--------------------------------------------------------------------------------
  1 | import type { Viewport } from '@xyflow/react'
  2 | import type { FlowProject } from '@/hooks/use-flow-projects'
  3 | 
  4 | /**
  5 |  * Gets the current localStorage usage in bytes and as a percentage of the available space
  6 |  */
  7 | export function getLocalStorageUsage(): {
  8 |   usage: number // Size in bytes
  9 |   usagePercent: number // Percentage of available space
 10 |   available: number // Estimated available space in bytes
 11 |   limit: number // Estimated limit in bytes
 12 | } {
 13 |   try {
 14 |     // Current usage
 15 |     let totalSize = 0
 16 |     for (let i = 0; i < localStorage.length; i++) {
 17 |       const key = localStorage.key(i) || ''
 18 |       const value = localStorage.getItem(key) || ''
 19 |       totalSize += (key.length + value.length) * 2 // UTF-16 uses 2 bytes per character
 20 |     }
 21 | 
 22 |     // Estimate of localStorage limit (typically around 5MB)
 23 |     const estimatedLimit = 5 * 1024 * 1024 // 5MB in bytes
 24 | 
 25 |     return {
 26 |       usage: totalSize,
 27 |       usagePercent: (totalSize / estimatedLimit) * 100,
 28 |       available: Math.max(0, estimatedLimit - totalSize),
 29 |       limit: estimatedLimit,
 30 |     }
 31 |   } catch (error) {
 32 |     console.error('Error calculating localStorage usage:', error)
 33 |     return {
 34 |       usage: 0,
 35 |       usagePercent: 0,
 36 |       available: 0,
 37 |       limit: 0,
 38 |     }
 39 |   }
 40 | }
 41 | 
 42 | /**
 43 |  * Formats bytes into a human-readable string (KB, MB)
 44 |  */
 45 | export function formatBytes(bytes: number, decimals = 2): string {
 46 |   if (bytes === 0) return '0 Bytes'
 47 | 
 48 |   const k = 1024
 49 |   const sizes = ['Bytes', 'KB', 'MB']
 50 |   const i = Math.floor(Math.log(bytes) / Math.log(k))
 51 | 
 52 |   return parseFloat((bytes / Math.pow(k, i)).toFixed(decimals)) + ' ' + sizes[i]
 53 | }
 54 | 
 55 | /**
 56 |  * Clears all flow projects from localStorage
 57 |  */
 58 | export function clearAllFlowProjects(): void {
 59 |   localStorage.removeItem('open-deep-research-flow-projects')
 60 |   localStorage.removeItem('open-deep-research-current-project')
 61 | }
 62 | 
 63 | /**
 64 |  * Exports all flow projects as a JSON string
 65 |  */
 66 | export function exportFlowProjects(): string {
 67 |   const projects =
 68 |     localStorage.getItem('open-deep-research-flow-projects') || '[]'
 69 |   return projects
 70 | }
 71 | 
 72 | /**
 73 |  * Imports flow projects from a JSON string
 74 |  * Returns success/failure status
 75 |  */
 76 | export function importFlowProjects(jsonProjects: string): boolean {
 77 |   try {
 78 |     // Validate JSON format
 79 |     const projects = JSON.parse(jsonProjects)
 80 | 
 81 |     if (!Array.isArray(projects)) {
 82 |       throw new Error('Invalid project data structure')
 83 |     }
 84 | 
 85 |     // Check if projects have the required structure
 86 |     for (const project of projects) {
 87 |       if (
 88 |         !project.id ||
 89 |         !project.name ||
 90 |         !project.createdAt ||
 91 |         !project.updatedAt
 92 |       ) {
 93 |         throw new Error('Invalid project data format')
 94 |       }
 95 |     }
 96 | 
 97 |     // Save to localStorage
 98 |     localStorage.setItem('open-deep-research-flow-projects', jsonProjects)
 99 | 
100 |     // If there's a current project ID, verify it still exists in the imported data
101 |     const currentProjectId = localStorage.getItem(
102 |       'open-deep-research-current-project'
103 |     )
104 |     if (currentProjectId) {
105 |       const exists = projects.some((p: any) => p.id === currentProjectId)
106 |       if (!exists) {
107 |         localStorage.removeItem('open-deep-research-current-project')
108 |       }
109 |     }
110 | 
111 |     return true
112 |   } catch (error) {
113 |     console.error('Failed to import projects:', error)
114 |     return false
115 |   }
116 | }
117 | 
118 | /**
119 |  * Creates a debounced function that delays invoking func until after wait milliseconds have elapsed
120 |  */
121 | export function debounce<T extends (...args: any[]) => any>(
122 |   func: T,
123 |   wait: number
124 | ): (...args: Parameters<T>) => void {
125 |   let timeout: NodeJS.Timeout | null = null
126 | 
127 |   return function (...args: Parameters<T>) {
128 |     if (timeout) {
129 |       clearTimeout(timeout)
130 |     }
131 | 
132 |     timeout = setTimeout(() => {
133 |       func(...args)
134 |       timeout = null
135 |     }, wait)
136 |   }
137 | }
138 | 
139 | /**
140 |  * Saves viewport state to the current project
141 |  */
142 | export function saveViewportToProject(
143 |   project: FlowProject | null,
144 |   viewport: Viewport,
145 |   updateProject: (data: Partial<Omit<FlowProject, 'id' | 'createdAt'>>) => void
146 | ): void {
147 |   if (!project) return
148 | 
149 |   try {
150 |     updateProject({
151 |       viewport,
152 |     })
153 |   } catch (error) {
154 |     console.error('Failed to save viewport to project:', error)
155 |   }
156 | }
157 | 
158 | /**
159 |  * Restores viewport state from the current project
160 |  */
161 | export function restoreViewportFromProject(
162 |   project: FlowProject | null,
163 |   setViewport: (viewport: Viewport) => void
164 | ): boolean {
165 |   if (!project?.viewport) return false
166 | 
167 |   try {
168 |     setViewport(project.viewport)
169 |     return true
170 |   } catch (error) {
171 |     console.error('Failed to restore project viewport:', error)
172 |     return false
173 |   }
174 | }
175 | 
176 | /**
177 |  * Creates a debounced viewport save function
178 |  */
179 | export function createDebouncedViewportSave(
180 |   getViewport: () => Viewport,
181 |   project: FlowProject | null,
182 |   updateProject: (data: Partial<Omit<FlowProject, 'id' | 'createdAt'>>) => void,
183 |   debounceTime: number = 500
184 | ): () => void {
185 |   return debounce(() => {
186 |     saveViewportToProject(project, getViewport(), updateProject)
187 |   }, debounceTime)
188 | }
189 | 
190 | /**
191 |  * Saves nodes to the current project
192 |  */
193 | export function saveNodesToProject<NodeType extends { id: string }>(
194 |   project: FlowProject | null,
195 |   nodes: NodeType[],
196 |   updateProject: (data: Partial<Omit<FlowProject, 'id' | 'createdAt'>>) => void
197 | ): void {
198 |   if (!project) return
199 | 
200 |   try {
201 |     updateProject({
202 |       nodes: nodes as any[],
203 |     })
204 |     console.log(`Nodes saved for project: ${project.name}`)
205 |   } catch (error) {
206 |     console.error('Failed to save nodes to project:', error)
207 |   }
208 | }
209 | 
210 | /**
211 |  * Saves edges to the current project
212 |  */
213 | export function saveEdgesToProject<EdgeType extends { id: string }>(
214 |   project: FlowProject | null,
215 |   edges: EdgeType[],
216 |   updateProject: (data: Partial<Omit<FlowProject, 'id' | 'createdAt'>>) => void
217 | ): void {
218 |   if (!project) return
219 | 
220 |   try {
221 |     updateProject({
222 |       edges: edges as any[],
223 |     })
224 |     console.log(`Edges saved for project: ${project.name}`)
225 |   } catch (error) {
226 |     console.error('Failed to save edges to project:', error)
227 |   }
228 | }
229 | 
230 | /**
231 |  * Creates a debounced nodes save function
232 |  */
233 | export function createDebouncedNodesSave<NodeType extends { id: string }>(
234 |   getNodes: () => NodeType[],
235 |   project: FlowProject | null,
236 |   updateProject: (data: Partial<Omit<FlowProject, 'id' | 'createdAt'>>) => void,
237 |   debounceTime: number = 500
238 | ): () => void {
239 |   return debounce(() => {
240 |     saveNodesToProject(project, getNodes(), updateProject)
241 |   }, debounceTime)
242 | }
243 | 
244 | /**
245 |  * Creates a debounced edges save function
246 |  */
247 | export function createDebouncedEdgesSave<EdgeType extends { id: string }>(
248 |   getEdges: () => EdgeType[],
249 |   project: FlowProject | null,
250 |   updateProject: (data: Partial<Omit<FlowProject, 'id' | 'createdAt'>>) => void,
251 |   debounceTime: number = 500
252 | ): () => void {
253 |   return debounce(() => {
254 |     saveEdgesToProject(project, getEdges(), updateProject)
255 |   }, debounceTime)
256 | }
257 | 


--------------------------------------------------------------------------------
/components/ui/dropdown-menu.tsx:
--------------------------------------------------------------------------------
  1 | "use client"
  2 | 
  3 | import * as React from "react"
  4 | import * as DropdownMenuPrimitive from "@radix-ui/react-dropdown-menu"
  5 | import { Check, ChevronRight, Circle } from "lucide-react"
  6 | 
  7 | import { cn } from "@/lib/utils"
  8 | 
  9 | const DropdownMenu = DropdownMenuPrimitive.Root
 10 | 
 11 | const DropdownMenuTrigger = DropdownMenuPrimitive.Trigger
 12 | 
 13 | const DropdownMenuGroup = DropdownMenuPrimitive.Group
 14 | 
 15 | const DropdownMenuPortal = DropdownMenuPrimitive.Portal
 16 | 
 17 | const DropdownMenuSub = DropdownMenuPrimitive.Sub
 18 | 
 19 | const DropdownMenuRadioGroup = DropdownMenuPrimitive.RadioGroup
 20 | 
 21 | const DropdownMenuSubTrigger = React.forwardRef<
 22 |   React.ElementRef<typeof DropdownMenuPrimitive.SubTrigger>,
 23 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.SubTrigger> & {
 24 |     inset?: boolean
 25 |   }
 26 | >(({ className, inset, children, ...props }, ref) => (
 27 |   <DropdownMenuPrimitive.SubTrigger
 28 |     ref={ref}
 29 |     className={cn(
 30 |       "flex cursor-default gap-2 select-none items-center rounded-sm px-2 py-1.5 text-sm outline-none focus:bg-accent data-[state=open]:bg-accent [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0",
 31 |       inset && "pl-8",
 32 |       className
 33 |     )}
 34 |     {...props}
 35 |   >
 36 |     {children}
 37 |     <ChevronRight className="ml-auto" />
 38 |   </DropdownMenuPrimitive.SubTrigger>
 39 | ))
 40 | DropdownMenuSubTrigger.displayName =
 41 |   DropdownMenuPrimitive.SubTrigger.displayName
 42 | 
 43 | const DropdownMenuSubContent = React.forwardRef<
 44 |   React.ElementRef<typeof DropdownMenuPrimitive.SubContent>,
 45 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.SubContent>
 46 | >(({ className, ...props }, ref) => (
 47 |   <DropdownMenuPrimitive.SubContent
 48 |     ref={ref}
 49 |     className={cn(
 50 |       "z-50 min-w-[8rem] overflow-hidden rounded-md border bg-popover p-1 text-popover-foreground shadow-lg data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2",
 51 |       className
 52 |     )}
 53 |     {...props}
 54 |   />
 55 | ))
 56 | DropdownMenuSubContent.displayName =
 57 |   DropdownMenuPrimitive.SubContent.displayName
 58 | 
 59 | const DropdownMenuContent = React.forwardRef<
 60 |   React.ElementRef<typeof DropdownMenuPrimitive.Content>,
 61 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Content>
 62 | >(({ className, sideOffset = 4, ...props }, ref) => (
 63 |   <DropdownMenuPrimitive.Portal>
 64 |     <DropdownMenuPrimitive.Content
 65 |       ref={ref}
 66 |       sideOffset={sideOffset}
 67 |       className={cn(
 68 |         "z-50 min-w-[8rem] overflow-hidden rounded-md border bg-popover p-1 text-popover-foreground shadow-md",
 69 |         "data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2",
 70 |         className
 71 |       )}
 72 |       {...props}
 73 |     />
 74 |   </DropdownMenuPrimitive.Portal>
 75 | ))
 76 | DropdownMenuContent.displayName = DropdownMenuPrimitive.Content.displayName
 77 | 
 78 | const DropdownMenuItem = React.forwardRef<
 79 |   React.ElementRef<typeof DropdownMenuPrimitive.Item>,
 80 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Item> & {
 81 |     inset?: boolean
 82 |   }
 83 | >(({ className, inset, ...props }, ref) => (
 84 |   <DropdownMenuPrimitive.Item
 85 |     ref={ref}
 86 |     className={cn(
 87 |       "relative flex cursor-default select-none items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none transition-colors focus:bg-accent focus:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50 [&>svg]:size-4 [&>svg]:shrink-0",
 88 |       inset && "pl-8",
 89 |       className
 90 |     )}
 91 |     {...props}
 92 |   />
 93 | ))
 94 | DropdownMenuItem.displayName = DropdownMenuPrimitive.Item.displayName
 95 | 
 96 | const DropdownMenuCheckboxItem = React.forwardRef<
 97 |   React.ElementRef<typeof DropdownMenuPrimitive.CheckboxItem>,
 98 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.CheckboxItem>
 99 | >(({ className, children, checked, ...props }, ref) => (
100 |   <DropdownMenuPrimitive.CheckboxItem
101 |     ref={ref}
102 |     className={cn(
103 |       "relative flex cursor-default select-none items-center rounded-sm py-1.5 pl-8 pr-2 text-sm outline-none transition-colors focus:bg-accent focus:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50",
104 |       className
105 |     )}
106 |     checked={checked}
107 |     {...props}
108 |   >
109 |     <span className="absolute left-2 flex h-3.5 w-3.5 items-center justify-center">
110 |       <DropdownMenuPrimitive.ItemIndicator>
111 |         <Check className="h-4 w-4" />
112 |       </DropdownMenuPrimitive.ItemIndicator>
113 |     </span>
114 |     {children}
115 |   </DropdownMenuPrimitive.CheckboxItem>
116 | ))
117 | DropdownMenuCheckboxItem.displayName =
118 |   DropdownMenuPrimitive.CheckboxItem.displayName
119 | 
120 | const DropdownMenuRadioItem = React.forwardRef<
121 |   React.ElementRef<typeof DropdownMenuPrimitive.RadioItem>,
122 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.RadioItem>
123 | >(({ className, children, ...props }, ref) => (
124 |   <DropdownMenuPrimitive.RadioItem
125 |     ref={ref}
126 |     className={cn(
127 |       "relative flex cursor-default select-none items-center rounded-sm py-1.5 pl-8 pr-2 text-sm outline-none transition-colors focus:bg-accent focus:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50",
128 |       className
129 |     )}
130 |     {...props}
131 |   >
132 |     <span className="absolute left-2 flex h-3.5 w-3.5 items-center justify-center">
133 |       <DropdownMenuPrimitive.ItemIndicator>
134 |         <Circle className="h-2 w-2 fill-current" />
135 |       </DropdownMenuPrimitive.ItemIndicator>
136 |     </span>
137 |     {children}
138 |   </DropdownMenuPrimitive.RadioItem>
139 | ))
140 | DropdownMenuRadioItem.displayName = DropdownMenuPrimitive.RadioItem.displayName
141 | 
142 | const DropdownMenuLabel = React.forwardRef<
143 |   React.ElementRef<typeof DropdownMenuPrimitive.Label>,
144 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Label> & {
145 |     inset?: boolean
146 |   }
147 | >(({ className, inset, ...props }, ref) => (
148 |   <DropdownMenuPrimitive.Label
149 |     ref={ref}
150 |     className={cn(
151 |       "px-2 py-1.5 text-sm font-semibold",
152 |       inset && "pl-8",
153 |       className
154 |     )}
155 |     {...props}
156 |   />
157 | ))
158 | DropdownMenuLabel.displayName = DropdownMenuPrimitive.Label.displayName
159 | 
160 | const DropdownMenuSeparator = React.forwardRef<
161 |   React.ElementRef<typeof DropdownMenuPrimitive.Separator>,
162 |   React.ComponentPropsWithoutRef<typeof DropdownMenuPrimitive.Separator>
163 | >(({ className, ...props }, ref) => (
164 |   <DropdownMenuPrimitive.Separator
165 |     ref={ref}
166 |     className={cn("-mx-1 my-1 h-px bg-muted", className)}
167 |     {...props}
168 |   />
169 | ))
170 | DropdownMenuSeparator.displayName = DropdownMenuPrimitive.Separator.displayName
171 | 
172 | const DropdownMenuShortcut = ({
173 |   className,
174 |   ...props
175 | }: React.HTMLAttributes<HTMLSpanElement>) => {
176 |   return (
177 |     <span
178 |       className={cn("ml-auto text-xs tracking-widest opacity-60", className)}
179 |       {...props}
180 |     />
181 |   )
182 | }
183 | DropdownMenuShortcut.displayName = "DropdownMenuShortcut"
184 | 
185 | export {
186 |   DropdownMenu,
187 |   DropdownMenuTrigger,
188 |   DropdownMenuContent,
189 |   DropdownMenuItem,
190 |   DropdownMenuCheckboxItem,
191 |   DropdownMenuRadioItem,
192 |   DropdownMenuLabel,
193 |   DropdownMenuSeparator,
194 |   DropdownMenuShortcut,
195 |   DropdownMenuGroup,
196 |   DropdownMenuPortal,
197 |   DropdownMenuSub,
198 |   DropdownMenuSubContent,
199 |   DropdownMenuSubTrigger,
200 |   DropdownMenuRadioGroup,
201 | }
202 | 


--------------------------------------------------------------------------------
/lib/documents.ts:
--------------------------------------------------------------------------------
  1 | import { type Report } from '@/types'
  2 | import {
  3 |   Document,
  4 |   Paragraph,
  5 |   TextRun,
  6 |   AlignmentType,
  7 |   Packer,
  8 |   Header,
  9 |   Footer,
 10 |   PageNumber,
 11 | } from 'docx'
 12 | import jsPDF from 'jspdf'
 13 | import MarkdownIt from 'markdown-it'
 14 | 
 15 | const md = new MarkdownIt()
 16 | 
 17 | export async function generateDocx(report: Report): Promise<Buffer> {
 18 |   try {
 19 |     console.log(
 20 |       'Starting DOCX generation with report:',
 21 |       JSON.stringify(report, null, 2)
 22 |     )
 23 | 
 24 |     const doc = new Document({
 25 |       sections: [
 26 |         {
 27 |           properties: {},
 28 |           headers: {
 29 |             default: new Header({
 30 |               children: [
 31 |                 new Paragraph({
 32 |                   children: [
 33 |                     new TextRun({
 34 |                       text: report.title || 'Untitled Report',
 35 |                       size: 48,
 36 |                       bold: true,
 37 |                     }),
 38 |                   ],
 39 |                   alignment: AlignmentType.CENTER,
 40 |                   spacing: { after: 800 },
 41 |                 }),
 42 |               ],
 43 |             }),
 44 |           },
 45 |           footers: {
 46 |             default: new Footer({
 47 |               children: [
 48 |                 new Paragraph({
 49 |                   children: [
 50 |                     new TextRun('Page '),
 51 |                     new TextRun({
 52 |                       children: [PageNumber.CURRENT],
 53 |                     }),
 54 |                     new TextRun(' of '),
 55 |                     new TextRun({
 56 |                       children: [PageNumber.TOTAL_PAGES],
 57 |                     }),
 58 |                   ],
 59 |                   alignment: AlignmentType.CENTER,
 60 |                 }),
 61 |               ],
 62 |             }),
 63 |           },
 64 |           children: [
 65 |             // Summary with increased spacing
 66 |             new Paragraph({
 67 |               children: [
 68 |                 new TextRun({
 69 |                   text: report.summary || '',
 70 |                   size: 24,
 71 |                 }),
 72 |               ],
 73 |               spacing: { before: 800, after: 800 },
 74 |               alignment: AlignmentType.JUSTIFIED,
 75 |             }),
 76 |             // Sections with increased spacing
 77 |             ...report.sections.flatMap((section) => [
 78 |               new Paragraph({
 79 |                 children: [
 80 |                   new TextRun({
 81 |                     text: section.title || '',
 82 |                     size: 32,
 83 |                     bold: true,
 84 |                   }),
 85 |                 ],
 86 |                 spacing: { before: 800, after: 400 },
 87 |                 alignment: AlignmentType.LEFT,
 88 |               }),
 89 |               new Paragraph({
 90 |                 children: [
 91 |                   new TextRun({
 92 |                     text: section.content || '',
 93 |                     size: 24,
 94 |                   }),
 95 |                 ],
 96 |                 spacing: { before: 400, after: 800 },
 97 |                 alignment: AlignmentType.JUSTIFIED,
 98 |               }),
 99 |             ]),
100 |           ],
101 |         },
102 |       ],
103 |     })
104 | 
105 |     console.log('Document instance created')
106 | 
107 |     try {
108 |       console.log('Starting document packing')
109 |       const buffer = await Packer.toBuffer(doc)
110 |       console.log('Document packed successfully, buffer size:', buffer.length)
111 |       return buffer
112 |     } catch (packError) {
113 |       console.error('Error packing document:', packError)
114 |       throw packError
115 |     }
116 |   } catch (error) {
117 |     console.error('Error in generateDocx:', error)
118 |     if (error instanceof Error) {
119 |       console.error('Error details:', {
120 |         message: error.message,
121 |         stack: error.stack,
122 |         name: error.name,
123 |       })
124 |     }
125 |     throw new Error(
126 |       `Failed to generate DOCX: ${
127 |         error instanceof Error ? error.message : 'Unknown error'
128 |       }`
129 |     )
130 |   }
131 | }
132 | 
133 | export function generatePdf(report: Report): Buffer {
134 |   try {
135 |     const doc = new jsPDF({
136 |       orientation: 'portrait',
137 |       unit: 'mm',
138 |       format: 'a4',
139 |     })
140 | 
141 |     const pageWidth = doc.internal.pageSize.width
142 |     const margin = 20
143 |     const contentWidth = pageWidth - 2 * margin
144 | 
145 |     // Helper function to add text with proper line breaks and page management
146 |     const addText = (
147 |       text: string,
148 |       y: number,
149 |       fontSize: number,
150 |       isBold: boolean = false,
151 |       isJustified: boolean = false,
152 |       isHTML: boolean = false
153 |     ): number => {
154 |       doc.setFontSize(fontSize)
155 |       doc.setFont('helvetica', isBold ? 'bold' : 'normal')
156 | 
157 |       // If the text contains markdown, convert it to plain text
158 |       let processedText = text
159 |       if (isHTML) {
160 |         // Remove HTML tags but preserve line breaks
161 |         processedText = text
162 |           .replace(/<br\s*\/?>/gi, '\n')
163 |           .replace(/<\/p>/gi, '\n')
164 |           .replace(/<[^>]*>/g, '')
165 |           // Handle markdown-style bold
166 |           .replace(/\*\*(.*?)\*\*/g, (_, p1) => {
167 |             doc.setFont('helvetica', 'bold')
168 |             const result = p1
169 |             doc.setFont('helvetica', isBold ? 'bold' : 'normal')
170 |             return result
171 |           })
172 |           // Handle markdown-style italic
173 |           .replace(/\*(.*?)\*/g, (_, p1) => {
174 |             doc.setFont('helvetica', 'italic')
175 |             const result = p1
176 |             doc.setFont('helvetica', isBold ? 'bold' : 'normal')
177 |             return result
178 |           })
179 |       }
180 | 
181 |       const lines = doc.splitTextToSize(processedText, contentWidth)
182 |       const lineHeight = fontSize * 0.3527 // Convert pt to mm
183 | 
184 |       lines.forEach((line: string) => {
185 |         if (y > 270) {
186 |           doc.addPage()
187 |           y = margin
188 |         }
189 | 
190 |         // Handle bullet points
191 |         if (line.trim().startsWith('•') || line.trim().startsWith('-')) {
192 |           doc.text('•', margin, y)
193 |           doc.text(line.trim().substring(1), margin + 5, y, {
194 |             align: isJustified ? 'justify' : 'left',
195 |             maxWidth: contentWidth - 5,
196 |           })
197 |         } else {
198 |           doc.text(line, margin, y, {
199 |             align: isJustified ? 'justify' : 'left',
200 |             maxWidth: contentWidth,
201 |           })
202 |         }
203 |         y += lineHeight + 1 // 1mm extra spacing between lines
204 |       })
205 | 
206 |       return y + lineHeight // Return new Y position
207 |     }
208 | 
209 |     // Start position
210 |     let currentY = margin
211 | 
212 |     // Title
213 |     currentY = addText(report.title, currentY, 24, true)
214 |     currentY += 5 // Reduced from 10 to 5
215 | 
216 |     // Convert markdown to HTML for processing
217 |     const summaryHtml = md.render(report.summary)
218 |     currentY = addText(summaryHtml, currentY, 12, false, true, true)
219 |     currentY += 3 // Reduced from 10 to 3
220 | 
221 |     // Sections
222 |     report.sections.forEach((section) => {
223 |       currentY += 2 // Reduced from 5 to 2
224 | 
225 |       // Section title
226 |       currentY = addText(section.title, currentY, 16, true)
227 |       currentY += 2 // Reduced from 5 to 2
228 | 
229 |       // Convert markdown to HTML for processing
230 |       const contentHtml = md.render(section.content)
231 |       currentY = addText(contentHtml, currentY, 12, false, true, true)
232 |       currentY += 2 // Reduced from 5 to 2
233 |     })
234 | 
235 |     // Add page numbers
236 |     const pageCount = doc.internal.pages.length - 1
237 |     for (let i = 1; i <= pageCount; i++) {
238 |       doc.setPage(i)
239 |       doc.setFontSize(10)
240 |       doc.setFont('helvetica', 'normal')
241 |       doc.text(`Page ${i} of ${pageCount}`, pageWidth / 2, 285, {
242 |         align: 'center',
243 |       })
244 |     }
245 | 
246 |     return Buffer.from(doc.output('arraybuffer'))
247 |   } catch (error) {
248 |     console.error('Error generating PDF:', error)
249 |     throw new Error('Failed to generate PDF')
250 |   }
251 | }
252 | 


--------------------------------------------------------------------------------
/components/flow/project-actions.tsx:
--------------------------------------------------------------------------------
  1 | import { useState, useRef } from 'react'
  2 | import { Download, Upload, Info, Database, AlertTriangle } from 'lucide-react'
  3 | import { Button } from '@/components/ui/button'
  4 | import {
  5 |   Dialog,
  6 |   DialogContent,
  7 |   DialogDescription,
  8 |   DialogFooter,
  9 |   DialogHeader,
 10 |   DialogTitle,
 11 | } from '@/components/ui/dialog'
 12 | import {
 13 |   DropdownMenu,
 14 |   DropdownMenuContent,
 15 |   DropdownMenuItem,
 16 |   DropdownMenuSeparator,
 17 |   DropdownMenuTrigger,
 18 | } from '@/components/ui/dropdown-menu'
 19 | import {
 20 |   Tooltip,
 21 |   TooltipContent,
 22 |   TooltipProvider,
 23 |   TooltipTrigger,
 24 | } from '@/components/ui/tooltip'
 25 | import { useToast } from '@/hooks/use-toast'
 26 | 
 27 | interface ProjectActionsProps {
 28 |   exportProjects: () => string
 29 |   importProjects: (jsonData: string) => boolean
 30 |   storageInfo: {
 31 |     usage: number
 32 |     usagePercent: number
 33 |     available: number
 34 |     limit: number
 35 |     formattedUsage: string
 36 |     formattedAvailable: string
 37 |   }
 38 |   refreshStorageInfo: () => void
 39 | }
 40 | 
 41 | export function ProjectActions({
 42 |   exportProjects,
 43 |   importProjects,
 44 |   storageInfo,
 45 |   refreshStorageInfo,
 46 | }: ProjectActionsProps) {
 47 |   const [isStorageInfoOpen, setIsStorageInfoOpen] = useState(false)
 48 |   const fileInputRef = useRef<HTMLInputElement>(null)
 49 |   const { toast } = useToast()
 50 | 
 51 |   const handleExport = () => {
 52 |     try {
 53 |       const jsonData = exportProjects()
 54 |       if (jsonData === '[]') {
 55 |         toast({
 56 |           title: 'No projects to export',
 57 |           description: "You don't have any projects saved to export.",
 58 |           variant: 'destructive',
 59 |         })
 60 |         return
 61 |       }
 62 | 
 63 |       const blob = new Blob([jsonData], { type: 'application/json' })
 64 |       const url = URL.createObjectURL(blob)
 65 |       const a = document.createElement('a')
 66 |       a.href = url
 67 |       a.download = `open-deep-research-projects-${
 68 |         new Date().toISOString().split('T')[0]
 69 |       }.json`
 70 |       document.body.appendChild(a)
 71 |       a.click()
 72 |       document.body.removeChild(a)
 73 |       URL.revokeObjectURL(url)
 74 | 
 75 |       toast({
 76 |         title: 'Projects exported',
 77 |         description: 'Your projects have been exported successfully.',
 78 |       })
 79 |     } catch (error) {
 80 |       console.error('Export error:', error)
 81 |       toast({
 82 |         title: 'Export failed',
 83 |         description: 'There was an error exporting your projects.',
 84 |         variant: 'destructive',
 85 |       })
 86 |     }
 87 |   }
 88 | 
 89 |   const handleImportClick = () => {
 90 |     if (fileInputRef.current) {
 91 |       fileInputRef.current.click()
 92 |     }
 93 |   }
 94 | 
 95 |   const handleFileChange = (e: React.ChangeEvent<HTMLInputElement>) => {
 96 |     const file = e.target.files?.[0]
 97 |     if (!file) return
 98 | 
 99 |     const reader = new FileReader()
100 |     reader.onload = (event) => {
101 |       try {
102 |         const jsonData = event.target?.result as string
103 |         const success = importProjects(jsonData)
104 | 
105 |         if (success) {
106 |           toast({
107 |             title: 'Projects imported',
108 |             description: 'Your projects have been imported successfully.',
109 |           })
110 |           refreshStorageInfo()
111 |         } else {
112 |           toast({
113 |             title: 'Import failed',
114 |             description: 'The file format is invalid or corrupted.',
115 |             variant: 'destructive',
116 |           })
117 |         }
118 |       } catch (error) {
119 |         console.error('Import error:', error)
120 |         toast({
121 |           title: 'Import failed',
122 |           description: 'There was an error importing your projects.',
123 |           variant: 'destructive',
124 |         })
125 |       }
126 |     }
127 | 
128 |     reader.readAsText(file)
129 | 
130 |     // Reset the input so the same file can be selected again
131 |     if (fileInputRef.current) {
132 |       fileInputRef.current.value = ''
133 |     }
134 |   }
135 | 
136 |   const getStorageStatusColor = (percent: number) => {
137 |     if (percent < 50) return 'bg-green-500'
138 |     if (percent < 80) return 'bg-yellow-500'
139 |     return 'bg-red-500'
140 |   }
141 | 
142 |   return (
143 |     <>
144 |       <input
145 |         type='file'
146 |         ref={fileInputRef}
147 |         onChange={handleFileChange}
148 |         accept='.json'
149 |         className='hidden'
150 |       />
151 | 
152 |       <TooltipProvider>
153 |         <DropdownMenu>
154 |           <Tooltip>
155 |             <TooltipTrigger asChild>
156 |               <DropdownMenuTrigger asChild>
157 |                 <Button variant='outline' size='sm' className='h-9 w-9'>
158 |                   <Database className='h-4 w-4' />
159 |                 </Button>
160 |               </DropdownMenuTrigger>
161 |             </TooltipTrigger>
162 |             <TooltipContent>
163 |               <p>Project Data & Storage</p>
164 |             </TooltipContent>
165 |           </Tooltip>
166 | 
167 |           <DropdownMenuContent align='end' className='w-56'>
168 |             <div className='px-2 py-1.5 text-sm font-medium text-gray-500'>
169 |               Storage & Backup
170 |             </div>
171 |             <DropdownMenuSeparator />
172 | 
173 |             <div className='px-3 py-2'>
174 |               <div className='text-xs text-gray-500 mb-1 flex justify-between'>
175 |                 <span>Storage Usage</span>
176 |                 <span>{storageInfo.formattedUsage} / 5 MB</span>
177 |               </div>
178 |               <div className='w-full h-1.5 bg-gray-200 rounded-full overflow-hidden'>
179 |                 <div
180 |                   className={`h-full ${getStorageStatusColor(
181 |                     storageInfo.usagePercent
182 |                   )}`}
183 |                   style={{
184 |                     width: `${Math.min(100, storageInfo.usagePercent)}%`,
185 |                   }}
186 |                 />
187 |               </div>
188 |             </div>
189 | 
190 |             <DropdownMenuSeparator />
191 |             <DropdownMenuItem onClick={handleExport} className='gap-2'>
192 |               <Download className='h-4 w-4' />
193 |               <div>
194 |                 <div className='text-sm'>Export Projects</div>
195 |                 <div className='text-xs text-gray-500'>
196 |                   Backup your projects as JSON
197 |                 </div>
198 |               </div>
199 |             </DropdownMenuItem>
200 |             <DropdownMenuItem onClick={handleImportClick} className='gap-2'>
201 |               <Upload className='h-4 w-4' />
202 |               <div>
203 |                 <div className='text-sm'>Import Projects</div>
204 |                 <div className='text-xs text-gray-500'>Restore from backup</div>
205 |               </div>
206 |             </DropdownMenuItem>
207 |             <DropdownMenuSeparator />
208 |             <DropdownMenuItem
209 |               onClick={() => setIsStorageInfoOpen(true)}
210 |               className='gap-2'
211 |             >
212 |               <Info className='h-4 w-4' />
213 |               <div className='text-sm'>Storage Details</div>
214 |             </DropdownMenuItem>
215 |           </DropdownMenuContent>
216 |         </DropdownMenu>
217 |       </TooltipProvider>
218 | 
219 |       <Dialog open={isStorageInfoOpen} onOpenChange={setIsStorageInfoOpen}>
220 |         <DialogContent>
221 |           <DialogHeader>
222 |             <DialogTitle>localStorage Usage</DialogTitle>
223 |             <DialogDescription>
224 |               Your research projects are stored in your browser&apos;s localStorage
225 |             </DialogDescription>
226 |           </DialogHeader>
227 | 
228 |           <div className='mt-4 space-y-4'>
229 |             <div>
230 |               <div className='flex justify-between text-sm mb-1'>
231 |                 <span>Storage Usage</span>
232 |                 <span>{storageInfo.formattedUsage} / 5 MB</span>
233 |               </div>
234 |               <div className='w-full h-2 bg-gray-200 rounded-full overflow-hidden'>
235 |                 <div
236 |                   className={`h-full ${getStorageStatusColor(
237 |                     storageInfo.usagePercent
238 |                   )}`}
239 |                   style={{
240 |                     width: `${Math.min(100, storageInfo.usagePercent)}%`,
241 |                   }}
242 |                 />
243 |               </div>
244 |             </div>
245 | 
246 |             {storageInfo.usagePercent > 80 && (
247 |               <div className='flex items-start gap-2 p-3 bg-amber-50 border border-amber-200 rounded-md'>
248 |                 <AlertTriangle className='h-5 w-5 text-amber-500 flex-shrink-0 mt-0.5' />
249 |                 <div className='text-sm text-amber-800'>
250 |                   <p className='font-medium'>Storage is running low</p>
251 |                   <p>
252 |                     Consider exporting and deleting some projects to free up
253 |                     space.
254 |                   </p>
255 |                 </div>
256 |               </div>
257 |             )}
258 | 
259 |             <div className='text-sm space-y-2'>
260 |               <p>
261 |                 <strong>Available Space:</strong>{' '}
262 |                 {storageInfo.formattedAvailable}
263 |               </p>
264 |               <p>
265 |                 <strong>Note:</strong> localStorage has a limit of approximately
266 |                 5MB per domain. Data is stored only in this browser and will be
267 |                 lost if you clear browser data.
268 |               </p>
269 |             </div>
270 |           </div>
271 | 
272 |           <DialogFooter>
273 |             <Button onClick={() => setIsStorageInfoOpen(false)}>Close</Button>
274 |           </DialogFooter>
275 |         </DialogContent>
276 |       </Dialog>
277 |     </>
278 |   )
279 | }
280 | 


--------------------------------------------------------------------------------
/app/api/search/route.ts:
--------------------------------------------------------------------------------
  1 | import { NextResponse } from 'next/server'
  2 | import { searchRatelimit } from '@/lib/redis'
  3 | import { CONFIG } from '@/lib/config'
  4 | 
  5 | const BING_ENDPOINT = 'https://api.bing.microsoft.com/v7.0/search'
  6 | const GOOGLE_ENDPOINT = 'https://customsearch.googleapis.com/customsearch/v1'
  7 | const EXA_ENDPOINT = 'https://api.exa.ai/search'
  8 | 
  9 | type TimeFilter = '24h' | 'week' | 'month' | 'year' | 'all'
 10 | 
 11 | function getBingFreshness(timeFilter: TimeFilter): string {
 12 |   switch (timeFilter) {
 13 |     case '24h':
 14 |       return 'Day'
 15 |     case 'week':
 16 |       return 'Week'
 17 |     case 'month':
 18 |       return 'Month'
 19 |     case 'year':
 20 |       return 'Year'
 21 |     default:
 22 |       return ''
 23 |   }
 24 | }
 25 | 
 26 | function getGoogleDateRestrict(timeFilter: TimeFilter): string | null {
 27 |   // Map our timeFilter to Google's dateRestrict values:
 28 |   // d1 for 24h, w1 for week, m1 for month, y1 for year
 29 |   switch (timeFilter) {
 30 |     case '24h':
 31 |       return 'd1'
 32 |     case 'week':
 33 |       return 'w1'
 34 |     case 'month':
 35 |       return 'm1'
 36 |     case 'year':
 37 |       return 'y1'
 38 |     default:
 39 |       return null
 40 |   }
 41 | }
 42 | 
 43 | export async function POST(request: Request) {
 44 |   try {
 45 |     const body = await request.json()
 46 |     const {
 47 |       query,
 48 |       timeFilter = 'all',
 49 |       provider = CONFIG.search.provider,
 50 |       isTestQuery = false,
 51 |     } = body
 52 | 
 53 |     if (!query) {
 54 |       return NextResponse.json(
 55 |         { error: 'Query parameter is required' },
 56 |         { status: 400 }
 57 |       )
 58 |     }
 59 | 
 60 |     // Return dummy results for test queries
 61 |     if (query.toLowerCase() === 'test' || isTestQuery) {
 62 |       return NextResponse.json({
 63 |         webPages: {
 64 |           value: [
 65 |             {
 66 |               id: 'test-1',
 67 |               url: 'https://example.com/test-1',
 68 |               name: 'Test Result 1',
 69 |               snippet:
 70 |                 'This is a test search result for testing purposes. It contains some sample text about research and analysis.',
 71 |             },
 72 |             {
 73 |               id: 'test-2',
 74 |               url: 'https://example.com/test-2',
 75 |               name: 'Test Result 2',
 76 |               snippet:
 77 |                 'Another test result with different content. This one discusses methodology and data collection.',
 78 |             },
 79 |             {
 80 |               id: 'test-3',
 81 |               url: 'https://example.com/test-3',
 82 |               name: 'Test Result 3',
 83 |               snippet:
 84 |                 'A third test result focusing on academic research and scientific papers.',
 85 |             },
 86 |           ],
 87 |         },
 88 |       })
 89 |     }
 90 | 
 91 |     // Only check rate limit if enabled
 92 |     if (CONFIG.rateLimits.enabled) {
 93 |       const { success } = await searchRatelimit.limit(query)
 94 |       if (!success) {
 95 |         return NextResponse.json(
 96 |           {
 97 |             error:
 98 |               'Too many requests. Please wait a moment before trying again.',
 99 |           },
100 |           { status: 429 }
101 |         )
102 |       }
103 |     }
104 | 
105 |     if (provider === 'exa') {
106 |       const exaApiKey = process.env.EXA_API_KEY
107 |       if (!exaApiKey) {
108 |         return NextResponse.json(
109 |           {
110 |             error:
111 |               'Exa search API is not properly configured. Please check your environment variables.',
112 |           },
113 |           { status: 500 }
114 |         )
115 |       }
116 | 
117 |       try {
118 |         const exaResponse = await fetch(EXA_ENDPOINT, {
119 |           method: 'POST',
120 |           headers: {
121 |             'Content-Type': 'application/json',
122 |             Authorization: `Bearer ${exaApiKey}`,
123 |           },
124 |           body: JSON.stringify({
125 |             query,
126 |             type: 'auto',
127 |             numResults: CONFIG.search.resultsPerPage,
128 |             contents: {
129 |               text: {
130 |                 maxCharacters: 500,
131 |               },
132 |             },
133 |           }),
134 |         })
135 | 
136 |         if (!exaResponse.ok) {
137 |           if (exaResponse.status === 429) {
138 |             return NextResponse.json(
139 |               {
140 |                 error: 'Rate limit exceeded. Please try again later.',
141 |               },
142 |               { status: 429 }
143 |             )
144 |           }
145 |           throw new Error(`Exa API error: ${exaResponse.status}`)
146 |         }
147 | 
148 |         const response = await exaResponse.json()
149 | 
150 |         if (!response?.results) {
151 |           throw new Error('Unexpected Exa API response format')
152 |         }
153 | 
154 |         // Transform Exa results to match our format
155 |         const transformedResults = {
156 |           webPages: {
157 |             value: response.results.map((item: any) => ({
158 |               id: item.id || item.url,
159 |               url: item.url,
160 |               name: item.title || 'Untitled',
161 |               snippet: item.text || '',
162 |               publishedDate: item.publishedDate || undefined,
163 |               author: item.author || undefined,
164 |               image: item.image || undefined,
165 |               favicon: item.favicon || undefined,
166 |               score: item.score || undefined,
167 |             })),
168 |           },
169 |         }
170 | 
171 |         return NextResponse.json(transformedResults)
172 |       } catch (error: any) {
173 |         console.error('Exa search error:', error)
174 |         return NextResponse.json(
175 |           {
176 |             error: 'Failed to fetch search results from Exa.',
177 |           },
178 |           { status: 500 }
179 |         )
180 |       }
181 |     } else if (provider === 'google') {
182 |       // Ensure required Google API variables are available.
183 |       const googleApiKey = process.env.GOOGLE_SEARCH_API_KEY
184 |       const googleCx = process.env.GOOGLE_SEARCH_CX
185 |       if (!googleApiKey || !googleCx) {
186 |         return NextResponse.json(
187 |           {
188 |             error:
189 |               'Google search API is not properly configured. Please check your environment variables.',
190 |           },
191 |           { status: 500 }
192 |         )
193 |       }
194 | 
195 |       const params = new URLSearchParams({
196 |         q: query,
197 |         key: googleApiKey,
198 |         cx: googleCx,
199 |         num: CONFIG.search.resultsPerPage.toString(),
200 |       })
201 | 
202 |       // Add Google's dateRestrict parameter if a time filter is applied
203 |       const dateRestrict = getGoogleDateRestrict(timeFilter as TimeFilter)
204 |       if (dateRestrict) {
205 |         params.append('dateRestrict', dateRestrict)
206 |       }
207 | 
208 |       // Set safe search parameter based on config
209 |       params.append('safe', CONFIG.search.safeSearch.google)
210 | 
211 |       const googleResponse = await fetch(
212 |         `${GOOGLE_ENDPOINT}?${params.toString()}`
213 |       )
214 | 
215 |       if (!googleResponse.ok) {
216 |         const errorData = await googleResponse.json().catch(() => null)
217 | 
218 |         // Check for quota exceeded error
219 |         if (errorData?.error?.message?.includes('Quota exceeded')) {
220 |           return NextResponse.json(
221 |             {
222 |               error:
223 |                 'Daily search limit reached. Please try again tomorrow or contact support for increased limits.',
224 |             },
225 |             { status: 429 }
226 |           )
227 |         }
228 | 
229 |         return NextResponse.json(
230 |           {
231 |             error:
232 |               'An error occurred while fetching search results. Please try again later.',
233 |           },
234 |           { status: googleResponse.status }
235 |         )
236 |       }
237 | 
238 |       const data = await googleResponse.json()
239 | 
240 |       // Transform Google search results to match our format
241 |       const transformedResults = {
242 |         webPages: {
243 |           value:
244 |             data.items?.map((item: any) => ({
245 |               id: item.cacheId || item.link,
246 |               url: item.link,
247 |               name: item.title,
248 |               snippet: item.snippet,
249 |             })) || [],
250 |         },
251 |       }
252 | 
253 |       return NextResponse.json(transformedResults)
254 |     } else {
255 |       // Default to Bing search
256 |       const subscriptionKey = process.env.AZURE_SUB_KEY
257 |       if (!subscriptionKey) {
258 |         return NextResponse.json(
259 |           {
260 |             error:
261 |               'Search API is not properly configured. Please check your environment variables.',
262 |           },
263 |           { status: 500 }
264 |         )
265 |       }
266 | 
267 |       const params = new URLSearchParams({
268 |         q: query,
269 |         count: CONFIG.search.resultsPerPage.toString(),
270 |         mkt: CONFIG.search.market,
271 |         safeSearch: CONFIG.search.safeSearch.bing,
272 |         textFormat: 'HTML',
273 |         textDecorations: 'true',
274 |       })
275 | 
276 |       // Add freshness parameter for Bing if a time filter is applied
277 |       const freshness = getBingFreshness(timeFilter as TimeFilter)
278 |       if (freshness) {
279 |         params.append('freshness', freshness)
280 |       }
281 | 
282 |       const bingResponse = await fetch(
283 |         `${BING_ENDPOINT}?${params.toString()}`,
284 |         {
285 |           headers: {
286 |             'Ocp-Apim-Subscription-Key': subscriptionKey,
287 |             'Accept-Language': 'en-US',
288 |           },
289 |         }
290 |       )
291 | 
292 |       if (!bingResponse.ok) {
293 |         if (bingResponse.status === 403) {
294 |           console.error('Bing Search API 403 Error:', {
295 |             status: bingResponse.status,
296 |             headers: Object.fromEntries(bingResponse.headers.entries()),
297 |             query,
298 |             timeFilter,
299 |           })
300 | 
301 |           try {
302 |             const errorBody = await bingResponse.json()
303 |             console.error('Bing Error Response:', errorBody)
304 |           } catch (e) {
305 |             console.error('Could not parse Bing error response', e)
306 |           }
307 | 
308 |           return NextResponse.json(
309 |             {
310 |               error:
311 |                 'Monthly search quota exceeded. Please try again next month or contact support for increased limits.',
312 |             },
313 |             { status: 403 }
314 |           )
315 |         }
316 |         const errorData = await bingResponse.json().catch(() => null)
317 |         return NextResponse.json(
318 |           {
319 |             error:
320 |               errorData?.message ||
321 |               `Search API returned error ${bingResponse.status}`,
322 |           },
323 |           { status: bingResponse.status }
324 |         )
325 |       }
326 | 
327 |       const data = await bingResponse.json()
328 |       return NextResponse.json(data)
329 |     }
330 |   } catch (error) {
331 |     console.error('Search API error:', error)
332 |     return NextResponse.json(
333 |       {
334 |         error:
335 |           error instanceof Error
336 |             ? error.message
337 |             : 'An unexpected error occurred while fetching search results',
338 |       },
339 |       { status: 500 }
340 |     )
341 |   }
342 | }
343 | 


--------------------------------------------------------------------------------
/components/flow/project-selector.tsx:
--------------------------------------------------------------------------------
  1 | import { useState } from 'react'
  2 | import { PlusCircle, FolderOpen, Trash2, Edit, Check, X } from 'lucide-react'
  3 | import { Button } from '@/components/ui/button'
  4 | import { Input } from '@/components/ui/input'
  5 | import {
  6 |   DropdownMenu,
  7 |   DropdownMenuContent,
  8 |   DropdownMenuItem,
  9 |   DropdownMenuSeparator,
 10 |   DropdownMenuTrigger,
 11 | } from '@/components/ui/dropdown-menu'
 12 | import {
 13 |   Dialog,
 14 |   DialogContent,
 15 |   DialogDescription,
 16 |   DialogFooter,
 17 |   DialogHeader,
 18 |   DialogTitle,
 19 | } from '@/components/ui/dialog'
 20 | import {
 21 |   Tooltip,
 22 |   TooltipContent,
 23 |   TooltipProvider,
 24 |   TooltipTrigger,
 25 | } from '@/components/ui/tooltip'
 26 | import { useToast } from '@/hooks/use-toast'
 27 | import type { FlowProject } from '@/hooks/use-flow-projects'
 28 | 
 29 | interface ProjectSelectorProps {
 30 |   projects: FlowProject[]
 31 |   currentProject: FlowProject | null
 32 |   onSelectProject: (project: FlowProject) => void
 33 |   onCreateProject: (name: string) => void
 34 |   onDeleteProject: (id: string) => void
 35 |   onRenameProject: (id: string, name: string) => void
 36 | }
 37 | 
 38 | export function ProjectSelector({
 39 |   projects,
 40 |   currentProject,
 41 |   onSelectProject,
 42 |   onCreateProject,
 43 |   onDeleteProject,
 44 |   onRenameProject,
 45 | }: ProjectSelectorProps) {
 46 |   const [isCreateDialogOpen, setIsCreateDialogOpen] = useState(false)
 47 |   const [isDeleteDialogOpen, setIsDeleteDialogOpen] = useState(false)
 48 |   const [newProjectName, setNewProjectName] = useState('')
 49 |   const [projectToDelete, setProjectToDelete] = useState<FlowProject | null>(
 50 |     null
 51 |   )
 52 |   const [editingProject, setEditingProject] = useState<FlowProject | null>(null)
 53 |   const [editName, setEditName] = useState('')
 54 |   const { toast } = useToast()
 55 | 
 56 |   const handleCreateProject = () => {
 57 |     if (!newProjectName.trim()) {
 58 |       toast({
 59 |         title: 'Project name required',
 60 |         description: 'Please enter a name for your project',
 61 |         variant: 'destructive',
 62 |       })
 63 |       return
 64 |     }
 65 | 
 66 |     onCreateProject(newProjectName)
 67 |     setNewProjectName('')
 68 |     setIsCreateDialogOpen(false)
 69 | 
 70 |     toast({
 71 |       title: 'Project created',
 72 |       description: `"${newProjectName}" has been created successfully.`,
 73 |     })
 74 |   }
 75 | 
 76 |   const handleDeleteProject = () => {
 77 |     if (projectToDelete) {
 78 |       onDeleteProject(projectToDelete.id)
 79 |       setProjectToDelete(null)
 80 |       setIsDeleteDialogOpen(false)
 81 | 
 82 |       toast({
 83 |         title: 'Project deleted',
 84 |         description: `"${projectToDelete.name}" has been deleted.`,
 85 |       })
 86 |     }
 87 |   }
 88 | 
 89 |   const handleSaveRename = (project: FlowProject) => {
 90 |     if (editName.trim()) {
 91 |       onRenameProject(project.id, editName)
 92 | 
 93 |       toast({
 94 |         title: 'Project renamed',
 95 |         description: `Project has been renamed to "${editName}".`,
 96 |       })
 97 |     }
 98 |     setEditingProject(null)
 99 |   }
100 | 
101 |   const formatDate = (dateString: string) => {
102 |     const date = new Date(dateString)
103 |     return new Intl.DateTimeFormat('en-US', {
104 |       month: 'short',
105 |       day: 'numeric',
106 |       year: 'numeric',
107 |       hour: 'numeric',
108 |       minute: 'numeric',
109 |     }).format(date)
110 |   }
111 | 
112 |   const handleSelectProject = (project: FlowProject) => {
113 |     if (currentProject?.id !== project.id) {
114 |       onSelectProject(project)
115 | 
116 |       toast({
117 |         title: 'Switching projects',
118 |         description: `Loading "${project.name}"...`,
119 |       })
120 | 
121 |       setTimeout(() => {
122 |         window.location.reload()
123 |       }, 300)
124 |     }
125 |   }
126 | 
127 |   return (
128 |     <>
129 |       <TooltipProvider>
130 |         <DropdownMenu>
131 |           <Tooltip>
132 |             <TooltipTrigger asChild>
133 |               <DropdownMenuTrigger asChild>
134 |                 <Button
135 |                   variant='outline'
136 |                   size='sm'
137 |                   className='h-9 text-sm font-medium gap-1.5 px-2.5'
138 |                 >
139 |                   <FolderOpen className='h-4 w-4' />
140 |                   <span className='max-w-[120px] truncate'>
141 |                     {currentProject ? currentProject.name : 'Select Project'}
142 |                   </span>
143 |                 </Button>
144 |               </DropdownMenuTrigger>
145 |             </TooltipTrigger>
146 |             <TooltipContent>
147 |               <p>Switch or manage projects</p>
148 |             </TooltipContent>
149 |           </Tooltip>
150 | 
151 |           <DropdownMenuContent align='end' className='w-[260px]'>
152 |             <div className='px-2 py-1.5 text-sm font-medium text-gray-500 flex justify-between items-center'>
153 |               <span>Your Projects</span>
154 |               <span className='text-xs text-gray-400'>
155 |                 {projects.length} project{projects.length !== 1 ? 's' : ''}
156 |               </span>
157 |             </div>
158 |             <DropdownMenuSeparator />
159 | 
160 |             <div className='max-h-[300px] overflow-y-auto py-1'>
161 |               {projects.length === 0 ? (
162 |                 <div className='px-2 py-4 text-center text-sm text-gray-500'>
163 |                   No projects yet. Create your first one!
164 |                 </div>
165 |               ) : (
166 |                 projects.map((project) => (
167 |                   <div key={project.id} className='px-1 py-0.5'>
168 |                     {editingProject?.id === project.id ? (
169 |                       <div className='flex items-center gap-1 p-1'>
170 |                         <Input
171 |                           value={editName}
172 |                           onChange={(e) => setEditName(e.target.value)}
173 |                           className='h-7 text-xs'
174 |                           autoFocus
175 |                         />
176 |                         <Button
177 |                           variant='ghost'
178 |                           size='icon'
179 |                           className='h-6 w-6'
180 |                           onClick={() => handleSaveRename(project)}
181 |                         >
182 |                           <Check className='h-3 w-3' />
183 |                         </Button>
184 |                         <Button
185 |                           variant='ghost'
186 |                           size='icon'
187 |                           className='h-6 w-6'
188 |                           onClick={() => setEditingProject(null)}
189 |                         >
190 |                           <X className='h-3 w-3' />
191 |                         </Button>
192 |                       </div>
193 |                     ) : (
194 |                       <DropdownMenuItem
195 |                         className={`flex justify-between items-center p-2 ${
196 |                           currentProject?.id === project.id
197 |                             ? 'bg-gray-100 dark:bg-gray-800'
198 |                             : ''
199 |                         }`}
200 |                       >
201 |                         <div
202 |                           className='flex-1 overflow-hidden text-ellipsis'
203 |                           onClick={() => handleSelectProject(project)}
204 |                         >
205 |                           <div className='font-medium'>{project.name}</div>
206 |                           <div className='text-xs text-gray-500'>
207 |                             Updated {formatDate(project.updatedAt)}
208 |                           </div>
209 |                         </div>
210 |                         <div className='flex gap-1 ml-2'>
211 |                           <Button
212 |                             variant='ghost'
213 |                             size='icon'
214 |                             className='h-6 w-6'
215 |                             onClick={(e) => {
216 |                               e.stopPropagation()
217 |                               setEditingProject(project)
218 |                               setEditName(project.name)
219 |                             }}
220 |                           >
221 |                             <Edit className='h-3 w-3' />
222 |                           </Button>
223 |                           <Button
224 |                             variant='ghost'
225 |                             size='icon'
226 |                             className='h-6 w-6 text-red-500 hover:text-red-600'
227 |                             onClick={(e) => {
228 |                               e.stopPropagation()
229 |                               setProjectToDelete(project)
230 |                               setIsDeleteDialogOpen(true)
231 |                             }}
232 |                           >
233 |                             <Trash2 className='h-3 w-3' />
234 |                           </Button>
235 |                         </div>
236 |                       </DropdownMenuItem>
237 |                     )}
238 |                   </div>
239 |                 ))
240 |               )}
241 |             </div>
242 | 
243 |             <DropdownMenuSeparator />
244 |             <DropdownMenuItem
245 |               className='cursor-pointer text-primary focus:text-primary focus:bg-primary/10'
246 |               onClick={() => setIsCreateDialogOpen(true)}
247 |             >
248 |               <PlusCircle className='mr-2 h-4 w-4' />
249 |               Create New Project
250 |             </DropdownMenuItem>
251 |           </DropdownMenuContent>
252 |         </DropdownMenu>
253 |       </TooltipProvider>
254 | 
255 |       <Dialog open={isCreateDialogOpen} onOpenChange={setIsCreateDialogOpen}>
256 |         <DialogContent>
257 |           <DialogHeader>
258 |             <DialogTitle>Create New Project</DialogTitle>
259 |             <DialogDescription>
260 |               Enter a name for your new research project
261 |             </DialogDescription>
262 |           </DialogHeader>
263 |           <Input
264 |             value={newProjectName}
265 |             onChange={(e) => setNewProjectName(e.target.value)}
266 |             placeholder='Project name'
267 |             className='mt-4'
268 |             autoFocus
269 |           />
270 |           <DialogFooter className='mt-4'>
271 |             <Button
272 |               variant='outline'
273 |               onClick={() => setIsCreateDialogOpen(false)}
274 |             >
275 |               Cancel
276 |             </Button>
277 |             <Button onClick={handleCreateProject}>Create Project</Button>
278 |           </DialogFooter>
279 |         </DialogContent>
280 |       </Dialog>
281 | 
282 |       <Dialog open={isDeleteDialogOpen} onOpenChange={setIsDeleteDialogOpen}>
283 |         <DialogContent>
284 |           <DialogHeader>
285 |             <DialogTitle>Delete Project</DialogTitle>
286 |             <DialogDescription>
287 |               Are you sure you want to delete &quot;{projectToDelete?.name}
288 |               &quot;? This action cannot be undone.
289 |             </DialogDescription>
290 |           </DialogHeader>
291 |           <DialogFooter className='mt-4'>
292 |             <Button
293 |               variant='outline'
294 |               onClick={() => setIsDeleteDialogOpen(false)}
295 |             >
296 |               Cancel
297 |             </Button>
298 |             <Button variant='destructive' onClick={handleDeleteProject}>
299 |               Delete Project
300 |             </Button>
301 |           </DialogFooter>
302 |         </DialogContent>
303 |       </Dialog>
304 |     </>
305 |   )
306 | }
307 | 


--------------------------------------------------------------------------------