refactor: improve OCR pipeline initialization

Co-Authored-By: ben <ben@prologe.io>
different-ai · Feb 14, 2025 · 15a0b72 · 15a0b72
1 parent 7d1c2b6
commit 15a0b72
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 20 deletions.
diff --git a/packages/web/app/api/ocr/init/route.ts b/packages/web/app/api/ocr/init/route.ts
@@ -1,15 +1 @@
-import { processOCRForInvoicesAndAdmin } from '@/lib/ocr/pipeline';
 
-// Initialize the OCR pipeline when this route is first accessed
-let initialized = false;
-
-export async function GET() {
-  if (!initialized) {
-    initialized = true;
-    // Start the OCR pipeline
-    setInterval(() => {
-      processOCRForInvoicesAndAdmin().catch(console.error);
-    }, 60000);
-  }
-  return new Response('OCR pipeline initialized', { status: 200 });
-}
diff --git a/packages/web/app/layout.tsx b/packages/web/app/layout.tsx
@@ -41,13 +41,10 @@ export default async function RootLayout({
 }: Readonly<{
   children: React.ReactNode;
 }>) {
-  // Initialize OCR pipeline
+  // Initialize OCR pipeline on server-side only
   if (process.env.NODE_ENV === 'production') {
-    try {
-      await fetch('/api/ocr/init');
-    } catch (error) {
-      console.error('Failed to initialize OCR pipeline:', error);
-    }
+    const { initializeOCRPipeline } = require('@/lib/ocr/init');
+    initializeOCRPipeline();
   }
   return (
     <html

diff --git a/packages/web/lib/ocr/init.ts b/packages/web/lib/ocr/init.ts
@@ -0,0 +1,14 @@
+import { processOCRForInvoicesAndAdmin } from './pipeline';
+
+let initialized = false;
+
+export function initializeOCRPipeline() {
+  if (!initialized && typeof window === 'undefined') {
+    initialized = true;
+    // Start the OCR pipeline
+    setInterval(() => {
+      processOCRForInvoicesAndAdmin().catch(console.error);
+    }, 60000);
+    console.log('OCR pipeline initialized');
+  }
+}