feat: implement image generation API route with extended timeout and error handling

app/api/v1/images/generate/route.ts

@@ -0,0 +1,54 @@
+import { NextRequest, NextResponse } from 'next/server';
+
+export const maxDuration = 300; // 5 minutes
+export const dynamic = 'force-dynamic';
+
+export async function POST(request: NextRequest) {
+  try {
+    const body = await request.json();
+
+    // Forward the request to the Python backend with extended timeout
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), 300000); // 5 minutes
+
+    const response = await fetch('http://localhost:8000/api/v1/images/generate', {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify(body),
+      signal: controller.signal,
+    });
+
+    clearTimeout(timeoutId);
+
+    if (!response.ok) {
+      throw new Error(`Backend responded with status: ${response.status}`);
+    }
+
+    const data = await response.json();
+    return NextResponse.json(data);
+
+  } catch (error) {
+    console.error('API route error:', error);
+    
+    if (error instanceof Error) {
+      if (error.name === 'AbortError') {
+        return NextResponse.json(
+          { success: false, message: 'Request timeout - the image generation is taking too long. Please try with fewer images or try again later.' },
+          { status: 408 }
+        );
+      }
+      
+      return NextResponse.json(
+        { success: false, message: `Request failed: ${error.message}` },
+        { status: 500 }
+      );
+    }
+
+    return NextResponse.json(
+      { success: false, message: 'An unexpected error occurred' },
+      { status: 500 }
+    );
+  }
+} 

backend/app/services/image_service.py

@@ -16,7 +16,7 @@ class ImageGenerationService:
     def __init__(self):
         self.client = OpenAI(
             api_key=os.getenv("OPENAI_API_KEY"),
-            timeout=60.0,  # Increase timeout for Hugging Face environment
+            timeout=300.0,  # 5 minutes timeout for multiple image generation
             max_retries=2,  # Reduce retries to fail faster
         )