Clean up route file

app/api/generate/route.ts

@@ -1,6 +1,6 @@
 // app/api/generate/route.ts
 import { NextRequest, NextResponse } from 'next/server';
-import { processAppRequest } from '@/lib/openai';
+import { generateAppCode } from '@/lib/openai';
 import { createSandbox } from '@/lib/e2b';
 import { componentSchema } from '@/lib/schemas';
 import { Benchify } from 'benchify';
@@ -22,35 +22,40 @@ const extendedComponentSchema = componentSchema.extend({
 
 export async function POST(request: NextRequest) {
     try {
-        const body = await request.json();
 
-        // Validate the request using extended schema
-        const validationResult = extendedComponentSchema.safeParse(body);
+        // Validate the request
+        const validation = await validateRequest(request);
 
-        if (!validationResult.success) {
-            return NextResponse.json(
-                { error: 'Invalid request format', details: validationResult.error.format() },
-                { status: 400 }
-            );
+        if (!validation.success) {
+            return validation.error;
         }
 
-        const { description, existingFiles, editInstruction, useBuggyCode, useFixer } = validationResult.data;
+        const { description, existingFiles, editInstruction, useBuggyCode, useFixer } = validation.data;
 
         // Process the app request using centralized logic
-        const filesToSandbox = await processAppRequest(description, existingFiles, editInstruction, useBuggyCode);
+        const filesToSandbox = await generateAppCode(description, existingFiles, editInstruction, useBuggyCode);
 
         let repairedFiles = filesToSandbox;
 
         // Repair the generated code using Benchify's API if requested
         if (useFixer) {
             try {
+                console.log("Trying fixer")
                 const { data } = await benchify.fixer.run({
                     files: filesToSandbox.map((file: { path: string; content: string }) => ({
                         path: file.path,
                         contents: file.content
-                    }))
+                    })),
+                    fixes: {
+                        css: false,
+                        imports: false,
+                        stringLiterals: true,
+                        tsSuggestions: false
+                    }
                 });
 
+                console.log('🔧 Benchify fixer data:', data);
+
                 if (data) {
                     const { success, diff } = data;
 
@@ -90,4 +95,25 @@ export async function POST(request: NextRequest) {
             { status: 500 }
         );
     }
-}
+}
+
+
+async function validateRequest(request: NextRequest) {
+    const body = await request.json();
+    const validationResult = extendedComponentSchema.safeParse(body);
+
+    if (!validationResult.success) {
+        return {
+            success: false as const,
+            error: NextResponse.json(
+                { error: 'Invalid request format', details: validationResult.error.format() },
+                { status: 400 }
+            )
+        };
+    }
+
+    return {
+        success: true as const,
+        data: validationResult.data
+    };
+}

lib/openai.ts

@@ -18,7 +18,7 @@ const fileSchema = z.object({
 });
 
 // Generate a new application using AI SDK
-export async function generateApp(
+export async function createNewApp(
     description: string,
 ): Promise<Array<{ path: string; content: string }>> {
     console.log("Creating app with description: ", description);
@@ -108,7 +108,7 @@ export async function editApp(
 }
 
 // Main function to handle both generation and editing
-export async function processAppRequest(
+export async function generateAppCode(
     description: string,
     existingFiles?: z.infer<typeof benchifyFileSchema>,
     editInstruction?: string,
@@ -147,7 +147,7 @@ export default App;`
             ];
         } else {
             console.log('🤖 Calling AI to generate app...');
-            return await generateApp(description);
+            return await createNewApp(description);
         }
     }
 }